110 lines
2.8 KiB
Go
110 lines
2.8 KiB
Go
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)
|
|
}
|