104 lines
2.3 KiB
Go
104 lines
2.3 KiB
Go
package system
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
|
|
"management/internal/erpserver/model/dto"
|
|
"management/internal/erpserver/model/system"
|
|
"management/internal/erpserver/repository"
|
|
"management/internal/pkg/database"
|
|
"management/internal/pkg/know/pearadmin"
|
|
|
|
"gorm.io/datatypes"
|
|
)
|
|
|
|
type configRepository struct {
|
|
repo *repository.Repository
|
|
}
|
|
|
|
func NewConfigRepository(repo *repository.Repository) system.ConfigRepository {
|
|
return &configRepository{
|
|
repo: repo,
|
|
}
|
|
}
|
|
|
|
func (r *configRepository) Initialize(ctx context.Context) error {
|
|
_, err := r.GetByKey(ctx, pearadmin.PearKey)
|
|
if err != nil {
|
|
if database.IsNoRows(err) {
|
|
b, e := json.Marshal(pearadmin.PearJson)
|
|
if e != nil {
|
|
return e
|
|
}
|
|
|
|
s := system.Config{
|
|
Key: pearadmin.PearKey,
|
|
Value: datatypes.JSON(b),
|
|
}
|
|
return r.Create(ctx, &s)
|
|
}
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *configRepository) Create(ctx context.Context, obj *system.Config) error {
|
|
return r.repo.DB(ctx).Create(obj).Error
|
|
}
|
|
|
|
func (r *configRepository) Update(ctx context.Context, obj *system.Config) error {
|
|
return r.repo.DB(ctx).Save(obj).Error
|
|
}
|
|
|
|
func (r *configRepository) Get(ctx context.Context, id int32) (*system.Config, error) {
|
|
var obj system.Config
|
|
err := r.repo.DB(ctx).First(&obj, id).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &obj, nil
|
|
}
|
|
|
|
func (r *configRepository) GetByKey(ctx context.Context, key string) (*system.Config, error) {
|
|
var obj system.Config
|
|
err := r.repo.DB(ctx).Where("key = ?", key).First(&obj).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &obj, nil
|
|
}
|
|
|
|
func (r *configRepository) GetValueByKey(ctx context.Context, key string) ([]byte, error) {
|
|
var obj system.Config
|
|
err := r.repo.DB(ctx).Where("key = ?", key).First(&obj).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return obj.Value, nil
|
|
}
|
|
|
|
func (r *configRepository) List(ctx context.Context, q dto.SearchDto) ([]*system.Config, int64, error) {
|
|
query := r.repo.DB(ctx).
|
|
Model(&system.Config{}).
|
|
Where("created_at BETWEEN ? AND ?", q.SearchTimeBegin, q.SearchTimeEnd)
|
|
|
|
var count int64
|
|
err := query.Count(&count).Error
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
var configs []*system.Config
|
|
err = query.
|
|
Order("id DESC").
|
|
Offset((q.Page - 1) * q.Rows).
|
|
Limit(q.Rows).
|
|
Find(&configs).
|
|
Error
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
return configs, count, nil
|
|
}
|