refact:openteam

2025-01-01 23:10:01 +08:00
parent 1f5e1c221c
commit 65d6d12972
18 changed files with 1126 additions and 106 deletions
--- a/team/dao/apikey.go
+++ b/team/dao/apikey.go
@@ -0,0 +1,145 @@
+package dao
+
+import (
+	"errors"
+	"opencatd-open/team/model"
+	"time"
+
+	"gorm.io/gorm"
+)
+
+var _ ApiKeyRepository = (*ApiKeyDAO)(nil)
+
+type ApiKeyRepository interface {
+	Create(apiKey *model.ApiKey) error
+	GetByID(id int64) (*model.ApiKey, error)
+	GetByName(name string) (*model.ApiKey, error)
+	GetByApiKey(apiKeyValue string) (*model.ApiKey, error)
+	Update(apiKey *model.ApiKey) error
+	Delete(id int64) error
+	List(offset, limit int, status *int) ([]model.ApiKey, error)
+	Enable(id int64) error
+	Disable(id int64) error
+	BatchEnable(ids []int64) error
+	BatchDisable(ids []int64) error
+	BatchDelete(ids []int64) error
+	Count() (int64, error)
+}
+
+type ApiKeyDAO struct {
+	db *gorm.DB
+}
+
+func NewApiKeyDAO(db *gorm.DB) *ApiKeyDAO {
+	return &ApiKeyDAO{db: db}
+}
+
+// CreateApiKey 创建ApiKey
+func (dao *ApiKeyDAO) Create(apiKey *model.ApiKey) error {
+	if apiKey == nil {
+		return errors.New("apiKey is nil")
+	}
+	return dao.db.Create(apiKey).Error
+}
+
+// GetApiKeyByID 根据ID获取ApiKey
+func (dao *ApiKeyDAO) GetByID(id int64) (*model.ApiKey, error) {
+	var apiKey model.ApiKey
+	err := dao.db.First(&apiKey, id).Error
+	if err != nil {
+		return nil, err
+	}
+	return &apiKey, nil
+}
+
+// GetApiKeyByName 根据名称获取ApiKey
+func (dao *ApiKeyDAO) GetByName(name string) (*model.ApiKey, error) {
+	var apiKey model.ApiKey
+	err := dao.db.Where("name = ?", name).First(&apiKey).Error
+	if err != nil {
+		return nil, err
+	}
+	return &apiKey, nil
+}
+
+// GetApiKeyByApiKey 根据ApiKey值获取ApiKey
+func (dao *ApiKeyDAO) GetByApiKey(apiKeyValue string) (*model.ApiKey, error) {
+	var apiKey model.ApiKey
+	err := dao.db.Where("api_key = ?", apiKeyValue).First(&apiKey).Error
+	if err != nil {
+		return nil, err
+	}
+	return &apiKey, nil
+}
+
+// UpdateApiKey 更新ApiKey信息
+func (dao *ApiKeyDAO) Update(apiKey *model.ApiKey) error {
+	if apiKey == nil {
+		return errors.New("apiKey is nil")
+	}
+	apiKey.UpdatedAt = time.Now()
+	return dao.db.Save(apiKey).Error
+}
+
+// DeleteApiKey 删除ApiKey
+func (dao *ApiKeyDAO) Delete(id int64) error {
+	return dao.db.Delete(&model.ApiKey{}, id).Error
+}
+
+// ListApiKeys 获取ApiKey列表
+func (dao *ApiKeyDAO) List(offset, limit int, status *int) ([]model.ApiKey, error) {
+	var apiKeys []model.ApiKey
+	db := dao.db.Offset(offset).Limit(limit)
+	if status != nil {
+		db = db.Where("status = ?", *status)
+	}
+	err := db.Find(&apiKeys).Error
+	if err != nil {
+		return nil, err
+	}
+	return apiKeys, nil
+}
+
+// EnableApiKey 启用ApiKey
+func (dao *ApiKeyDAO) Enable(id int64) error {
+	return dao.db.Model(&model.ApiKey{}).Where("id = ?", id).Update("status", 0).Error
+}
+
+// DisableApiKey 禁用ApiKey
+func (dao *ApiKeyDAO) Disable(id int64) error {
+	return dao.db.Model(&model.ApiKey{}).Where("id = ?", id).Update("status", 1).Error
+}
+
+// BatchEnableApiKeys 批量启用ApiKey
+func (dao *ApiKeyDAO) BatchEnable(ids []int64) error {
+	if len(ids) == 0 {
+		return errors.New("ids is empty")
+	}
+	return dao.db.Model(&model.ApiKey{}).Where("id IN ?", ids).Update("status", 0).Error
+}
+
+// BatchDisableApiKeys 批量禁用ApiKey
+func (dao *ApiKeyDAO) BatchDisable(ids []int64) error {
+	if len(ids) == 0 {
+		return errors.New("ids is empty")
+	}
+	return dao.db.Model(&model.ApiKey{}).Where("id IN ?", ids).Update("status", 1).Error
+}
+
+// BatchDeleteApiKey 批量删除ApiKey
+func (dao *ApiKeyDAO) BatchDelete(ids []int64) error {
+	if len(ids) == 0 {
+		return errors.New("ids is empty")
+	}
+	return dao.db.Delete(&model.ApiKey{}, ids).Error
+}
+
+// CountApiKeys 获取ApiKey总数
+func (dao *ApiKeyDAO) Count() (int64, error) {
+	var count int64
+	err := dao.db.Model(&model.ApiKey{}).Count(&count).Error
+	if err != nil {
+		return 0, err
+	}
+	return count, nil
+}
--- a/team/dao/token.go
+++ b/team/dao/token.go
@@ -0,0 +1,120 @@
+package dao
+
+import (
+	"errors"
+	"opencatd-open/team/model"
+
+	"gorm.io/gorm"
+)
+
+// 确保 TokenDAO 实现了 TokenDAOInterface 接口
+var _ TokenDAOInterface = (*TokenDAO)(nil)
+
+type TokenDAOInterface interface {
+	Create(token *model.Token) error
+	GetByID(id int) (*model.Token, error)
+	GetByKey(key string) (*model.Token, error)
+	GetByUserID(userID int) ([]model.Token, error)
+	Update(token *model.Token) error
+	Delete(id int) error
+	List(offset, limit int) ([]model.Token, error)
+	Disable(id int) error
+	Enable(id int) error
+	BatchDisable(ids []int) error
+	BatchEnable(ids []int) error
+	BatchDelete(ids []int) error
+}
+
+type TokenDAO struct {
+	db *gorm.DB
+}
+
+func NewTokenDAO(db *gorm.DB) *TokenDAO {
+	return &TokenDAO{db: db}
+}
+
+// CreateToken 创建 Token
+func (dao *TokenDAO) Create(token *model.Token) error {
+	if token == nil {
+		return errors.New("token is nil")
+	}
+	return dao.db.Create(token).Error
+}
+
+// GetTokenByID 根据 ID 获取 Token
+func (dao *TokenDAO) GetByID(id int) (*model.Token, error) {
+	var token model.Token
+	err := dao.db.First(&token, id).Error
+	if err != nil {
+		return nil, err
+	}
+	return &token, nil
+}
+
+// GetTokenByKey 根据 Key 获取 Token
+func (dao *TokenDAO) GetByKey(key string) (*model.Token, error) {
+	var token model.Token
+	err := dao.db.Where("key = ?", key).First(&token).Error
+	if err != nil {
+		return nil, err
+	}
+	return &token, nil
+}
+
+// GetTokensByUserID 根据 UserID 获取 Token 列表
+func (dao *TokenDAO) GetByUserID(userID int) ([]model.Token, error) {
+	var tokens []model.Token
+	err := dao.db.Where("user_id = ?", userID).Find(&tokens).Error
+	if err != nil {
+		return nil, err
+	}
+	return tokens, nil
+}
+
+// UpdateToken 更新 Token 信息
+func (dao *TokenDAO) Update(token *model.Token) error {
+	if token == nil {
+		return errors.New("token is nil")
+	}
+	return dao.db.Save(token).Error
+}
+
+// DeleteToken 删除 Token
+func (dao *TokenDAO) Delete(id int) error {
+	return dao.db.Delete(&model.Token{}, id).Error
+}
+
+// ListTokens 获取 Token 列表
+func (dao *TokenDAO) List(offset, limit int) ([]model.Token, error) {
+	var tokens []model.Token
+	err := dao.db.Offset(offset).Limit(limit).Find(&tokens).Error
+	if err != nil {
+		return nil, err
+	}
+	return tokens, nil
+}
+
+// DisableToken 禁用 Token
+func (dao *TokenDAO) Disable(id int) error {
+	return dao.db.Model(&model.Token{}).Where("id = ?", id).Update("status", false).Error
+}
+
+// EnableToken 启用 Token
+func (dao *TokenDAO) Enable(id int) error {
+	return dao.db.Model(&model.Token{}).Where("id = ?", id).Update("status", true).Error
+}
+
+// BatchDisableTokens 批量禁用 Token
+func (dao *TokenDAO) BatchDisable(ids []int) error {
+	return dao.db.Model(&model.Token{}).Where("id IN ?", ids).Update("status", false).Error
+}
+
+// BatchEnableTokens 批量启用 Token
+func (dao *TokenDAO) BatchEnable(ids []int) error {
+	return dao.db.Model(&model.Token{}).Where("id IN ?", ids).Update("status", true).Error
+}
+
+// BatchDeleteTokens 批量删除 Token
+func (dao *TokenDAO) BatchDelete(ids []int) error {
+	return dao.db.Where("id IN ?", ids).Delete(&model.Token{}).Error
+}
--- a/team/dao/user.go
+++ b/team/dao/user.go
@@ -0,0 +1,123 @@
+package dao
+
+import (
+	"errors"
+	"opencatd-open/team/model"
+	"time"
+
+	"gorm.io/gorm"
+)
+
+// 确保 UserDAO 实现了 UserRepository 接口
+var _ UserRepository = (*UserDAO)(nil)
+
+// UserRepository 定义用户数据访问操作的接口
+type UserRepository interface {
+	Create(user *model.User) error
+	GetByID(id int64) (*model.User, error)
+	GetByUsername(username string) (*model.User, error)
+	Update(user *model.User) error
+	Delete(id int64) error
+	List(offset, limit int) ([]model.User, error)
+	Enable(id int64) error
+	Disable(id int64) error
+	BatchEnable(ids []int64) error
+	BatchDisable(ids []int64) error
+	BatchDelete(ids []int64) error
+}
+
+type UserDAO struct {
+	db *gorm.DB
+}
+
+func NewUserDAO(db *gorm.DB) *UserDAO {
+	return &UserDAO{db: db}
+}
+
+// CreateUser 创建用户
+func (dao *UserDAO) Create(user *model.User) error {
+	if user == nil {
+		return errors.New("user is nil")
+	}
+	return dao.db.Create(user).Error
+}
+
+// GetUserByID 根据ID获取用户
+func (dao *UserDAO) GetByID(id int64) (*model.User, error) {
+	var user model.User
+	err := dao.db.First(&user, id).Error
+	if err != nil {
+		return nil, err
+	}
+	return &user, nil
+}
+
+// GetUserByUsername 根据用户名获取用户
+func (dao *UserDAO) GetByUsername(username string) (*model.User, error) {
+	var user model.User
+	err := dao.db.Where("user_name = ?", username).First(&user).Error
+	if err != nil {
+		return nil, err
+	}
+	return &user, nil
+}
+
+// UpdateUser 更新用户信息
+func (dao *UserDAO) Update(user *model.User) error {
+	if user == nil {
+		return errors.New("user is nil")
+	}
+	user.UpdatedAt = time.Now()
+	return dao.db.Save(user).Error
+}
+
+// DeleteUser 删除用户
+func (dao *UserDAO) Delete(id int64) error {
+	return dao.db.Delete(&model.User{}, id).Error
+	// return dao.db.Model(&model.User{}).Where("id = ?", id).Update("status", 2).Error
+}
+
+// ListUsers 获取用户列表
+func (dao *UserDAO) List(offset, limit int) ([]model.User, error) {
+	var users []model.User
+	err := dao.db.Offset(offset).Limit(limit).Find(&users).Error
+	if err != nil {
+		return nil, err
+	}
+	return users, nil
+}
+
+// EnableUser 启用User
+func (dao *UserDAO) Enable(id int64) error {
+	return dao.db.Model(&model.User{}).Where("id = ?", id).Update("status", 0).Error
+}
+
+// DisableUser 禁用User
+func (dao *UserDAO) Disable(id int64) error {
+	return dao.db.Model(&model.User{}).Where("id = ?", id).Update("status", 1).Error
+}
+
+// BatchEnableUsers 批量启用User
+func (dao *UserDAO) BatchEnable(ids []int64) error {
+	if len(ids) == 0 {
+		return errors.New("ids is empty")
+	}
+	return dao.db.Model(&model.User{}).Where("id IN ?", ids).Update("status", 0).Error
+}
+
+// BatchDisableUsers 批量禁用User
+func (dao *UserDAO) BatchDisable(ids []int64) error {
+	if len(ids) == 0 {
+		return errors.New("ids is empty")
+	}
+	return dao.db.Model(&model.User{}).Where("id IN ?", ids).Update("status", 1).Error
+}
+
+// BatchDeleteUser 批量删除用户
+func (dao *UserDAO) BatchDelete(ids []int64) error {
+	if len(ids) == 0 {
+		return errors.New("ids is empty")
+	}
+	return dao.db.Where("id IN ?", ids).Delete(&model.User{}).Error
+	// return dao.db.Model(&model.User{}).Where("id IN ?", ids).Update("status", 2).Error
+}