lanyuanxiaoyao
4c6b49099d
- 新增 backend/.golangci.yml 配置 12 个 linter（forbidigo、errorlint、errcheck、staticcheck、revive、gocritic、gosec、bodyclose、noctx、nilerr、goimports、gocyclo） - 新增 lefthook.yml 配置 pre-commit hook 自动运行 lint - 修复存量代码违规：errors.Is/As 替换、zap.Error 替换、import 排序、errcheck 修复 - 更新 README 补充编码规范说明 - 归档 backend-code-lint 变更
86 lines
2.3 KiB
Go
86 lines
2.3 KiB
Go
package service
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"nex/backend/internal/domain"
|
|
"nex/backend/internal/repository"
|
|
"nex/backend/pkg/modelid"
|
|
|
|
appErrors "nex/backend/pkg/errors"
|
|
)
|
|
|
|
type providerService struct {
|
|
providerRepo repository.ProviderRepository
|
|
modelRepo repository.ModelRepository
|
|
cache *RoutingCache
|
|
}
|
|
|
|
func NewProviderService(providerRepo repository.ProviderRepository, modelRepo repository.ModelRepository, cache *RoutingCache) ProviderService {
|
|
return &providerService{providerRepo: providerRepo, modelRepo: modelRepo, cache: cache}
|
|
}
|
|
|
|
func (s *providerService) Create(provider *domain.Provider) error {
|
|
if err := modelid.ValidateProviderID(provider.ID); err != nil {
|
|
return appErrors.ErrInvalidProviderID
|
|
}
|
|
provider.Enabled = true
|
|
err := s.providerRepo.Create(provider)
|
|
if err != nil {
|
|
if isUniqueConstraintError(err) {
|
|
return appErrors.ErrConflict
|
|
}
|
|
return err
|
|
}
|
|
s.cache.SetProvider(provider)
|
|
return nil
|
|
}
|
|
|
|
func (s *providerService) Get(id string) (*domain.Provider, error) {
|
|
return s.providerRepo.GetByID(id)
|
|
}
|
|
|
|
func (s *providerService) List() ([]domain.Provider, error) {
|
|
return s.providerRepo.List()
|
|
}
|
|
|
|
func (s *providerService) Update(id string, updates map[string]interface{}) error {
|
|
if _, ok := updates["id"]; ok {
|
|
return appErrors.ErrImmutableField
|
|
}
|
|
err := s.providerRepo.Update(id, updates)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
s.cache.InvalidateProvider(id)
|
|
return nil
|
|
}
|
|
|
|
func (s *providerService) Delete(id string) error {
|
|
err := s.providerRepo.Delete(id)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
s.cache.InvalidateProvider(id)
|
|
return nil
|
|
}
|
|
|
|
// ListEnabledModels 返回所有启用的模型（用于 Models 接口本地聚合）
|
|
func (s *providerService) ListEnabledModels() ([]domain.Model, error) {
|
|
return s.modelRepo.ListEnabled()
|
|
}
|
|
|
|
// GetModelByProviderAndName 按 provider_id 和 model_name 查询模型（用于 ModelInfo 接口本地查询）
|
|
func (s *providerService) GetModelByProviderAndName(providerID, modelName string) (*domain.Model, error) {
|
|
return s.modelRepo.FindByProviderAndModelName(providerID, modelName)
|
|
}
|
|
|
|
// isUniqueConstraintError 判断是否为数据库唯一约束冲突错误
|
|
func isUniqueConstraintError(err error) bool {
|
|
if err == nil {
|
|
return false
|
|
}
|
|
msg := strings.ToLower(err.Error())
|
|
return strings.Contains(msg, "unique constraint") || strings.Contains(msg, "duplicate")
|
|
}
|