0.29.1原版

2026-07-02 19:14:14 +08:00
commit d94008f0fb
947 changed files with 174905 additions and 0 deletions
@@ -0,0 +1,179 @@
+package store
+
+import (
+	"context"
+	"strconv"
+
+	"github.com/pkg/errors"
+)
+
+// Role is the type of a role.
+type Role string
+
+const (
+	// RoleAdmin is the ADMIN role.
+	RoleAdmin Role = "ADMIN"
+	// RoleUser is the USER role.
+	RoleUser Role = "USER"
+)
+
+func (e Role) String() string {
+	switch e {
+	case RoleAdmin:
+		return "ADMIN"
+	default:
+		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
+	Description  string
+}
+
+type UpdateUser struct {
+	ID int32
+
+	UpdatedTs    *int64
+	RowStatus    *RowStatus
+	Username     *string
+	Role         *Role
+	Email        *string
+	Nickname     *string
+	Password     *string
+	AvatarURL    *string
+	PasswordHash *string
+	Description  *string
+}
+
+type FindUser struct {
+	ID     *int32
+	IDList []int32
+
+	UsernameList []string
+
+	RowStatus *RowStatus
+	Username  *string
+	Role      *Role
+	Email     *string
+	Nickname  *string
+	Search    *string
+
+	// Domain specific fields
+	Filters []string
+
+	// The maximum number of users to return.
+	Limit *int
+}
+
+type DeleteUser struct {
+	ID int32
+}
+
+func userCacheKey(userID int32) string {
+	return strconv.Itoa(int(userID))
+}
+
+func (s *Store) CreateUser(ctx context.Context, create *User) (*User, error) {
+	user, err := s.driver.CreateUser(ctx, create)
+	if err != nil {
+		return nil, err
+	}
+
+	s.userCache.Set(ctx, userCacheKey(user.ID), user)
+	return user, nil
+}
+
+// CreateUserIfNoUsers creates a user only when the instance has no users.
+// The in-process lock prevents concurrent first-user setup requests from
+// creating multiple admins in the same server process.
+func (s *Store) CreateUserIfNoUsers(ctx context.Context, create *User) (*User, bool, error) {
+	s.userCreateMu.Lock()
+	defer s.userCreateMu.Unlock()
+
+	limitOne := 1
+	users, err := s.driver.ListUsers(ctx, &FindUser{Limit: &limitOne})
+	if err != nil {
+		return nil, false, err
+	}
+	if len(users) > 0 {
+		return nil, false, nil
+	}
+
+	user, err := s.driver.CreateUser(ctx, create)
+	if err != nil {
+		return nil, false, err
+	}
+	s.userCache.Set(ctx, userCacheKey(user.ID), user)
+	return user, true, nil
+}
+
+func (s *Store) UpdateUser(ctx context.Context, update *UpdateUser) (*User, error) {
+	user, err := s.driver.UpdateUser(ctx, update)
+	if err != nil {
+		return nil, err
+	}
+
+	s.userCache.Set(ctx, userCacheKey(user.ID), user)
+	return user, nil
+}
+
+func (s *Store) ListUsers(ctx context.Context, find *FindUser) ([]*User, error) {
+	list, err := s.driver.ListUsers(ctx, find)
+	if err != nil {
+		return nil, err
+	}
+
+	for _, user := range list {
+		s.userCache.Set(ctx, userCacheKey(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.Get(ctx, userCacheKey(*find.ID)); ok {
+			user, ok := cache.(*User)
+			if ok {
+				return 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.Set(ctx, userCacheKey(user.ID), user)
+	return user, nil
+}
+
+func (s *Store) DeleteUser(ctx context.Context, delete *DeleteUser) (*DeleteUserResult, error) {
+	result, err := s.driver.DeleteUser(ctx, delete)
+	if err != nil {
+		return nil, err
+	}
+	if result == nil {
+		return nil, errors.New("unexpected nil delete user result")
+	}
+	s.deleteUserCache(ctx, delete.ID, result)
+	return result, nil
+}