package model import ( "time" "github.com/drone/drone/shared/database" "github.com/russross/meddler" ) type Build struct { ID int64 `json:"id" meddler:"build_id,pk"` RepoID int64 `json:"-" meddler:"build_repo_id"` Number int `json:"number" meddler:"build_number"` Event string `json:"event" meddler:"build_event"` Status string `json:"status" meddler:"build_status"` Enqueued int64 `json:"enqueued_at" meddler:"build_enqueued"` Created int64 `json:"created_at" meddler:"build_created"` Started int64 `json:"started_at" meddler:"build_started"` Finished int64 `json:"finished_at" meddler:"build_finished"` Commit string `json:"commit" meddler:"build_commit"` Branch string `json:"branch" meddler:"build_branch"` Ref string `json:"ref" meddler:"build_ref"` Refspec string `json:"refspec" meddler:"build_refspec"` Remote string `json:"remote" meddler:"build_remote"` Title string `json:"title" meddler:"build_title"` Message string `json:"message" meddler:"build_message"` Timestamp int64 `json:"timestamp" meddler:"build_timestamp"` Author string `json:"author" meddler:"build_author"` Avatar string `json:"author_avatar" meddler:"build_avatar"` Email string `json:"author_email" meddler:"build_email"` Link string `json:"link_url" meddler:"build_link"` } type BuildGroup struct { Date string Builds []*Build } func GetBuild(db meddler.DB, id int64) (*Build, error) { var build = new(Build) var err = meddler.Load(db, buildTable, build, id) return build, err } func GetBuildNumber(db meddler.DB, repo *Repo, number int) (*Build, error) { var build = new(Build) var err = meddler.QueryRow(db, build, database.Rebind(buildNumberQuery), repo.ID, number) return build, err } func GetBuildRef(db meddler.DB, repo *Repo, ref string) (*Build, error) { var build = new(Build) var err = meddler.QueryRow(db, build, database.Rebind(buildRefQuery), repo.ID, ref) return build, err } func GetBuildCommit(db meddler.DB, repo *Repo, sha, branch string) (*Build, error) { var build = new(Build) var err = meddler.QueryRow(db, build, database.Rebind(buildCommitQuery), repo.ID, sha, branch) return build, err } func GetBuildLast(db meddler.DB, repo *Repo, branch string) (*Build, error) { var build = new(Build) var err = meddler.QueryRow(db, build, database.Rebind(buildLastQuery), repo.ID, branch) return build, err } func GetBuildLastBefore(db meddler.DB, repo *Repo, branch string, number int64) (*Build, error) { var build = new(Build) var err = meddler.QueryRow(db, build, database.Rebind(buildLastBeforeQuery), repo.ID, branch, number) return build, err } func GetBuildList(db meddler.DB, repo *Repo) ([]*Build, error) { var builds = []*Build{} var err = meddler.QueryAll(db, &builds, database.Rebind(buildListQuery), repo.ID) return builds, err } func CreateBuild(db meddler.DB, build *Build, jobs ...*Job) error { var number int db.QueryRow(buildNumberLast, build.RepoID).Scan(&number) build.Number = number + 1 build.Created = time.Now().UTC().Unix() build.Enqueued = build.Created err := meddler.Insert(db, buildTable, build) if err != nil { return err } for i, job := range jobs { job.BuildID = build.ID job.Number = i + 1 job.Enqueued = build.Created err = InsertJob(db, job) if err != nil { return err } } return nil } func UpdateBuild(db meddler.DB, build *Build) error { return meddler.Update(db, buildTable, build) } const buildTable = "builds" const buildListQuery = ` SELECT * FROM builds WHERE build_repo_id = ? ORDER BY build_number DESC LIMIT 50 ` const buildNumberQuery = ` SELECT * FROM builds WHERE build_repo_id = ? AND build_number = ? LIMIT 1; ` const buildLastQuery = ` SELECT * FROM builds WHERE build_repo_id = ? AND build_branch = ? ORDER BY build_number DESC LIMIT 1 ` const buildLastBeforeQuery = ` SELECT * FROM builds WHERE build_repo_id = ? AND build_branch = ? AND build_id < ? ORDER BY build_number DESC LIMIT 1 ` const buildCommitQuery = ` SELECT * FROM builds WHERE build_repo_id = ? AND build_commit = ? AND build_branch = ? LIMIT 1 ` const buildRefQuery = ` SELECT * FROM builds WHERE build_repo_id = ? AND build_ref = ? LIMIT 1 ` const buildNumberLast = ` SELECT MAX(build_number) FROM builds WHERE build_repo_id = ? `