123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269 |
- package store
- import (
- "context"
- "strings"
- )
- // Role is the type of a role.
- type Role string
- const (
- // RoleHost is the HOST role.
- RoleHost Role = "HOST"
- // RoleAdmin is the ADMIN role.
- RoleAdmin Role = "ADMIN"
- // RoleUser is the USER role.
- RoleUser Role = "USER"
- )
- func (e Role) String() string {
- switch e {
- case RoleHost:
- return "HOST"
- case RoleAdmin:
- return "ADMIN"
- case RoleUser:
- return "USER"
- }
- return "USER"
- }
- type User struct {
- ID int32
- // Standard fields
- RowStatus RowStatus
- CreatedTs int64
- UpdatedTs int64
- // Domain specific fields
- Username string
- Role Role
- Email string
- Nickname string
- PasswordHash string
- AvatarURL string
- }
- type UpdateUser struct {
- ID int32
- UpdatedTs *int64
- RowStatus *RowStatus
- Username *string
- Role *Role
- Email *string
- Nickname *string
- Password *string
- AvatarURL *string
- PasswordHash *string
- }
- type FindUser struct {
- ID *int32
- RowStatus *RowStatus
- Username *string
- Role *Role
- Email *string
- Nickname *string
- }
- type DeleteUser struct {
- ID int32
- }
- func (s *Store) CreateUser(ctx context.Context, create *User) (*User, error) {
- stmt := `
- INSERT INTO user (
- username,
- role,
- email,
- nickname,
- password_hash
- )
- VALUES (?, ?, ?, ?, ?)
- RETURNING id, avatar_url, created_ts, updated_ts, row_status
- `
- if err := s.db.QueryRowContext(
- ctx,
- stmt,
- create.Username,
- create.Role,
- create.Email,
- create.Nickname,
- create.PasswordHash,
- ).Scan(
- &create.ID,
- &create.AvatarURL,
- &create.CreatedTs,
- &create.UpdatedTs,
- &create.RowStatus,
- ); err != nil {
- return nil, err
- }
- user := create
- s.userCache.Store(user.ID, user)
- return user, nil
- }
- func (s *Store) UpdateUser(ctx context.Context, update *UpdateUser) (*User, error) {
- set, args := []string{}, []any{}
- if v := update.UpdatedTs; v != nil {
- set, args = append(set, "updated_ts = ?"), append(args, *v)
- }
- if v := update.RowStatus; v != nil {
- set, args = append(set, "row_status = ?"), append(args, *v)
- }
- if v := update.Username; v != nil {
- set, args = append(set, "username = ?"), append(args, *v)
- }
- if v := update.Email; v != nil {
- set, args = append(set, "email = ?"), append(args, *v)
- }
- if v := update.Nickname; v != nil {
- set, args = append(set, "nickname = ?"), append(args, *v)
- }
- if v := update.AvatarURL; v != nil {
- set, args = append(set, "avatar_url = ?"), append(args, *v)
- }
- if v := update.PasswordHash; v != nil {
- set, args = append(set, "password_hash = ?"), append(args, *v)
- }
- args = append(args, update.ID)
- query := `
- UPDATE user
- SET ` + strings.Join(set, ", ") + `
- WHERE id = ?
- RETURNING id, username, role, email, nickname, password_hash, avatar_url, created_ts, updated_ts, row_status
- `
- user := &User{}
- if err := s.db.QueryRowContext(ctx, query, args...).Scan(
- &user.ID,
- &user.Username,
- &user.Role,
- &user.Email,
- &user.Nickname,
- &user.PasswordHash,
- &user.AvatarURL,
- &user.CreatedTs,
- &user.UpdatedTs,
- &user.RowStatus,
- ); err != nil {
- return nil, err
- }
- s.userCache.Store(user.ID, user)
- return user, nil
- }
- func (s *Store) ListUsers(ctx context.Context, find *FindUser) ([]*User, error) {
- where, args := []string{"1 = 1"}, []any{}
- if v := find.ID; v != nil {
- where, args = append(where, "id = ?"), append(args, *v)
- }
- if v := find.Username; v != nil {
- where, args = append(where, "username = ?"), append(args, *v)
- }
- if v := find.Role; v != nil {
- where, args = append(where, "role = ?"), append(args, *v)
- }
- if v := find.Email; v != nil {
- where, args = append(where, "email = ?"), append(args, *v)
- }
- if v := find.Nickname; v != nil {
- where, args = append(where, "nickname = ?"), append(args, *v)
- }
- query := `
- SELECT
- id,
- username,
- role,
- email,
- nickname,
- password_hash,
- avatar_url,
- created_ts,
- updated_ts,
- row_status
- FROM user
- WHERE ` + strings.Join(where, " AND ") + `
- ORDER BY created_ts DESC, row_status DESC
- `
- rows, err := s.db.QueryContext(ctx, query, args...)
- if err != nil {
- return nil, err
- }
- defer rows.Close()
- list := make([]*User, 0)
- for rows.Next() {
- var user User
- if err := rows.Scan(
- &user.ID,
- &user.Username,
- &user.Role,
- &user.Email,
- &user.Nickname,
- &user.PasswordHash,
- &user.AvatarURL,
- &user.CreatedTs,
- &user.UpdatedTs,
- &user.RowStatus,
- ); err != nil {
- return nil, err
- }
- list = append(list, &user)
- }
- if err := rows.Err(); err != nil {
- return nil, err
- }
- for _, user := range list {
- s.userCache.Store(user.ID, user)
- }
- return list, nil
- }
- func (s *Store) GetUser(ctx context.Context, find *FindUser) (*User, error) {
- if find.ID != nil {
- if cache, ok := s.userCache.Load(*find.ID); ok {
- return cache.(*User), nil
- }
- }
- list, err := s.ListUsers(ctx, find)
- if err != nil {
- return nil, err
- }
- if len(list) == 0 {
- return nil, nil
- }
- user := list[0]
- s.userCache.Store(user.ID, user)
- return user, nil
- }
- func (s *Store) DeleteUser(ctx context.Context, delete *DeleteUser) error {
- result, err := s.db.ExecContext(ctx, `
- DELETE FROM user WHERE id = ?
- `, delete.ID)
- if err != nil {
- return err
- }
- if _, err := result.RowsAffected(); err != nil {
- return err
- }
- if err := s.Vacuum(ctx); err != nil {
- // Prevent linter warning.
- return err
- }
- s.userCache.Delete(delete.ID)
- return nil
- }
|