160 lines
3.1 KiB
Go
160 lines
3.1 KiB
Go
package db
|
|
|
|
import (
|
|
"database/sql"
|
|
"encoding/json"
|
|
"time"
|
|
|
|
"code.dumpstack.io/tools/out-of-tree/api"
|
|
)
|
|
|
|
func createJobTable(db *sql.DB) (err error) {
|
|
_, err = db.Exec(`
|
|
CREATE TABLE IF NOT EXISTS job (
|
|
id INTEGER PRIMARY KEY,
|
|
uuid TEXT,
|
|
group_uuid TEXT,
|
|
repo TEXT,
|
|
"commit" TEXT,
|
|
config TEXT,
|
|
target TEXT,
|
|
created INT,
|
|
started INT,
|
|
finished INT,
|
|
status TEXT DEFAULT "new"
|
|
)`)
|
|
return
|
|
}
|
|
|
|
func AddJob(db *sql.DB, job *api.Job) (err error) {
|
|
stmt, err := db.Prepare(`INSERT INTO job (uuid, group_uuid, repo, "commit", ` +
|
|
`config, target, created, started, finished) ` +
|
|
`VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9);`)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
defer stmt.Close()
|
|
|
|
config := api.Marshal(job.Artifact)
|
|
target := api.Marshal(job.Target)
|
|
|
|
res, err := stmt.Exec(job.UUID, job.Group,
|
|
job.RepoName, job.Commit, config, target,
|
|
job.Created.Unix(), job.Started.Unix(),
|
|
job.Finished.Unix(),
|
|
)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
job.ID, err = res.LastInsertId()
|
|
return
|
|
}
|
|
|
|
func UpdateJob(db *sql.DB, job *api.Job) (err error) {
|
|
stmt, err := db.Prepare(`UPDATE job ` +
|
|
`SET uuid=$1, group_uuid=$2, repo=$3, ` +
|
|
`"commit"=$4, config=$5, target=$6, ` +
|
|
`created=$7, started=$8, finished=$9, ` +
|
|
`status=$10 ` +
|
|
`WHERE id=$11`)
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer stmt.Close()
|
|
|
|
config := api.Marshal(job.Artifact)
|
|
target := api.Marshal(job.Target)
|
|
|
|
_, err = stmt.Exec(job.UUID, job.Group,
|
|
job.RepoName, job.Commit,
|
|
config, target,
|
|
job.Created.Unix(), job.Started.Unix(),
|
|
job.Finished.Unix(), job.Status, job.ID)
|
|
return
|
|
}
|
|
|
|
func scanJob(scan func(dest ...any) error) (job api.Job, err error) {
|
|
var config, target []byte
|
|
var created, started, finished int64
|
|
err = scan(&job.ID, &job.UUID, &job.Group,
|
|
&job.RepoName, &job.Commit, &config, &target,
|
|
&created, &started, &finished, &job.Status)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
err = json.Unmarshal(config, &job.Artifact)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
err = json.Unmarshal(target, &job.Target)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
job.Created = time.Unix(created, 0)
|
|
job.Started = time.Unix(started, 0)
|
|
job.Finished = time.Unix(finished, 0)
|
|
return
|
|
}
|
|
|
|
func Jobs(db *sql.DB) (jobs []api.Job, err error) {
|
|
stmt, err := db.Prepare(`SELECT id, uuid, group_uuid, ` +
|
|
`repo, "commit", config, target, created, ` +
|
|
`started, finished, status FROM job`)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
defer stmt.Close()
|
|
|
|
rows, err := stmt.Query()
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer rows.Close()
|
|
|
|
for rows.Next() {
|
|
var job api.Job
|
|
job, err = scanJob(rows.Scan)
|
|
if err != nil {
|
|
return
|
|
}
|
|
jobs = append(jobs, job)
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func Job(db *sql.DB, uuid string) (job api.Job, err error) {
|
|
stmt, err := db.Prepare(`SELECT id, uuid, group_uuid, ` +
|
|
`repo, "commit", config, target, ` +
|
|
`created, started, finished, status ` +
|
|
`FROM job WHERE uuid=$1`)
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer stmt.Close()
|
|
|
|
return scanJob(stmt.QueryRow(uuid).Scan)
|
|
}
|
|
|
|
func JobStatus(db *sql.DB, uuid string) (st api.Status, err error) {
|
|
stmt, err := db.Prepare(`SELECT status FROM job ` +
|
|
`WHERE uuid=$1`)
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer stmt.Close()
|
|
|
|
err = stmt.QueryRow(uuid).Scan(&st)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
return
|
|
}
|