refact:openteam

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
+}