146 lines
3.6 KiB
Go
146 lines
3.6 KiB
Go
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
|
|
}
|