87 lines
1.5 KiB
Go
87 lines
1.5 KiB
Go
package store
|
|
|
|
import (
|
|
"database/sql"
|
|
|
|
sq "github.com/Masterminds/squirrel"
|
|
|
|
"git.ctrlz.es/mgdelacroix/rmsn/model"
|
|
)
|
|
|
|
var userColumns = []string{"id", "name", "mail", "username", "password"}
|
|
|
|
type UserStore struct {
|
|
Conn *sql.DB
|
|
}
|
|
|
|
func NewUserStore(s *Store) *UserStore {
|
|
return &UserStore{Conn: s.Conn}
|
|
}
|
|
|
|
func (us *UserStore) Q() sq.StatementBuilderType {
|
|
return sq.StatementBuilder.RunWith(us.Conn)
|
|
}
|
|
|
|
func (us *UserStore) usersFromRows(rows *sql.Rows) ([]*model.User, error) {
|
|
users := []*model.User{}
|
|
|
|
for rows.Next() {
|
|
var user model.User
|
|
|
|
err := rows.Scan(
|
|
&user.ID,
|
|
&user.Name,
|
|
&user.Mail,
|
|
&user.Username,
|
|
&user.Password,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
users = append(users, &user)
|
|
}
|
|
|
|
return users, nil
|
|
}
|
|
|
|
func (us *UserStore) GetByID(id int) (*model.User, error) {
|
|
query := us.Q().Select(userColumns...).
|
|
From("users").
|
|
Where(sq.Eq{"id": id})
|
|
|
|
rows, err := query.Query()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
users, err := us.usersFromRows(rows)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if len(users) == 0 {
|
|
return nil, sql.ErrNoRows
|
|
}
|
|
|
|
return users[0], nil
|
|
}
|
|
|
|
func (us *UserStore) Create(user *model.User) (*model.User, error) {
|
|
query := us.Q().Insert("users").
|
|
Columns(userColumns[1:]...).
|
|
Values(user.Name, user.Mail, user.Username, user.Password)
|
|
|
|
res, err := query.Exec()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
id, err := res.LastInsertId()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return us.GetByID(int(id))
|
|
}
|