第一次提交

2026-04-29 15:54:14 +08:00
parent 50d2eadbd1
commit e81df5ce5a
51 changed files with 4571 additions and 0 deletions
--- a/service/model_service.go
+++ b/service/model_service.go
@@ -0,0 +1,125 @@
+package service
+
+import (
+	"context"
+	"errors"
+
+	"model-asynch/dao"
+	"model-asynch/model/dto"
+	"model-asynch/model/entity"
+)
+
+var Model = &modelService{}
+
+type modelService struct{}
+
+func (s *modelService) Create(ctx context.Context, req *dto.CreateModelReq) (res *dto.CreateModelRes, err error) {
+	m := &entity.AsynchModel{
+		ModelName:         req.ModelName,
+		ModelsType:        normalizeModelsType(req.ModelsType),
+		BaseURL:           req.BaseURL,
+		Route:             req.Route,
+		HttpMethod:        req.HttpMethod,
+		HeadMsg:           req.HeadMsg,
+		Form:              req.Form,
+		Enabled:           req.Enabled,
+		MaxConcurrency:    req.MaxConcurrency,
+		QueueLimit:        req.QueueLimit,
+		TimeoutSeconds:    req.TimeoutSeconds,
+		ExpectedSeconds:   req.ExpectedSeconds,
+		RetryTimes:        req.RetryTimes,
+		RetryQueueMaxSecs: req.RetryQueueMaxSeconds,
+		AutoCleanSeconds:  req.AutoCleanSeconds,
+		Remark:            req.Remark,
+	}
+	if m.HttpMethod == "" {
+		m.HttpMethod = "POST"
+	}
+	if m.Enabled == 0 {
+		m.Enabled = 1
+	}
+	if m.MaxConcurrency <= 0 {
+		m.MaxConcurrency = 10
+	}
+	if m.QueueLimit <= 0 {
+		m.QueueLimit = 1000
+	}
+	if m.TimeoutSeconds <= 0 {
+		m.TimeoutSeconds = 60
+	}
+	if m.AutoCleanSeconds <= 0 {
+		m.AutoCleanSeconds = 86400
+	}
+	id, err := dao.Model.Insert(ctx, m)
+	if err != nil {
+		return nil, err
+	}
+	return &dto.CreateModelRes{ID: id}, nil
+}
+
+func (s *modelService) Update(ctx context.Context, req *dto.UpdateModelReq) error {
+	data := map[string]any{}
+	if req.BaseURL != "" {
+		data[entity.AsynchModelCol.BaseURL] = req.BaseURL
+	}
+	if req.Route != "" {
+		data[entity.AsynchModelCol.Route] = req.Route
+	}
+	if req.HttpMethod != nil && *req.HttpMethod != "" {
+		data[entity.AsynchModelCol.HttpMethod] = *req.HttpMethod
+	}
+	if req.HeadMsg != nil {
+		data[entity.AsynchModelCol.HeadMsg] = *req.HeadMsg
+	}
+	if req.Form != nil {
+		data[entity.AsynchModelCol.FormJSON] = req.Form
+	}
+	if req.ModelsType != nil {
+		data[entity.AsynchModelCol.ModelsType] = normalizeModelsType(*req.ModelsType)
+	}
+	if req.Enabled != nil {
+		data[entity.AsynchModelCol.Enabled] = *req.Enabled
+	}
+	if req.MaxConcurrency != nil {
+		data[entity.AsynchModelCol.MaxConcurrency] = *req.MaxConcurrency
+	}
+	if req.QueueLimit != nil {
+		data[entity.AsynchModelCol.QueueLimit] = *req.QueueLimit
+	}
+	if req.TimeoutSeconds != nil {
+		data[entity.AsynchModelCol.TimeoutSeconds] = *req.TimeoutSeconds
+	}
+	if req.ExpectedSeconds != nil {
+		data[entity.AsynchModelCol.ExpectedSeconds] = *req.ExpectedSeconds
+	}
+	if req.RetryTimes != nil {
+		data[entity.AsynchModelCol.RetryTimes] = *req.RetryTimes
+	}
+	if req.RetryQueueMaxSeconds != nil {
+		data[entity.AsynchModelCol.RetryQueueMaxSecs] = *req.RetryQueueMaxSeconds
+	}
+	if req.AutoCleanSeconds != nil {
+		data[entity.AsynchModelCol.AutoCleanSeconds] = *req.AutoCleanSeconds
+	}
+	if req.Remark != nil {
+		data[entity.AsynchModelCol.Remark] = *req.Remark
+	}
+	if len(data) == 0 {
+		return errors.New("无可更新字段")
+	}
+	_, err := dao.Model.UpdateByID(ctx, req.ID, data)
+	return err
+}
+
+func (s *modelService) Delete(ctx context.Context, id int64) error {
+	_, err := dao.Model.DeleteByID(ctx, id)
+	return err
+}
+
+func (s *modelService) Get(ctx context.Context, id int64) (*entity.AsynchModel, error) {
+	return dao.Model.GetByID(ctx, id)
+}
+
+func (s *modelService) List(ctx context.Context, pageNum, pageSize int, modelNameLike string) (list []*entity.AsynchModel, total int64, err error) {
+	return dao.Model.List(ctx, pageNum, pageSize, modelNameLike)
+}