out-of-tree/daemon/db/job.go

package db

import (
	"database/sql"
	"encoding/json"

	"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,
		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) ` +
		`VALUES ($1, $2, $3, $4, $5, $6);`)
	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,
	)
	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, status=$7 WHERE id=$8`)
	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.Status, job.ID)
	return
}

func Jobs(db *sql.DB) (jobs []api.Job, err error) {
	stmt, err := db.Prepare(`SELECT id, uuid, group_uuid, repo, "commit", ` +
		`config, target, 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
		var config, target []byte
		err = rows.Scan(&job.ID, &job.UUID, &job.Group,
			&job.RepoName, &job.Commit,
			&config, &target, &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
		}

		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, status ` +
		`FROM job WHERE uuid=$1`)
	if err != nil {
		return
	}
	defer stmt.Close()

	err = stmt.QueryRow(uuid).Scan(&job.ID, &job.UUID,
		&job.Group, &job.RepoName, &job.Commit,
		&job.Artifact, &job.Target, &job.Status)
	if err != nil {
		return
	}

	return
}

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
}