harness-drone/server/resource/user/manager.go
2014-06-04 14:25:38 -07:00

111 lines
2.3 KiB
Go

package user
import (
"database/sql"
"time"
"github.com/russross/meddler"
)
type UserManager interface {
// Find finds the User by ID.
Find(id int64) (*User, error)
// FindLogin finds the User by remote login.
FindLogin(remote, login string) (*User, error)
// FindToken finds the User by token.
FindToken(token string) (*User, error)
// List finds all registered users of the system.
List() ([]*User, error)
// Insert persists the User to the datastore.
Insert(user *User) error
// Update persists changes to the User to the datastore.
Update(user *User) error
// Delete removes the User from the datastore.
Delete(user *User) error
}
// userManager manages a list of users in a SQL database.
type userManager struct {
*sql.DB
}
// SQL query to retrieve a User by remote login.
const findLoginQuery = `
SELECT *
FROM users
WHERE user_remote=?
AND user_login=?
LIMIT 1
`
// SQL query to retrieve a User by remote login.
const findTokenQuery = `
SELECT *
FROM users
WHERE user_token=?
LIMIT 1
`
// SQL query to retrieve a list of all users.
const listQuery = `
SELECT *
FROM users
ORDER BY user_name ASC
`
// SQL statement to delete a User by ID.
const deleteStmt = `
DELETE FROM users WHERE user_id=?
`
// NewManager initiales a new UserManager intended to
// manage and persist commits.
func NewManager(db *sql.DB) UserManager {
return &userManager{db}
}
func (db *userManager) Find(id int64) (*User, error) {
dst := User{}
err := meddler.Load(db, "users", &dst, id)
return &dst, err
}
func (db *userManager) FindLogin(remote, login string) (*User, error) {
dst := User{}
err := meddler.QueryRow(db, &dst, findLoginQuery, remote, login)
return &dst, err
}
func (db *userManager) FindToken(token string) (*User, error) {
dst := User{}
err := meddler.QueryRow(db, &dst, findTokenQuery, token)
return &dst, err
}
func (db *userManager) List() ([]*User, error) {
var dst []*User
err := meddler.QueryAll(db, &dst, listQuery)
return dst, err
}
func (db *userManager) Insert(user *User) error {
user.Created = time.Now().Unix()
user.Updated = time.Now().Unix()
return meddler.Insert(db, "users", user)
}
func (db *userManager) Update(user *User) error {
user.Updated = time.Now().Unix()
return meddler.Update(db, "users", user)
}
func (db *userManager) Delete(user *User) error {
_, err := db.Exec(deleteStmt, user.ID)
return err
}