prompts-core

2026-05-12 13:59:15 +08:00
parent 9080d2cebe
commit b179fab2a1
34 changed files with 3892 additions and 0 deletions
--- a/dao/compose_session_dao.go
+++ b/dao/compose_session_dao.go
@@ -0,0 +1,100 @@
+package dao
+
+import (
+	"context"
+
+	"prompts-core/consts/public"
+	"prompts-core/model/entity"
+
+	"gitea.com/red-future/common/db/gfdb"
+)
+
+var ComposeSession = &composeSessionDao{}
+
+type composeSessionDao struct{}
+
+func (d *composeSessionDao) Insert(ctx context.Context, m *entity.ComposeSession) (id int64, err error) {
+	r, err := gfdb.DB(ctx).Model(ctx, public.TableNameComposeSession).Data(m).Insert()
+	if err != nil {
+		return 0, err
+	}
+	return r.LastInsertId()
+}
+
+func (d *composeSessionDao) GetById(ctx context.Context, Id int64) (m *entity.ComposeSession, err error) {
+	r, err := gfdb.DB(ctx).Model(ctx, public.TableNameComposeSession).
+		Where("deleted_at IS NULL").
+		Where(entity.ComposeSessionCol.Id, Id).
+		One()
+	if err != nil {
+		return nil, err
+	}
+	if r.IsEmpty() {
+		return nil, nil
+	}
+	err = r.Struct(&m)
+	return
+}
+
+func (d *composeSessionDao) GetBySessionId(ctx context.Context, sessionId string) (m *entity.ComposeSession, err error) {
+	r, err := gfdb.DB(ctx).Model(ctx, public.TableNameComposeSession).
+		Where("deleted_at IS NULL").
+		Where(entity.ComposeSessionCol.SessionId, sessionId).
+		One()
+	if err != nil {
+		return nil, err
+	}
+	if r.IsEmpty() {
+		return nil, nil
+	}
+	err = r.Struct(&m)
+	return
+}
+
+func (d *composeSessionDao) UpdateById(ctx context.Context, id int64, data map[string]any) (rows int64, err error) {
+	data[entity.ComposeSessionCol.Updater] = ""
+	r, err := gfdb.DB(ctx).Model(ctx, public.TableNameComposeSession).
+		Where(entity.ComposeSessionCol.Id, id).
+		Data(data).
+		Update()
+	if err != nil {
+		return 0, err
+	}
+	return r.RowsAffected()
+}
+
+func (d *composeSessionDao) List(ctx context.Context, page, size int, where map[string]any) (list []*entity.ComposeSession, total int, err error) {
+	model := gfdb.DB(ctx).Model(ctx, public.TableNameComposeSession).
+		Where("deleted_at IS NULL")
+
+	// 动态拼接查询条件
+	for k, v := range where {
+		model = model.Where(k, v)
+	}
+
+	// 查询总数
+	total, err = model.Count()
+	if err != nil {
+		return nil, 0, err
+	}
+
+	// 分页查询
+	err = model.Order("created_at DESC").
+		Page(page, size).
+		Scan(&list)
+
+	return
+}
+
+func (d *composeSessionDao) DeleteBySessionId(ctx context.Context, sessionId string) (rows int64, err error) {
+	r, err := gfdb.DB(ctx).Model(ctx, public.TableNameComposeSession).
+		Where(entity.ComposeSessionCol.SessionId, sessionId).
+		Data(map[string]any{
+			entity.ComposeSessionCol.DeletedAt: "NOW()",
+		}).
+		Update()
+	if err != nil {
+		return 0, err
+	}
+	return r.RowsAffected()
+}
--- a/dao/compose_task_dao.go
+++ b/dao/compose_task_dao.go
@@ -0,0 +1,49 @@
+package dao
+
+import (
+	"context"
+
+	"prompts-core/consts/public"
+	"prompts-core/model/entity"
+
+	"gitea.com/red-future/common/db/gfdb"
+)
+
+var ComposeTask = &composeTaskDao{}
+
+type composeTaskDao struct{}
+
+func (d *composeTaskDao) Insert(ctx context.Context, m *entity.ComposeTask) (id int64, err error) {
+	r, err := gfdb.DB(ctx).Model(ctx, public.TableNameComposeTask).Data(m).Insert()
+	if err != nil {
+		return 0, err
+	}
+	return r.LastInsertId()
+}
+
+func (d *composeTaskDao) GetByTaskId(ctx context.Context, taskId string) (m *entity.ComposeTask, err error) {
+	r, err := gfdb.DB(ctx).Model(ctx, public.TableNameComposeTask).
+		Where("deleted_at IS NULL").
+		Where(entity.ComposeTaskCol.TaskId, taskId).
+		One()
+	if err != nil {
+		return nil, err
+	}
+	if r.IsEmpty() {
+		return nil, nil
+	}
+	err = r.Struct(&m)
+	return
+}
+
+func (d *composeTaskDao) UpdateByTaskId(ctx context.Context, taskId string, data map[string]any) (rows int64, err error) {
+	data[entity.ComposeTaskCol.Updater] = ""
+	r, err := gfdb.DB(ctx).Model(ctx, public.TableNameComposeTask).
+		Where(entity.ComposeTaskCol.TaskId, taskId).
+		Data(data).
+		Update()
+	if err != nil {
+		return 0, err
+	}
+	return r.RowsAffected()
+}
--- a/dao/model_dao.go
+++ b/dao/model_dao.go
@@ -0,0 +1,57 @@
+package dao
+
+import (
+	"context"
+	"fmt"
+	"prompts-core/consts/public"
+	"prompts-core/model/entity"
+
+	"gitea.com/red-future/common/db/gfdb"
+)
+
+var Model = &modelDao{}
+
+type modelDao struct{}
+
+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) GetByIsChatModel(ctx context.Context) (m *entity.AsynchModel, err error) {
+	r, err := gfdb.DB(ctx).Model(ctx, public.TableNameModel).
+		Where(entity.AsynchModelCol.IsChatModel, 1).
+		One()
+	if err != nil {
+		return nil, err
+	}
+	if r.IsEmpty() {
+		return nil, nil
+	}
+	err = r.Struct(&m)
+	return
+}
+
+// GetBySuperAdmin 查询超级管理员（tenant_id=1）的模型
+func (d *modelDao) GetBySuperAdmin(ctx context.Context, modelName string) (m *entity.AsynchModel, err error) {
+	sql := fmt.Sprintf("SELECT * FROM %s WHERE model_name = ? AND tenant_id = 1 AND deleted_at IS NULL LIMIT 1", public.TableNameModel)
+	r, err := gfdb.DB(ctx).GetAll(ctx, sql, modelName)
+	if err != nil {
+		return nil, err
+	}
+	if len(r) == 0 {
+		return nil, nil
+	}
+
+	err = r[0].Struct(&m)
+	return
+}
--- a/dao/prompt_dao.go
+++ b/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
+}