model-gateway/dao/model_dao.go

package dao

import (
	"context"
	"fmt"

	"model-asynch/consts/public"
	"model-asynch/model/dto"
	"model-asynch/model/entity"

	"gitea.com/red-future/common/db/gfdb"
	"github.com/gogf/gf/v2/frame/g"
	"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) Update(ctx context.Context, m *dto.UpdateModelReq) (rows int64, err error) {
	// 触发 gfdb 的 updateHook 自动填充 updater，需要显式带 updater 字段
	r, err := gfdb.DB(ctx).Model(ctx, public.TableNameModel).
		OmitEmpty().
		Where(entity.AsynchModelCol.Id, m.ID).
		Data(m).
		Update()
	if err != nil {
		return 0, err
	}
	return r.RowsAffected()
}

func (d *modelDao) UpdateByID(ctx context.Context, m *dto.UpdateModelReq) (rows int64, err error) {
	// 专用于切换会话模型，只更新 is_chat_model 字段
	r, err := gfdb.DB(ctx).Model(ctx, public.TableNameModel).
		Where(entity.AsynchModelCol.Id, m.ID).
		Data(g.Map{
			"is_chat_model": m.IsChatModel,
		}).
		Update()
	if err != nil {
		return 0, err
	}
	return r.RowsAffected()
}

func (d *modelDao) DeleteByID(ctx context.Context, id string) (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) Get(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, modelType int) (list []*entity.AsynchModel, total int64, err error) {
	model := gfdb.DB(ctx).Model(ctx, public.TableNameModel).
		OrderDesc(entity.AsynchModelCol.CreatedAt)
	if modelNameLike != "" {
		model = model.WhereLike(entity.AsynchModelCol.ModelName, "%"+modelNameLike+"%")
	}
	if modelType != 0 {
		model = model.Where(entity.AsynchModelCol.ModelsType, modelType)
	}
	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
}

// ListByCreatorAndPlatform 普通用户：平台公共(tenant_id=0) + 自己创建的(creator=xxx)
func (d *modelDao) ListByCreatorAndPlatform(ctx context.Context, creator string, pageNum, pageSize int, modelNameLike string) (list []*entity.AsynchModel, total int64, err error) {
	// 构建 Where 条件
	whereSQL := "deleted_at IS NULL AND (tenant_id = 1 OR creator = ?)" //1 代表超级管理员
	args := []any{creator}

	if modelNameLike != "" {
		whereSQL += " AND model_name LIKE ?"
		args = append(args, "%"+modelNameLike+"%")
	}

	// 查总数
	countSQL := fmt.Sprintf("SELECT COUNT(1) FROM %s WHERE %s", public.TableNameModel, whereSQL)
	countResult, err := gfdb.DB(ctx).GetAll(ctx, countSQL, args...)
	if err != nil {
		return nil, 0, err
	}
	if len(countResult) > 0 {
		total = gconv.Int64(countResult[0]["count"])
	}

	// 查列表
	querySQL := fmt.Sprintf("SELECT * FROM %s WHERE %s ORDER BY created_at DESC", public.TableNameModel, whereSQL)
	if pageNum > 0 && pageSize > 0 {
		offset := (pageNum - 1) * pageSize
		querySQL += fmt.Sprintf(" LIMIT %d OFFSET %d", pageSize, offset)
	}

	r, err := gfdb.DB(ctx).GetAll(ctx, querySQL, args...)
	if err != nil {
		return nil, 0, err
	}

	err = r.Structs(&list)
	return
}

func (d *modelDao) GetByCreatorAndPlatform(ctx context.Context, creator string, modelNameLike string, modelType int) (list []*entity.AsynchModel, err error) {
	whereSQL := "deleted_at IS NULL AND (tenant_id = 1 OR creator = ?)"
	args := []any{creator}

	if modelNameLike != "" {
		whereSQL += " AND model_name LIKE ?"
		args = append(args, "%"+modelNameLike+"%")
	}
	if modelType != 0 {
		whereSQL += " AND models_type = ?"
		args = append(args, modelType)
	}

	querySQL := fmt.Sprintf("SELECT * FROM %s WHERE %s ORDER BY created_at DESC", public.TableNameModel, whereSQL)

	r, err := gfdb.DB(ctx).GetAll(ctx, querySQL, args...)
	if err != nil {
		return nil, err
	}

	err = r.Structs(&list)
	return
}

func (d *modelDao) GetByIsChatModel(ctx context.Context, userName string) (m *entity.AsynchModel, err error) {
	r, err := gfdb.DB(ctx).Model(ctx, public.TableNameModel).
		Where(entity.AsynchModelCol.IsChatModel, 1).
		Where(entity.AsynchModelCol.Creator, userName).
		One()
	if err != nil {
		return nil, err
	}
	if r.IsEmpty() {
		return nil, nil
	}
	err = r.Struct(&m)
	return
}

// ListAll 用于分组展示：查询全部模型（不按类型过滤，类型拆分在 service 层处理）
func (d *modelDao) ListAll(ctx context.Context) (list []*entity.AsynchModel, err error) {
	r, err := gfdb.DB(ctx).Model(ctx, public.TableNameModel).
		OrderDesc(entity.AsynchModelCol.CreatedAt).
		All()
	if err != nil {
		return nil, err
	}
	err = r.Structs(&list)
	return
}