105 lines
2.7 KiB
Go
105 lines
2.7 KiB
Go
|
package dao
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
|
|||
|
|
"model-asynch/consts/public"
|
|||
|
|
"model-asynch/model/entity"
|
|||
|
|
|
|||
|
|
"gitea.com/red-future/common/db/gfdb"
|
|||
|
|
"github.com/gogf/gf/v2/util/gconv"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
var Model = &modelDao{}
|
|||
|
|
|
|||
|
|
type modelDao struct{}
|
|||
|
|
|
|||
|
|
func (d *modelDao) Insert(ctx context.Context, m *entity.AsynchModel) (id int64, err error) {
|
|||
|
|
r, err := gfdb.DB(ctx).Model(ctx, public.TableNameModel).Data(m).Insert()
|
|||
|
|
if err != nil {
|
|||
|
|
return 0, err
|
|||
|
|
}
|
|||
|
|
return r.LastInsertId()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (d *modelDao) UpdateByID(ctx context.Context, id int64, data map[string]any) (rows int64, err error) {
|
|||
|
|
// 触发 gfdb 的 updateHook 自动填充 updater，需要显式带 updater 字段
|
|||
|
|
data[entity.AsynchModelCol.Updater] = ""
|
|||
|
|
r, err := gfdb.DB(ctx).Model(ctx, public.TableNameModel).
|
|||
|
|
Where(entity.AsynchModelCol.Id, id).
|
|||
|
|
Data(data).
|
|||
|
|
Update()
|
|||
|
|
if err != nil {
|
|||
|
|
return 0, err
|
|||
|
|
}
|
|||
|
|
return r.RowsAffected()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (d *modelDao) DeleteByID(ctx context.Context, id int64) (rows int64, err error) {
|
|||
|
|
r, err := gfdb.DB(ctx).Model(ctx, public.TableNameModel).
|
|||
|
|
Where(entity.AsynchModelCol.Id, id).
|
|||
|
|
Delete()
|
|||
|
|
if err != nil {
|
|||
|
|
return 0, err
|
|||
|
|
}
|
|||
|
|
return r.RowsAffected()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (d *modelDao) GetByModelName(ctx context.Context, modelName string) (m *entity.AsynchModel, err error) {
|
|||
|
|
r, err := gfdb.DB(ctx).Model(ctx, public.TableNameModel).
|
|||
|
|
Where(entity.AsynchModelCol.ModelName, modelName).
|
|||
|
|
One()
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
if r.IsEmpty() {
|
|||
|
|
return nil, nil
|
|||
|
|
}
|
|||
|
|
err = r.Struct(&m)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (d *modelDao) GetByID(ctx context.Context, id int64) (m *entity.AsynchModel, err error) {
|
|||
|
|
r, err := gfdb.DB(ctx).Model(ctx, public.TableNameModel).
|
|||
|
|
Where(entity.AsynchModelCol.Id, id).
|
|||
|
|
One()
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
if r.IsEmpty() {
|
|||
|
|
return nil, nil
|
|||
|
|
}
|
|||
|
|
err = r.Struct(&m)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (d *modelDao) List(ctx context.Context, pageNum, pageSize int, modelNameLike string) (list []*entity.AsynchModel, total int64, err error) {
|
|||
|
|
model := gfdb.DB(ctx).Model(ctx, public.TableNameModel).Where("deleted_at IS NULL").OrderDesc(entity.AsynchModelCol.CreatedAt)
|
|||
|
|
if modelNameLike != "" {
|
|||
|
|
model = model.WhereLike(entity.AsynchModelCol.ModelName, "%"+modelNameLike+"%")
|
|||
|
|
}
|
|||
|
|
if pageNum > 0 && pageSize > 0 {
|
|||
|
|
model = model.Page(pageNum, pageSize)
|
|||
|
|
}
|
|||
|
|
r, totalInt, err := model.AllAndCount(false)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, 0, err
|
|||
|
|
}
|
|||
|
|
total = gconv.Int64(totalInt)
|
|||
|
|
err = r.Structs(&list)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ListAll 用于分组展示：查询全部模型（不按类型过滤，类型拆分在 service 层处理）
|
|||
|
|
func (d *modelDao) ListAll(ctx context.Context) (list []*entity.AsynchModel, err error) {
|
|||
|
|
r, err := gfdb.DB(ctx).Model(ctx, public.TableNameModel).
|
|||
|
|
Where("deleted_at IS NULL").
|
|||
|
|
OrderDesc(entity.AsynchModelCol.CreatedAt).
|
|||
|
|
All()
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
err = r.Structs(&list)
|
|||
|
|
return
|
|||
|
|
}
|