3d617de577
将仓储层/基础设施改为 Ent + 原生 SQL 执行路径，并移除 AutoMigrate 与 GORM 依赖。 重构内容包括： - 仓储层改用 Ent/SQL（含 usage_log/account 等复杂查询），统一错误映射 - 基础设施与 setup 初始化切换为 Ent + SQL migrations - 集成测试与 fixtures 迁移到 Ent 事务模型 - 清理遗留 GORM 模型/依赖，补充迁移与文档说明 - 增加根目录 Makefile 便于前后端编译 测试： - go test -tags unit ./... - go test -tags integration ./...
106 lines
2.5 KiB
Go
106 lines
2.5 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/Wei-Shaw/sub2api/ent"
|
|
"github.com/Wei-Shaw/sub2api/ent/setting"
|
|
"github.com/Wei-Shaw/sub2api/internal/service"
|
|
)
|
|
|
|
type settingRepository struct {
|
|
client *ent.Client
|
|
}
|
|
|
|
func NewSettingRepository(client *ent.Client) service.SettingRepository {
|
|
return &settingRepository{client: client}
|
|
}
|
|
|
|
func (r *settingRepository) Get(ctx context.Context, key string) (*service.Setting, error) {
|
|
m, err := r.client.Setting.Query().Where(setting.KeyEQ(key)).Only(ctx)
|
|
if err != nil {
|
|
if ent.IsNotFound(err) {
|
|
return nil, service.ErrSettingNotFound
|
|
}
|
|
return nil, err
|
|
}
|
|
return &service.Setting{
|
|
ID: m.ID,
|
|
Key: m.Key,
|
|
Value: m.Value,
|
|
UpdatedAt: m.UpdatedAt,
|
|
}, nil
|
|
}
|
|
|
|
func (r *settingRepository) GetValue(ctx context.Context, key string) (string, error) {
|
|
setting, err := r.Get(ctx, key)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return setting.Value, nil
|
|
}
|
|
|
|
func (r *settingRepository) Set(ctx context.Context, key, value string) error {
|
|
now := time.Now()
|
|
return r.client.Setting.
|
|
Create().
|
|
SetKey(key).
|
|
SetValue(value).
|
|
SetUpdatedAt(now).
|
|
OnConflictColumns(setting.FieldKey).
|
|
UpdateNewValues().
|
|
Exec(ctx)
|
|
}
|
|
|
|
func (r *settingRepository) GetMultiple(ctx context.Context, keys []string) (map[string]string, error) {
|
|
if len(keys) == 0 {
|
|
return map[string]string{}, nil
|
|
}
|
|
settings, err := r.client.Setting.Query().Where(setting.KeyIn(keys...)).All(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
result := make(map[string]string)
|
|
for _, s := range settings {
|
|
result[s.Key] = s.Value
|
|
}
|
|
return result, nil
|
|
}
|
|
|
|
func (r *settingRepository) SetMultiple(ctx context.Context, settings map[string]string) error {
|
|
if len(settings) == 0 {
|
|
return nil
|
|
}
|
|
|
|
now := time.Now()
|
|
builders := make([]*ent.SettingCreate, 0, len(settings))
|
|
for key, value := range settings {
|
|
builders = append(builders, r.client.Setting.Create().SetKey(key).SetValue(value).SetUpdatedAt(now))
|
|
}
|
|
return r.client.Setting.
|
|
CreateBulk(builders...).
|
|
OnConflictColumns(setting.FieldKey).
|
|
UpdateNewValues().
|
|
Exec(ctx)
|
|
}
|
|
|
|
func (r *settingRepository) GetAll(ctx context.Context) (map[string]string, error) {
|
|
settings, err := r.client.Setting.Query().All(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
result := make(map[string]string)
|
|
for _, s := range settings {
|
|
result[s.Key] = s.Value
|
|
}
|
|
return result, nil
|
|
}
|
|
|
|
func (r *settingRepository) Delete(ctx context.Context, key string) error {
|
|
_, err := r.client.Setting.Delete().Where(setting.KeyEQ(key)).Exec(ctx)
|
|
return err
|
|
}
|