prompts-core

dao/prompt_dao.go

@@ -0,0 +1,97 @@
+package dao
+
+import (
+	"context"
+
+	"prompts-core/consts/public"
+	"prompts-core/model/entity"
+
+	"gitea.com/red-future/common/db/gfdb"
+	"github.com/gogf/gf/v2/util/gconv"
+)
+
+var Prompt = &promptDao{}
+
+type promptDao struct{}
+
+func (d *promptDao) Insert(ctx context.Context, m *entity.PromptConfig) (id int64, err error) {
+	r, err := gfdb.DB(ctx).Model(ctx, public.TableNamePromptConfig).Data(m).Insert()
+	if err != nil {
+		return 0, err
+	}
+	return r.LastInsertId()
+}
+
+func (d *promptDao) UpdateByID(ctx context.Context, id int64, data map[string]any) (rows int64, err error) {
+	// 触发 gfdb 的 updateHook 自动填充 updater，需要显式带 updater 字段
+	data[entity.PromptConfigCol.Updater] = ""
+	r, err := gfdb.DB(ctx).Model(ctx, public.TableNamePromptConfig).
+		Where(entity.PromptConfigCol.Id, id).
+		Data(data).
+		Update()
+	if err != nil {
+		return 0, err
+	}
+	return r.RowsAffected()
+}
+
+func (d *promptDao) DeleteByID(ctx context.Context, id int64) (rows int64, err error) {
+	r, err := gfdb.DB(ctx).Model(ctx, public.TableNamePromptConfig).
+		Where(entity.PromptConfigCol.Id, id).
+		Delete()
+	if err != nil {
+		return 0, err
+	}
+	return r.RowsAffected()
+}
+
+func (d *promptDao) GetByID(ctx context.Context, id int64) (m *entity.PromptConfig, err error) {
+	r, err := gfdb.DB(ctx).Model(ctx, public.TableNamePromptConfig).
+		Where(entity.PromptConfigCol.Id, id).
+		One()
+	if err != nil {
+		return nil, err
+	}
+	if r.IsEmpty() {
+		return nil, nil
+	}
+	err = r.Struct(&m)
+	return
+}
+
+func (d *promptDao) GetLatestEnabledByModelTypeID(ctx context.Context, modelTypeID int) (m *entity.PromptConfig, err error) {
+	r, err := gfdb.DB(ctx).Model(ctx, public.TableNamePromptConfig).
+		Where("deleted_at IS NULL").
+		Where(entity.PromptConfigCol.ModelTypeId, modelTypeID).
+		Where(entity.PromptConfigCol.Enabled, 1).
+		OrderDesc(entity.PromptConfigCol.CreatedAt).
+		One()
+	if err != nil {
+		return nil, err
+	}
+	if r.IsEmpty() {
+		return nil, nil
+	}
+	err = r.Struct(&m)
+	return
+}
+
+func (d *promptDao) List(ctx context.Context, pageNum, pageSize int, modelTypeID *int, modelTypeLike string) (list []*entity.PromptConfig, total int64, err error) {
+	model := gfdb.DB(ctx).Model(ctx, public.TableNamePromptConfig).Where("deleted_at IS NULL").OrderDesc(entity.PromptConfigCol.CreatedAt)
+	if modelTypeID != nil && *modelTypeID > 0 {
+		model = model.Where(entity.PromptConfigCol.ModelTypeId, *modelTypeID)
+	}
+	if modelTypeLike != "" {
+		model = model.WhereLike(entity.PromptConfigCol.ModelType, "%"+modelTypeLike+"%")
+	}
+	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
+}