85 lines
2.1 KiB
Go
85 lines
2.1 KiB
Go
package database
|
|
|
|
import (
|
|
. "github.com/drone/drone/pkg/model"
|
|
"github.com/russross/meddler"
|
|
)
|
|
|
|
// Name of the Build table in the database
|
|
const buildTable = "builds"
|
|
|
|
// SQL Queries to retrieve a list of all Commits belonging to a Repo.
|
|
const buildStmt = `
|
|
SELECT id, commit_id, slug, status, started, finished, duration, created, updated, stdout, buildscript
|
|
FROM builds
|
|
WHERE commit_id = ?
|
|
ORDER BY slug ASC
|
|
`
|
|
|
|
// SQL Queries to retrieve a Build by id.
|
|
const buildFindStmt = `
|
|
SELECT id, commit_id, slug, status, started, finished, duration, created, updated, stdout, buildscript
|
|
FROM builds
|
|
WHERE id = ?
|
|
LIMIT 1
|
|
`
|
|
|
|
// SQL Queries to retrieve a Commit by name and repo id.
|
|
const buildFindSlugStmt = `
|
|
SELECT id, commit_id, slug, status, started, finished, duration, created, updated, stdout, buildscript
|
|
FROM builds
|
|
WHERE slug = ? AND commit_id = ?
|
|
LIMIT 1
|
|
`
|
|
|
|
// SQL Queries to fail all builds that are running or pending
|
|
const buildFailUnfinishedStmt = `
|
|
UPDATE builds
|
|
SET status = 'Failure'
|
|
WHERE status IN ('Started', 'Pending')
|
|
`
|
|
|
|
// SQL Queries to delete a Commit.
|
|
const buildDeleteStmt = `
|
|
DELETE FROM builds WHERE id = ?
|
|
`
|
|
|
|
// Returns the Build with the given ID.
|
|
func GetBuild(id int64) (*Build, error) {
|
|
build := Build{}
|
|
err := meddler.QueryRow(db, &build, buildFindStmt, id)
|
|
return &build, err
|
|
}
|
|
|
|
// Returns the Build with the given slug.
|
|
func GetBuildSlug(slug string, commit int64) (*Build, error) {
|
|
build := Build{}
|
|
err := meddler.QueryRow(db, &build, buildFindSlugStmt, slug, commit)
|
|
return &build, err
|
|
}
|
|
|
|
// Creates a new Build.
|
|
func SaveBuild(build *Build) error {
|
|
return meddler.Save(db, buildTable, build)
|
|
}
|
|
|
|
// Deletes an existing Build.
|
|
func DeleteBuild(id int64) error {
|
|
_, err := db.Exec(buildDeleteStmt, id)
|
|
return err
|
|
}
|
|
|
|
// Returns a list of all Builds associated
|
|
// with the specified Commit ID and branch.
|
|
func ListBuilds(id int64) ([]*Build, error) {
|
|
var builds []*Build
|
|
err := meddler.QueryAll(db, &builds, buildStmt, id)
|
|
return builds, err
|
|
}
|
|
|
|
// FailUnfinishedBuilds sets status=Failure to all builds
|
|
// in the Pending and Started states
|
|
func FailUnfinishedBuilds() error {
|
|
_, err := db.Exec(buildFailUnfinishedStmt)
|
|
return err
|
|
}
|