catface_backend_go/bootstrap/init.go

package bootstrap

import (
	_ "catface/app/core/destroy" // 监听程序退出信号，用于资源的释放
	"catface/app/global/my_errors"
	"catface/app/global/variable"
	"catface/app/http/validator/common/register_validator"
	"catface/app/service/sys_log_hook"
	"catface/app/utils/casbin_v2"
	"catface/app/utils/gorm_v2"
	"catface/app/utils/llm_factory"
	"catface/app/utils/snow_flake"
	"catface/app/utils/validator_translation"
	"catface/app/utils/websocket/core"
	"catface/app/utils/yml_config"
	"catface/app/utils/zap_factory"
	"log"
	"os"

	"github.com/elastic/go-elasticsearch/v8"
)

func checkRequiredFolders() {
	//1.检查配置文件是否存在
	if _, err := os.Stat(variable.BasePath + "/config/config.yml"); err != nil {
		log.Fatal(my_errors.ErrorsConfigYamlNotExists + err.Error())
	}
	if _, err := os.Stat(variable.BasePath + "/config/gorm_v2.yml"); err != nil {
		log.Fatal(my_errors.ErrorsConfigGormNotExists + err.Error())
	}
	if _, err := os.Stat(variable.BasePath + "/config/prompts.yml"); err != nil {
		log.Fatal(my_errors.ErrorsPromptsYmlNotExists + err.Error())
	}
	//2.检查public目录是否存在
	if _, err := os.Stat(variable.BasePath + "/public/"); err != nil {
		log.Fatal(my_errors.ErrorsPublicNotExists + err.Error())
	}
	//3.检查storage/logs 目录是否存在
	if _, err := os.Stat(variable.BasePath + "/store/logs/"); err != nil {
		log.Fatal(my_errors.ErrorsStorageLogsNotExists + err.Error())
	}
	// todo
}

func init() {
	// 1. 初始化 项目根路径，参见 variable 常量包，相关路径：app\global\variable\variable.go

	//2.检查配置文件以及日志目录等非编译性的必要条件
	checkRequiredFolders()

	//3.初始化表单参数验证器，注册在容器（Web、Api共用容器）
	register_validator.WebRegisterValidator()
	register_validator.ApiRegisterValidator()

	// 4.启动针对配置文件(confgi.yml、gorm_v2.yml)变化的监听， 配置文件操作指针，初始化为全局变量
	variable.ConfigYml = yml_config.CreateYamlFactory()
	variable.ConfigYml.ConfigFileChangeListen()
	// config>gorm_v2.yml 启动文件变化监听事件
	variable.ConfigGormv2Yml = variable.ConfigYml.Clone("gorm_v2")
	variable.ConfigGormv2Yml.ConfigFileChangeListen()

	variable.PromptsYml = variable.ConfigYml.Clone("prompts")
	variable.PromptsYml.ConfigFileChangeListen()

	// 5.初始化全局日志句柄，并载入日志钩子处理函数
	variable.ZapLog = zap_factory.CreateZapFactory(sys_log_hook.ZapLogHandler)

	// 6.根据配置初始化 gorm mysql 全局 *gorm.Db
	if variable.ConfigGormv2Yml.GetInt("Gormv2.Mysql.IsInitGlobalGormMysql") == 1 {
		if dbMysql, err := gorm_v2.GetOneMysqlClient(); err != nil {
			log.Fatal(my_errors.ErrorsGormInitFail + err.Error())
		} else {
			variable.GormDbMysql = dbMysql
		}
	}
	// 根据配置初始化 gorm sqlserver 全局 *gorm.Db
	if variable.ConfigGormv2Yml.GetInt("Gormv2.Sqlserver.IsInitGlobalGormSqlserver") == 1 {
		if dbSqlserver, err := gorm_v2.GetOneSqlserverClient(); err != nil {
			log.Fatal(my_errors.ErrorsGormInitFail + err.Error())
		} else {
			variable.GormDbSqlserver = dbSqlserver
		}
	}
	// 根据配置初始化 gorm postgresql 全局 *gorm.Db
	if variable.ConfigGormv2Yml.GetInt("Gormv2.PostgreSql.IsInitGlobalGormPostgreSql") == 1 {
		if dbPostgre, err := gorm_v2.GetOnePostgreSqlClient(); err != nil {
			log.Fatal(my_errors.ErrorsGormInitFail + err.Error())
		} else {
			variable.GormDbPostgreSql = dbPostgre
		}
	}

	// 7.雪花算法全局变量
	variable.SnowFlake = snow_flake.CreateSnowflakeFactory()

	// 8.websocket Hub中心启动
	if variable.ConfigYml.GetInt("Websocket.Start") == 1 {
		// websocket 管理中心hub全局初始化一份
		variable.WebsocketHub = core.CreateHubFactory()
		if Wh, ok := variable.WebsocketHub.(*core.Hub); ok {
			go Wh.Run()
		}
	}

	// 9.casbin 依据配置文件设置参数(IsInit=1)初始化
	if variable.ConfigYml.GetInt("Casbin.IsInit") == 1 {
		var err error
		if variable.Enforcer, err = casbin_v2.InitCasbinEnforcer(); err != nil {
			log.Fatal(err.Error())
		}
	}
	//10.全局注册 validator 错误翻译器,zh 代表中文，en 代表英语
	if err := validator_translation.InitTrans("zh"); err != nil {
		log.Fatal(my_errors.ErrorsValidatorTransInitFail + err.Error())
	}

	// 11. GLM 资源池管理 初始化
	variable.GlmClientHub = llm_factory.InitGlmClientHub(
		variable.ConfigYml.GetInt("Glm.MaxIdle"),
		variable.ConfigYml.GetInt("Glm.MaxActive"),
		variable.ConfigYml.GetInt("Glm.LifeTime"),
		variable.ConfigYml.GetString("Glm.ApiKey"),
		variable.ConfigYml.GetString("Glm.DefaultModelName"),
		variable.PromptsYml.GetString("Prompt.InitPrompt"),
	)

	// 12. ES 客户端启动
	if variable.ConfigYml.GetInt("ElasticSearch.Start") == 1 {
		var err error
		variable.ElasticClient, err = elasticsearch.NewClient(elasticsearch.Config{
			Addresses: []string{variable.ConfigYml.GetString("ElasticSearch.Addr")},
		})
		if err != nil {
			log.Fatal(my_errors.ErrorsInitConnFail + err.Error())
		}
	}

}