prompts-core

service/prompt_service.go

@@ -0,0 +1,92 @@
+package service
+
+import (
+	"context"
+	"encoding/json"
+	"errors"
+
+	"prompts-core/dao"
+	"prompts-core/model/dto"
+	"prompts-core/model/entity"
+)
+
+var Prompt = &promptService{}
+
+type promptService struct{}
+
+func (s *promptService) Create(ctx context.Context, req *dto.CreatePromptReq) (res *dto.CreatePromptRes, err error) {
+	// promptInfo 兜底校验：必须可序列化为 JSON
+	if req.PromptInfo == nil {
+		return nil, errors.New("promptInfo不能为空")
+	}
+	if _, err := json.Marshal(req.PromptInfo); err != nil {
+		return nil, errors.New("promptInfo不是合法JSON")
+	}
+	if req.ResponseJsonSchema == nil {
+		return nil, errors.New("responseJsonSchema不能为空")
+	}
+	if _, err := json.Marshal(req.ResponseJsonSchema); err != nil {
+		return nil, errors.New("responseJsonSchema不是合法JSON")
+	}
+
+	m := &entity.PromptConfig{
+		ModelTypeId:        req.ModelTypeId,
+		ModelType:          req.ModelType,
+		PromptInfo:         req.PromptInfo,
+		ResponseJsonSchema: req.ResponseJsonSchema,
+		Enabled:            1,
+		Version:            req.Version,
+	}
+
+	id, err := dao.Prompt.Insert(ctx, m)
+	if err != nil {
+		return nil, err
+	}
+	return &dto.CreatePromptRes{ID: id}, nil
+}
+
+func (s *promptService) Update(ctx context.Context, req *dto.UpdatePromptReq) error {
+	data := map[string]any{}
+	if req.ModelTypeId != nil && *req.ModelTypeId > 0 {
+		data[entity.PromptConfigCol.ModelTypeId] = *req.ModelTypeId
+	}
+	if req.ModelType != nil && *req.ModelType != "" {
+		data[entity.PromptConfigCol.ModelType] = *req.ModelType
+	}
+	if req.PromptInfo != nil {
+		if _, err := json.Marshal(req.PromptInfo); err != nil {
+			return errors.New("promptInfo不是合法JSON")
+		}
+		data[entity.PromptConfigCol.PromptInfo] = req.PromptInfo
+	}
+	if req.ResponseJsonSchema != nil {
+		if _, err := json.Marshal(req.ResponseJsonSchema); err != nil {
+			return errors.New("responseJsonSchema不是合法JSON")
+		}
+		data[entity.PromptConfigCol.ResponseJsonSchema] = req.ResponseJsonSchema
+	}
+	if req.Enabled != nil {
+		data[entity.PromptConfigCol.Enabled] = *req.Enabled
+	}
+	if req.Version != nil {
+		data[entity.PromptConfigCol.Version] = *req.Version
+	}
+	if len(data) == 0 {
+		return errors.New("无可更新字段")
+	}
+	_, err := dao.Prompt.UpdateByID(ctx, req.ID, data)
+	return err
+}
+
+func (s *promptService) Delete(ctx context.Context, id int64) error {
+	_, err := dao.Prompt.DeleteByID(ctx, id)
+	return err
+}
+
+func (s *promptService) Get(ctx context.Context, id int64) (*entity.PromptConfig, error) {
+	return dao.Prompt.GetByID(ctx, id)
+}
+
+func (s *promptService) List(ctx context.Context, pageNum, pageSize int, modelTypeID *int, modelTypeLike string) (list []*entity.PromptConfig, total int64, err error) {
+	return dao.Prompt.List(ctx, pageNum, pageSize, modelTypeID, modelTypeLike)
+}