refactor(service): 重构服务代码结构并更新配置

dao/model_dao.go

@@ -2,62 +2,27 @@ package dao
 
 import (
 	"context"
-	"fmt"
 	"prompts-core/consts/public"
 	"prompts-core/model/entity"
 
 	"gitea.com/red-future/common/db/gfdb"
-	"gitea.com/red-future/common/utils"
 )
 
 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()
+// Get 获取模型
+func (d *modelDao) Get(ctx context.Context, req *entity.AsynchModel, fields ...string) (m *entity.AsynchModel, err error) {
+	r, err := gfdb.DB(ctx, public.DbNameModelGateway).Model(ctx, public.TableNameModel).
+		OmitEmpty().
+		Where(entity.AsynchModelCol.Creator, req.Creator).
+		Where(entity.AsynchModelCol.IsChatModel, req.IsChatModel).
+		Where(entity.AsynchModelCol.ModelName, req.ModelName).
+		Fields(fields).One()
 	if err != nil {
-		return nil, err
-	}
-	if r.IsEmpty() {
-		return nil, nil
+		return
 	}
 	err = r.Struct(&m)
 	return
 }
-
-func (d *modelDao) GetByIsChatModel(ctx context.Context) (m *entity.AsynchModel, err error) {
-	userInfo, err := utils.GetUserInfo(ctx)
-	if err != nil {
-		return nil, err
-	}
-	r, err := gfdb.DB(ctx).Model(ctx, public.TableNameModel).
-		Where(entity.AsynchModelCol.IsChatModel, 1).
-		Where(entity.AsynchModelCol.Creator, userInfo.UserName).
-		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
-}