Refactored to use Model and Database folders

server/database/commit.go

@@ -0,0 +1,236 @@
+package database
+
+import (
+	"database/sql"
+	"time"
+
+	"github.com/drone/drone/shared/model"
+	"github.com/russross/meddler"
+)
+
+type CommitManager interface {
+	// Find finds the commit by ID.
+	Find(id int64) (*model.Commit, error)
+
+	// FindSha finds the commit for the branch and sha.
+	FindSha(repo int64, branch, sha string) (*model.Commit, error)
+
+	// FindLatest finds the most recent commit for the branch.
+	FindLatest(repo int64, branch string) (*model.Commit, error)
+
+	// FindOutput finds the commit's output.
+	FindOutput(commit int64) ([]byte, error)
+
+	// List finds recent commits for the repository
+	List(repo int64) ([]*model.Commit, error)
+
+	// ListBranch finds recent commits for the repository and branch.
+	ListBranch(repo int64, branch string) ([]*model.Commit, error)
+
+	// ListBranches finds most recent commit for each branch.
+	ListBranches(repo int64) ([]*model.Commit, error)
+
+	// ListUser finds most recent commits for a user.
+	ListUser(repo int64) ([]*model.CommitRepo, error)
+
+	// Insert persists the commit to the datastore.
+	Insert(commit *model.Commit) error
+
+	// Update persists changes to the commit to the datastore.
+	Update(commit *model.Commit) error
+
+	// UpdateOutput persists a commit's stdout to the datastore.
+	UpdateOutput(commit *model.Commit, out []byte) error
+
+	// Delete removes the commit from the datastore.
+	Delete(commit *model.Commit) error
+
+	// CancelAll will update the status of all Started or Pending
+	// builds to a status of Killed (cancelled).
+	CancelAll() error
+}
+
+// commitManager manages a list of commits in a SQL database.
+type commitManager struct {
+	*sql.DB
+}
+
+// NewCommitManager initiales a new CommitManager intended to
+// manage and persist commits.
+func NewCommitManager(db *sql.DB) CommitManager {
+	return &commitManager{db}
+}
+
+// SQL query to retrieve the latest Commits for each branch.
+const listBranchesQuery = `
+SELECT *
+FROM commits
+WHERE commit_id IN (
+    SELECT MAX(commit_id)
+    FROM commits
+    WHERE repo_id=?
+    AND commit_status NOT IN ('Started', 'Pending')
+    GROUP BY commit_branch)
+ ORDER BY commit_branch ASC
+ `
+
+// SQL query to retrieve the latest Commits for a specific branch.
+const listBranchQuery = `
+SELECT *
+FROM commits
+WHERE repo_id=?
+AND   commit_branch=?
+ORDER BY commit_id DESC
+LIMIT 20
+ `
+
+// SQL query to retrieve the latest Commits for a user's repositories.
+const listUserCommitsQuery = `
+SELECT r.repo_remote, r.repo_host, r.repo_owner, r.repo_name, c.*
+FROM commits c, repos r, perms p
+WHERE c.repo_id=r.repo_id
+AND   r.repo_id=p.repo_id
+AND   p.user_id=?
+AND   c.commit_status NOT IN ('Started', 'Pending')
+ORDER BY commit_id DESC
+LIMIT 20
+`
+
+// SQL query to retrieve the latest Commits across all branches.
+const listCommitsQuery = `
+SELECT *
+FROM commits
+WHERE repo_id=? 
+ORDER BY commit_id DESC
+LIMIT 20
+`
+
+// SQL query to retrieve a Commit by branch and sha.
+const findCommitQuery = `
+SELECT *
+FROM commits
+WHERE repo_id=?
+AND   commit_branch=?
+AND   commit_sha=?
+LIMIT 1
+`
+
+// SQL query to retrieve the most recent Commit for a branch.
+const findLatestCommitQuery = `
+SELECT *
+FROM commits
+WHERE commit_id IN (
+    SELECT MAX(commit_id)
+    FROM commits
+    WHERE repo_id=?
+    AND commit_branch=?)
+`
+
+// SQL query to retrieve a Commit's stdout.
+const findOutputQuery = `
+SELECT output_raw
+FROM output
+WHERE commit_id = ?
+`
+
+// SQL statement to insert a Commit's stdout.
+const insertOutputStmt = `
+INSERT INTO output (commit_id, output_raw) values (?,?);
+`
+
+// SQL statement to update a Commit's stdout.
+const updateOutputStmt = `
+UPDATE output SET output_raw = ? WHERE commit_id = ?;
+`
+
+// SQL statement to delete a Commit by ID.
+const deleteCommitStmt = `
+DELETE FROM commits WHERE commit_id = ?;
+`
+
+// SQL statement to cancel all running Commits.
+const cancelCommitStmt = `
+UPDATE commits SET
+commit_status = ?,
+commit_started = ?,
+commit_finished = ?
+WHERE commit_status IN ('Started', 'Pending');
+`
+
+func (db *commitManager) Find(id int64) (*model.Commit, error) {
+	dst := model.Commit{}
+	err := meddler.Load(db, "commits", &dst, id)
+	return &dst, err
+}
+
+func (db *commitManager) FindSha(repo int64, branch, sha string) (*model.Commit, error) {
+	dst := model.Commit{}
+	err := meddler.QueryRow(db, &dst, findCommitQuery, repo, branch, sha)
+	return &dst, err
+}
+
+func (db *commitManager) FindLatest(repo int64, branch string) (*model.Commit, error) {
+	dst := model.Commit{}
+	err := meddler.QueryRow(db, &dst, findLatestCommitQuery, repo, branch)
+	return &dst, err
+}
+
+func (db *commitManager) FindOutput(commit int64) ([]byte, error) {
+	var dst string
+	err := db.QueryRow(findOutputQuery, commit).Scan(&dst)
+	return []byte(dst), err
+}
+
+func (db *commitManager) List(repo int64) ([]*model.Commit, error) {
+	var dst []*model.Commit
+	err := meddler.QueryAll(db, &dst, listCommitsQuery, repo)
+	return dst, err
+}
+
+func (db *commitManager) ListBranch(repo int64, branch string) ([]*model.Commit, error) {
+	var dst []*model.Commit
+	err := meddler.QueryAll(db, &dst, listBranchQuery, repo, branch)
+	return dst, err
+}
+
+func (db *commitManager) ListBranches(repo int64) ([]*model.Commit, error) {
+	var dst []*model.Commit
+	err := meddler.QueryAll(db, &dst, listBranchesQuery, repo)
+	return dst, err
+}
+
+func (db *commitManager) ListUser(user int64) ([]*model.CommitRepo, error) {
+	var dst []*model.CommitRepo
+	err := meddler.QueryAll(db, &dst, listUserCommitsQuery, user)
+	return dst, err
+}
+
+func (db *commitManager) Insert(commit *model.Commit) error {
+	commit.Created = time.Now().Unix()
+	commit.Updated = time.Now().Unix()
+	return meddler.Insert(db, "commits", commit)
+}
+
+func (db *commitManager) Update(commit *model.Commit) error {
+	commit.Updated = time.Now().Unix()
+	return meddler.Update(db, "commits", commit)
+}
+
+func (db *commitManager) UpdateOutput(commit *model.Commit, out []byte) error {
+	_, err := db.Exec(insertOutputStmt, commit.ID, out)
+	if err != nil {
+		return nil
+	}
+	_, err = db.Exec(updateOutputStmt, out, commit.ID)
+	return err
+}
+
+func (db *commitManager) Delete(commit *model.Commit) error {
+	_, err := db.Exec(deleteCommitStmt, commit.ID)
+	return err
+}
+
+func (db *commitManager) CancelAll() error {
+	_, err := db.Exec(cancelCommitStmt, model.StatusKilled, time.Now().Unix(), time.Now().Unix())
+	return err
+}