完成文本审查服务开发

api/service/moderation/moderation_manager.go

@@ -0,0 +1,58 @@
+package moderation
+
+import (
+	"geekai/core/types"
+
+	logger2 "geekai/logger"
+)
+
+// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
+// * Copyright 2023 The Geek-AI Authors. All rights reserved.
+// * Use of this source code is governed by a Apache-2.0 license
+// * that can be found in the LICENSE file.
+// * @Author yangjian102621@163.com
+// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
+
+var logger = logger2.GetLogger()
+
+type Service interface {
+	Moderate(text string) (types.ModerationResult, error)
+}
+
+type ServiceManager struct {
+	gitee   *GiteeAIModeration
+	baidu   *BaiduAIModeration
+	tencent *TencentAIModeration
+	active  string
+}
+
+func NewServiceManager(gitee *GiteeAIModeration, baidu *BaiduAIModeration, tencent *TencentAIModeration) *ServiceManager {
+	return &ServiceManager{
+		gitee:   gitee,
+		baidu:   baidu,
+		tencent: tencent,
+	}
+}
+
+func (s *ServiceManager) GetService() Service {
+	switch s.active {
+	case types.ModerationBaidu:
+		return s.baidu
+	case types.ModerationTencent:
+		return s.tencent
+	default:
+		return s.gitee
+	}
+}
+
+func (s *ServiceManager) UpdateConfig(config types.ModerationConfig) {
+	switch config.Active {
+	case types.ModerationGitee:
+		s.gitee.UpdateConfig(config.Gitee)
+	case types.ModerationBaidu:
+		s.baidu.UpdateConfig(config.Baidu)
+	case types.ModerationTencent:
+		s.tencent.UpdateConfig(config.Tencent)
+	}
+	s.active = config.Active
+}