first commit

2026-04-23 13:53:09 +08:00
commit 9de47fa5b8
34 changed files with 2764 additions and 0 deletions
--- a/service/model_service.go
+++ b/service/model_service.go
@@ -0,0 +1,109 @@
+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,
+		BaseURL:          req.BaseURL,
+		Route:            req.Route,
+		HttpMethod:       req.HttpMethod,
+		APIKey:           req.APIKey,
+		Enabled:          req.Enabled,
+		MaxConcurrency:   req.MaxConcurrency,
+		QueueLimit:       req.QueueLimit,
+		TimeoutMs:        req.TimeoutMs,
+		RetryTimes:       req.RetryTimes,
+		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.TimeoutMs <= 0 {
+		m.TimeoutMs = 60000
+	}
+	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.APIKey != nil {
+		data[entity.AsynchModelCol.APIKey] = *req.APIKey
+	}
+	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.TimeoutMs != nil {
+		data[entity.AsynchModelCol.TimeoutMs] = *req.TimeoutMs
+	}
+	if req.RetryTimes != nil {
+		data[entity.AsynchModelCol.RetryTimes] = *req.RetryTimes
+	}
+	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) (list []*entity.AsynchModel, total int64, err error) {
+	return dao.Model.List(ctx, pageNum, pageSize)
+}