package database import ( "database/sql" "github.com/drone/drone/shared/model" "github.com/russross/meddler" ) type RemoteManager interface { // Find finds the Remote by ID. Find(id int64) (*model.Remote, error) // FindHost finds the Remote by hostname. FindHost(name string) (*model.Remote, error) // List finds all registered Remotes of the system. List() ([]*model.Remote, error) // Insert persists the Remotes to the datastore. Insert(server *model.Remote) error // Update persists changes to the Remotes to the datastore. Update(server *model.Remote) error // Delete removes the Remotes from the datastore. Delete(server *model.Remote) error } // remoteManager manages a list of remotes in a SQL database. type remoteManager struct { *sql.DB } // SQL query to retrieve a Remote by remote login. const findRemoteQuery = ` SELECT * FROM remotes WHERE remote_host=? LIMIT 1 ` // SQL query to retrieve a list of all Remotes. const listRemoteQuery = ` SELECT * FROM remotes ` // SQL statement to delete a Remote by ID. const deleteRemoteStmt = ` DELETE FROM remotes WHERE remote_id=? ` // NewRemoteManager initiales a new RemoteManager intended to // manage and persist servers. func NewRemoteManager(db *sql.DB) RemoteManager { return &remoteManager{db} } func (db *remoteManager) Find(id int64) (*model.Remote, error) { dst := model.Remote{} err := meddler.Load(db, "remotes", &dst, id) return &dst, err } func (db *remoteManager) FindHost(host string) (*model.Remote, error) { dst := model.Remote{} err := meddler.QueryRow(db, &dst, findRemoteQuery, host) return &dst, err } func (db *remoteManager) List() ([]*model.Remote, error) { var dst []*model.Remote err := meddler.QueryAll(db, &dst, listRemoteQuery) return dst, err } func (db *remoteManager) Insert(remote *model.Remote) error { return meddler.Insert(db, "remotes", remote) } func (db *remoteManager) Update(remote *model.Remote) error { return meddler.Update(db, "remotes", remote) } func (db *remoteManager) Delete(remote *model.Remote) error { _, err := db.Exec(deleteRemoteStmt, remote.ID) return err }