83577a7d5d
removed amber files. replacing with angular removed queue package in favor or worker package removed channel package in favor of pubsub package
111 lines
2.6 KiB
Go
111 lines
2.6 KiB
Go
package database
|
|
|
|
import (
|
|
"database/sql"
|
|
|
|
"github.com/drone/drone/shared/model"
|
|
"github.com/russross/meddler"
|
|
)
|
|
|
|
type ServerManager interface {
|
|
// Find finds the Server by ID.
|
|
Find(id int64) (*model.Server, error)
|
|
|
|
// FindName finds the Server by name.
|
|
FindName(name string) (*model.Server, error)
|
|
|
|
// FindName finds the Server by name.
|
|
FindSMTP() (*model.SMTPServer, error)
|
|
|
|
// List finds all registered Servers of the system.
|
|
List() ([]*model.Server, error)
|
|
|
|
// Insert persists the Server to the datastore.
|
|
Insert(server *model.Server) error
|
|
|
|
// Update persists changes to the Server to the datastore.
|
|
Update(server *model.Server) error
|
|
|
|
// UpdateSMTP persists changes to the SMTP Server to the datastore.
|
|
UpdateSMTP(server *model.SMTPServer) error
|
|
|
|
// Delete removes the Server from the datastore.
|
|
Delete(server *model.Server) error
|
|
}
|
|
|
|
// serverManager manages a list of users in a SQL database.
|
|
type serverManager struct {
|
|
*sql.DB
|
|
}
|
|
|
|
// SQL query to retrieve a Server by remote login.
|
|
const findServerQuery = `
|
|
SELECT *
|
|
FROM servers
|
|
WHERE server_name=?
|
|
LIMIT 1
|
|
`
|
|
|
|
// SQL query to retrieve a list of all Servers.
|
|
const listServerQuery = `
|
|
SELECT *
|
|
FROM servers
|
|
`
|
|
|
|
// SQL statement to delete a Server by ID.
|
|
const deleteServerStmt = `
|
|
DELETE FROM servers WHERE server_id=?
|
|
`
|
|
|
|
// NewServerManager initiales a new ServerManager intended to
|
|
// manage and persist servers.
|
|
func NewServerManager(db *sql.DB) ServerManager {
|
|
return &serverManager{db}
|
|
}
|
|
|
|
func (db *serverManager) Find(id int64) (*model.Server, error) {
|
|
dst := model.Server{}
|
|
err := meddler.Load(db, "servers", &dst, id)
|
|
return &dst, err
|
|
}
|
|
|
|
func (db *serverManager) FindName(name string) (*model.Server, error) {
|
|
dst := model.Server{}
|
|
err := meddler.QueryRow(db, &dst, findServerQuery, name)
|
|
return &dst, err
|
|
}
|
|
|
|
func (db *serverManager) FindSMTP() (*model.SMTPServer, error) {
|
|
dst := model.SMTPServer{}
|
|
err := meddler.Load(db, "smtp", &dst, 1)
|
|
if err != nil && err != sql.ErrNoRows {
|
|
return &dst, err
|
|
}
|
|
return &dst, nil
|
|
}
|
|
|
|
func (db *serverManager) List() ([]*model.Server, error) {
|
|
var dst []*model.Server
|
|
err := meddler.QueryAll(db, &dst, listServerQuery)
|
|
return dst, err
|
|
}
|
|
|
|
func (db *serverManager) Insert(server *model.Server) error {
|
|
return meddler.Insert(db, "servers", server)
|
|
}
|
|
|
|
func (db *serverManager) Update(server *model.Server) error {
|
|
return meddler.Update(db, "servers", server)
|
|
}
|
|
|
|
func (db *serverManager) UpdateSMTP(server *model.SMTPServer) error {
|
|
server.ID = 0
|
|
meddler.Insert(db, "smtp", server)
|
|
server.ID = 1
|
|
return meddler.Update(db, "smtp", server)
|
|
}
|
|
|
|
func (db *serverManager) Delete(server *model.Server) error {
|
|
_, err := db.Exec(deleteServerStmt, server.ID)
|
|
return err
|
|
}
|