harness-drone/store/step/step.go
Eoin McAfee df2da1c646
Feature/dron 101 cards 2 (#3158)
* refactor create / find / delete end points for cards
2021-10-26 08:10:01 +01:00

236 lines
5.1 KiB
Go

// Copyright 2019 Drone IO, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package step
import (
"context"
"github.com/drone/drone/core"
"github.com/drone/drone/store/shared/db"
)
// New returns a new StepStore.
func New(db *db.DB) core.StepStore {
return &stepStore{db}
}
type stepStore struct {
db *db.DB
}
func (s *stepStore) List(ctx context.Context, id int64) ([]*core.Step, error) {
var out []*core.Step
err := s.db.View(func(queryer db.Queryer, binder db.Binder) error {
params := map[string]interface{}{"step_stage_id": id}
stmt, args, err := binder.BindNamed(queryStage, params)
if err != nil {
return err
}
rows, err := queryer.Query(stmt, args...)
if err != nil {
return err
}
out, err = scanRows(rows)
return err
})
return out, err
}
func (s *stepStore) Find(ctx context.Context, id int64) (*core.Step, error) {
out := &core.Step{ID: id}
err := s.db.View(func(queryer db.Queryer, binder db.Binder) error {
params := toParams(out)
query, args, err := binder.BindNamed(queryKey, params)
if err != nil {
return err
}
row := queryer.QueryRow(query, args...)
return scanRow(row, out)
})
return out, err
}
func (s *stepStore) FindNumber(ctx context.Context, id int64, number int) (*core.Step, error) {
out := &core.Step{StageID: id, Number: number}
err := s.db.View(func(queryer db.Queryer, binder db.Binder) error {
params := toParams(out)
query, args, err := binder.BindNamed(queryNumber, params)
if err != nil {
return err
}
row := queryer.QueryRow(query, args...)
return scanRow(row, out)
})
return out, err
}
func (s *stepStore) Create(ctx context.Context, step *core.Step) error {
if s.db.Driver() == db.Postgres {
return s.createPostgres(ctx, step)
}
return s.create(ctx, step)
}
func (s *stepStore) create(ctx context.Context, step *core.Step) error {
step.Version = 1
return s.db.Lock(func(execer db.Execer, binder db.Binder) error {
params := toParams(step)
stmt, args, err := binder.BindNamed(stmtInsert, params)
if err != nil {
return err
}
res, err := execer.Exec(stmt, args...)
if err != nil {
return err
}
step.ID, err = res.LastInsertId()
return err
})
}
func (s *stepStore) createPostgres(ctx context.Context, step *core.Step) error {
step.Version = 1
return s.db.Lock(func(execer db.Execer, binder db.Binder) error {
params := toParams(step)
stmt, args, err := binder.BindNamed(stmtInsertPg, params)
if err != nil {
return err
}
return execer.QueryRow(stmt, args...).Scan(&step.ID)
})
}
func (s *stepStore) Update(ctx context.Context, step *core.Step) error {
versionNew := step.Version + 1
versionOld := step.Version
err := s.db.Lock(func(execer db.Execer, binder db.Binder) error {
params := toParams(step)
params["step_version_old"] = versionOld
params["step_version_new"] = versionNew
stmt, args, err := binder.BindNamed(stmtUpdate, params)
if err != nil {
return err
}
res, err := execer.Exec(stmt, args...)
if err != nil {
return err
}
effected, err := res.RowsAffected()
if err != nil {
return err
}
if effected == 0 {
return db.ErrOptimisticLock
}
return nil
})
if err == nil {
step.Version = versionNew
}
return err
}
const queryBase = `
SELECT
step_id
,step_stage_id
,step_number
,step_name
,step_status
,step_error
,step_errignore
,step_exit_code
,step_started
,step_stopped
,step_version
,step_depends_on
,step_image
,step_detached
,step_schema
`
const queryKey = queryBase + `
FROM steps
WHERE step_id = :step_id
`
const queryNumber = queryBase + `
FROM steps
WHERE step_stage_id = :step_stage_id
AND step_number = :step_number
`
const queryStage = queryBase + `
FROM steps
WHERE step_stage_id = :step_stage_id
`
const stmtUpdate = `
UPDATE steps
SET
step_name = :step_name
,step_status = :step_status
,step_error = :step_error
,step_errignore = :step_errignore
,step_exit_code = :step_exit_code
,step_started = :step_started
,step_stopped = :step_stopped
,step_version = :step_version_new
,step_depends_on = :step_depends_on
,step_image = :step_image
,step_detached = :step_detached
,step_schema = :step_schema
WHERE step_id = :step_id
AND step_version = :step_version_old
`
const stmtInsert = `
INSERT INTO steps (
step_stage_id
,step_number
,step_name
,step_status
,step_error
,step_errignore
,step_exit_code
,step_started
,step_stopped
,step_version
,step_depends_on
,step_image
,step_detached
,step_schema
) VALUES (
:step_stage_id
,:step_number
,:step_name
,:step_status
,:step_error
,:step_errignore
,:step_exit_code
,:step_started
,:step_stopped
,:step_version
,:step_depends_on
,:step_image
,:step_detached
,:step_schema
)
`
const stmtInsertPg = stmtInsert + `
RETURNING step_id
`