56 lines
1.1 KiB
Go
56 lines
1.1 KiB
Go
|
package datastore
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
|
||
|
|
"github.com/drone/drone/model"
|
||
|
|
"github.com/russross/meddler"
|
||
|
|
)
|
||
|
|
|
||
|
|
type jobstore struct {
|
||
|
|
*sql.DB
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *jobstore) Get(id int64) (*model.Job, error) {
|
||
|
|
var job = new(model.Job)
|
||
|
|
var err = meddler.Load(db, jobTable, job, id)
|
||
|
|
return job, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *jobstore) GetNumber(build *model.Build, num int) (*model.Job, error) {
|
||
|
|
var job = new(model.Job)
|
||
|
|
var err = meddler.QueryRow(db, job, rebind(jobNumberQuery), build.ID, num)
|
||
|
|
return job, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *jobstore) GetList(build *model.Build) ([]*model.Job, error) {
|
||
|
|
var jobs = []*model.Job{}
|
||
|
|
var err = meddler.QueryAll(db, &jobs, rebind(jobListQuery), build.ID)
|
||
|
|
return jobs, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *jobstore) Create(job *model.Job) error {
|
||
|
|
return meddler.Insert(db, jobTable, job)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *jobstore) Update(job *model.Job) error {
|
||
|
|
return meddler.Update(db, jobTable, job)
|
||
|
|
}
|
||
|
|
|
||
|
|
const jobTable = "jobs"
|
||
|
|
|
||
|
|
const jobListQuery = `
|
||
|
|
SELECT *
|
||
|
|
FROM jobs
|
||
|
|
WHERE job_build_id = ?
|
||
|
|
ORDER BY job_number ASC
|
||
|
|
`
|
||
|
|
|
||
|
|
const jobNumberQuery = `
|
||
|
|
SELECT *
|
||
|
|
FROM jobs
|
||
|
|
WHERE job_build_id = ?
|
||
|
|
AND job_number = ?
|
||
|
|
LIMIT 1
|
||
|
|
`
|