eta_api/utils/config.go

package utils

import (
	"errors"
	"fmt"
	"strconv"

	beeLogger "github.com/beego/bee/v2/logger"
	"github.com/beego/beego/v2/server/web"
	"github.com/qiniu/qmgo"
	"github.com/spf13/viper"
)

var (
	RunMode          string //运行模式
	MYSQL_URL        string //数据库连接
	MYSQL_URL_RDDP   string //数据库连接
	MYSQL_URL_EDB    string
	MYSQL_URL_DATA   string
	MYSQL_URL_GL     string
	MYSQL_LOG_URL    string
	MYSQL_WEEKLY_URL string //用户主库
	MYSQL_AI_URL     string //ETA-AI 数据库
	MgoUrlData       string // mongodb数据库连接配置

	REDIS_CACHE string //缓存地址
	Re            error        //redis错误
	Rc            RedisClient  //redis缓存
	MgoDataCli    *qmgo.Client // mongodb客户端连接
	MgoDataDbName string       // mongodb指标数据的库名
)

var (
	STATIC_DIR       string
	DesKey           string // 接口返回加密KEY
	APPNAME          string //项目中文名称
	EmailSendToUsers string // 邮件提醒人员
	JhGnAppKey       string // 聚合短信，国内AppKey
	JhGjAppKey       string // 聚合短信，国际AppKey
	ResourceProxyUrl string // 代理资源地址
	RESOURCE_DIR     string
)

var (
	ES_URL      string // ES服务器地址
	ES_USERNAME string // ES账号
	ES_PASSWORD string // ES密码
)

var (
	EDB_LIB_URL         string
	APP_EDB_LIB_NAME_EN string
	EDB_LIB_Md5_KEY     string

	PublicChartHost string

	MySteelPostUrl string

	SendWxTemplateMsgUrl string

	SendTemplateMsgAuthorization string

	HandleVideoDecibelUrl string

	Ppt2ImageUrl string

	HongzeOpenApiHost string

	EnglishShareUrl string

	EnPolicyReportUrl string

	GoogleTranslateUrl string

	AlarmMsgUrl string

	ChatUrl string

	EtaBridgeUrl       string // 桥接服务地址
	EtaBridgeAppNameEn string // 桥接服务英文名称-鉴权用
	EtaBridgeMd5Key    string // 桥接服务Md5密钥-鉴权用
	EtaBridgeDesKey    string // 桥接服务Des密钥-解密数据用
	EtaBridgeLoginUrl  string // 第三方登录鉴权接口地址
	EtaBridgeLogoutUrl string // 第三方登出接口地址

	ETAMiniBridgeUrl    string // 桥接服务地址
	ETA_MINI_APPID      string // 桥接服务-鉴权用
	ETA_MINI_APP_SECRET string // 桥接服务-鉴权用
)

var (
	WxId                string //微信原始ID
	WxAppId             string // 公众号appid
	WxAppSecret         string // 公众号秘钥
	TemplateIdByProduct string //产品运行报告通知-模板ID

	WxYbAppId     string // 研报小程序appid
	WxYbId        string // 研报小程序微信原始ID
	WxYbAppSecret string // 研报小程序秘钥
	WxYbPlatform  int    //国能研报来源
)

var (
	DATA_INDEX_NAME                string //数据指标库索引
	CHART_INDEX_NAME               string //研究图库索引
	EsReportIndexName              string //研报ES索引
	EsEnglishReportIndexName       string //英文研报ES索引
	MY_CHART_INDEX_NAME            string //研究图库（MY ETA）索引
	EsSemanticAnalysisDocIndexName string //ES语义分析文档索引名
	SmartReportIndexName           string //智能研报ES索引
)

var (
	XfHostUrl string // 科大讯飞--语音合成
)

var (
	ObjectStorageClient string // 目前有oss minio，默认oss
	PptUpdateApi        string //ppt 通过后端接口的方式上传
)

var (
	Bucketname       string
	Endpoint         string
	Imghost          string
	UploadDir        string
	Upload_Audio_Dir string
	AccessKeyId      string
	AccessKeySecret  string
)

var (
	AliStsScheme       string
	RegionId           string
	RoleArn            string
	RoleSessionName    string
	RAMAccessKeyId     string
	RAMAccessKeySecret string
	STSTokenCacheKey   string
)

var (
	AliyunEmailAccountName       string
	AliyunEmailAccessKeyId       string
	AliyunEmailAccessKeySecret   string
	AliyunEmailReplyAddress      string
	AliyunEmailReplyAddressAlias string
)

var (
	TencentSDKSecretId  string // 腾讯云主账号SecretId
	TencentSDKSecretKey string // 腾讯云主账号SecretKey

	TencentEmailFromEmailAddress string // 腾讯云邮件发信地址
	TencentEmailTemplateID       uint64 // 云邮件模板ID
)

var LibreOfficePath string

var (
	CrmEtaServerUrl     string
	CrmEtaServerDes3Key string
)

var (
	ETA_FORUM_HUB_URL     string
	ETA_FORUM_HUB_NAME_EN string
	ETA_FORUM_HUB_MD5_KEY string
)

var BusinessCode string

var (
	LogPath     string //调用过程中的日志存放地址
	LogFile     string
	LogDataPath string //调用过程中图表相关的日志存放地址
	LogDataFile string
	BinLogPath  string //数据库相关的日志存放地址
	BinLogFile  string
	ApiLogPath  string //接口请求地址和接口返回值日志存放地址
	ApiLogFile  string
	LogMaxDays  int //日志最大保留天数
)

var (
	MinIoBucketname       string
	MinIoEndpoint         string
	MinIoBackEndpoint     string
	MinIoImghost          string
	MinIoUploadDir        string
	MinIoUpload_Audio_Dir string
	MinIoAccessKeyId      string
	MinIoAccessKeySecret  string
	MinIoUseSSL           string
	MinIoPort             string
	MinIoRegion           string
)

var PythonUrlReport2Img string

var EtaAiUrl string

var (
	EtaAppid  string
	EtaSecret string
)

var (
	S3Endpoint        string
	S3BackEndpoint    string
	S3BucketName      string
	S3UploadDir       string
	S3AccessKeyId     string
	S3AccessKeySecret string
	S3Host            string
	S3Region          string
	S3ForceStyle      string
	S3EndpointPort    string
	S3Protocol        string
	S3DisableSSL      string
	S3OpenAcl         string
)

var (
	ChromePath string // chrome浏览器地址
)

var (
	ErrMsgConfigFile string
)

var (
	ViperConfig *viper.Viper
)

var (
	UseMongo bool // 是否使用mongo
)

func init() {
	tmpRunMode, err := web.AppConfig.String("run_mode")
	if err != nil {
		panic(any("配置文件读取run_mode错误 " + err.Error()))
	}
	RunMode = tmpRunMode
	if RunMode == "" {
		configPath := `/home/code/config/eta_gn/eta_api/conf/app.conf`
		fmt.Println("configPath:", configPath)
		err = web.LoadAppConfig("ini", configPath)
		if err != nil {
			fmt.Println("web.LoadAppConfig Err:" + err.Error())
		}

		tmpRunMode, err = web.AppConfig.String("run_mode")
		if err != nil {
			panic(any("配置文件读取run_mode错误 " + err.Error()))
		}
		RunMode = tmpRunMode
	}
	fmt.Println("RunMode:", RunMode)

	config, err := web.AppConfig.GetSection(RunMode)
	if err != nil {
		panic(any("配置文件读取错误 " + err.Error()))
	}
	beeLogger.Log.Info(RunMode + " 模式")
	MYSQL_URL = config["mysql_url"]
	MYSQL_URL_RDDP = config["mysql_url_rddp"]
	MYSQL_URL_EDB = config["mysql_url_edb"]
	MYSQL_URL_DATA = config["mysql_url_data"]
	MYSQL_URL_GL = config["mysql_url_gl"]
	MYSQL_LOG_URL = config["mysql_url_log"]

	MYSQL_WEEKLY_URL = config["mysql_url_weekly"]

	MYSQL_WEEKLY_URL = config["mysql_url_weekly"]
	MYSQL_AI_URL = config["mysql_url_ai"]

	MgoUrlData = config["mgo_url_data"]

	REDIS_CACHE = config["beego_cache"]
	if len(REDIS_CACHE) <= 0 {
		panic(any("redis链接参数没有配置"))
	}

	redisClient, err := initRedis(config["redis_type"], config["beego_cache"])
	if err != nil {
		fmt.Println("redis链接异常：", err)
		panic(any(Re))
	}

	Rc = redisClient

	appNameCn, err := web.AppConfig.String("app_name_cn")
	if err != nil {
		panic(any("配置文件读取app_name_cn错误 " + err.Error()))
	}
	APPNAME = appNameCn


	DesKey = config["des_key"]

	EmailSendToUsers = config["email_send_to_users"]

	JhGnAppKey = config["jh_gn_app_key"]
	JhGjAppKey = config["jh_gj_app_key"]

	Ppt2ImageUrl = config["ppt2_image_url"]

	STATIC_DIR = config["static_dir"]

	ResourceProxyUrl = config["resource_proxy_url"]
	RESOURCE_DIR = config["resource_dir"]
	if RESOURCE_DIR == "" {
		RESOURCE_DIR = "eta_gn/"
	}

	{
		EDB_LIB_URL = config["edb_lib_url"]
		APP_EDB_LIB_NAME_EN = config["app_edb_lib_name_en"]
		EDB_LIB_Md5_KEY = config["edb_lib_md5_key"]

		PublicChartHost = config["public_chart_host"]

		MySteelPostUrl = config["mysteel_post_url"]

		SendWxTemplateMsgUrl = config["send_wx_template_msg_url"]

		SendTemplateMsgAuthorization = config["send_template_msg_authorization"]

		HandleVideoDecibelUrl = config["handle_video_decibel_url"]

		HongzeOpenApiHost = config["hongze_open_api_host"]

		EnglishShareUrl = config["english_share_url"]

		EnPolicyReportUrl = config["en_policy_report_url"]

		GoogleTranslateUrl = config["google_translate_url"]

		AlarmMsgUrl = config["alarm_msg_url"]

		ChatUrl = config["chat_url"]

		EtaBridgeUrl = config["eta_bridge_url"]               // 桥接服务地址
		EtaBridgeAppNameEn = config["eta_bridge_app_name_en"] // 桥接服务英文名称-鉴权用
		EtaBridgeMd5Key = config["eta_bridge_md5_key"]        // 桥接服务Md5密钥-鉴权用
		EtaBridgeDesKey = config["eta_bridge_des_key"]        // 桥接服务Des密钥-解密数据用
		EtaBridgeLoginUrl = config["eta_bridge_login_url"]    // eta桥接服务-登录接口地址
		EtaBridgeLogoutUrl = config["eta_bridge_logout_url"]  // eta桥接服务-登出接口地址
	}

	{
		LogPath = config["log_path"]
		LogFile = config["log_file"]
		LogDataPath = config["log_data_path"]
		LogDataFile = config["log_data_file"]
		BinLogPath = config["binlog_path"]
		BinLogFile = config["binlog_file"]
		ApiLogPath = config["apilog_path"]
		ApiLogFile = config["apilog_file"]
		logMaxDaysStr := config["log_max_day"]
		LogMaxDays, _ = strconv.Atoi(logMaxDaysStr)
	}

	{
		ES_URL = config["es_url"]
		ES_USERNAME = config["es_username"]
		ES_PASSWORD = config["es_password"]
	}

	{
		WxAppId = config["wx_appid"]
		WxAppSecret = config["wx_app_secret"]
		WxId = config["wx_id"]
		TemplateIdByProduct = config["template_id_by_product"]

		WxYbAppId = config["wx_yb_appid"]          //国能研报小程序
		WxYbId = config["wx_yb_id"]                // 研报小程序微信原始ID
		WxYbAppSecret = config["wx_yb_app_secret"] // 研报小程序秘钥
		wxYbPlatformStr := config["wx_yb_platform"]
		if wxYbPlatformStr != `` {
			WxYbPlatform, err = strconv.Atoi(wxYbPlatformStr) //国能研报来源
			if err != nil {
				panic(any("配置文件读取wx_yb_platform错误 " + err.Error()))
			}
		}
	}

	ObjectStorageClient = config["object_storage_client"]

	PptUpdateApi = config["ppt_update_api"]
	{
		Endpoint = config["endpoint"]
		Bucketname = config["bucket_name"]
		Imghost = config["img_host"]
		UploadDir = config["upload_dir"]
		Upload_Audio_Dir = config["upload_audio_dir"]
		AccessKeyId = config["access_key_id"]
		AccessKeySecret = config["access_key_secret"]
	}

	{
		AliStsScheme = config["ali_sts_scheme"]
		RegionId = config["region_id"]
		RoleArn = config["role_arn"]
		RoleSessionName = config["role_session_name"]
		RAMAccessKeyId = config["ram_access_key_id"]
		RAMAccessKeySecret = config["ram_access_key_secret"]
		STSTokenCacheKey = config["sts_token_cache_key"]
	}

	{
		AliyunEmailAccountName = config["aliyun_email_account_name"]
		AliyunEmailAccessKeyId = config["aliyun_email_access_key_id"]
		AliyunEmailAccessKeySecret = config["aliyun_email_access_key_secret"]
		AliyunEmailReplyAddress = config["aliyun_email_reply_address"]
		AliyunEmailReplyAddressAlias = config["aliyun_email_reply_address_alias"]
	}

	{
		TencentSDKSecretId = config["tencent_sdk_secret_id"]                      // 腾讯云主账号SecretId
		TencentSDKSecretKey = config["tencent_sdk_secret_key"]                    // 腾讯云主账号SecretKey
		TencentEmailFromEmailAddress = config["tencent_email_from_email_address"] // 腾讯云邮件发信地址

		tencentEmailTemplateIDtr := config["tencent_email_template_id"]
		if tencentEmailTemplateIDtr != `` {
			tmpTencentEmailTemplateID, err := strconv.Atoi(tencentEmailTemplateIDtr) //国能研报来源
			if err != nil {
				panic(any("配置文件读取tencent_email_template_id错误 " + err.Error()))
			}
			TencentEmailTemplateID = uint64(tmpTencentEmailTemplateID)
		}
	}

	{
		XfHostUrl = config["xf_host_url"]
	}

	{
		DATA_INDEX_NAME = config["data_index_name"]
		CHART_INDEX_NAME = config["chart_index_name"]
		MY_CHART_INDEX_NAME = config["my_chart_index_name"]
		EsReportIndexName = config["es_report_index_name"]
		EsEnglishReportIndexName = config["es_english_report_index_name"]
		EsSemanticAnalysisDocIndexName = config["es_semantic_analysis_doc_index_name"]
		SmartReportIndexName = config["es_smart_report_index_name"]
	}

	CrmEtaServerUrl = config["crm_eta_server_url"]
	CrmEtaServerDes3Key = config["crm_eta_server_des3_key"]
	LibreOfficePath = config["libre_office_path"]

	{
		ETA_FORUM_HUB_URL = config["eta_forum_hub_url"]
		ETA_FORUM_HUB_NAME_EN = config["eta_forum_hub_name_en"]
		ETA_FORUM_HUB_MD5_KEY = config["eta_forum_hub_md5_key"]
	}
	BusinessCode = config["business_code"]
	{
		ETAMiniBridgeUrl = config["eta_mini_bridge_url"]
		ETA_MINI_APPID = config["eta_mini_bridge_appid"]       // 桥接服务-鉴权用
		ETA_MINI_APP_SECRET = config["eta_mini_bridge_secret"] // 桥接服务-鉴权用
	}

	{
		MinIoEndpoint = config["minio_endpoint"]
		MinIoBackEndpoint = config["minio_back_endpoint"]
		MinIoBucketname = config["minio_bucket_name"]
		MinIoImghost = config["minio_img_host"]
		MinIoUploadDir = config["minio_upload_dir"]
		MinIoUpload_Audio_Dir = config["minio_upload_audio_dir"]
		MinIoAccessKeyId = config["minio_access_key_id"]
		MinIoAccessKeySecret = config["minio_access_key_secret"]
		MinIoUseSSL = config["minio_use_ssl"]
		MinIoPort = config["minio_port"]
		MinIoRegion = config["minio_region"]
	}

	{
		S3Endpoint = config["s3_endpoint"]
		S3BackEndpoint = config["s3_back_endpoint"]
		S3BucketName = config["s3_bucket_name"]
		S3Host = config["s3_host"]
		S3AccessKeyId = config["s3_access_key_id"]
		S3AccessKeySecret = config["s3_access_key_secret"]
		S3UploadDir = config["s3_upload_dir"]
		S3Region = config["s3_region"]
		S3ForceStyle = config["s3_force_style"]
		S3EndpointPort = config["s3_endpoint_port"]
		S3Protocol = config["s3_protocol"]
		S3DisableSSL = config["s3_disable_ssl"]
		S3OpenAcl = config["s3_open_acl"]
	}

	PythonUrlReport2Img = config["python_url_report2img"]

	EtaAiUrl = config["eta_ai_url"]
	EtaAppid = config["eta_appid"]
	EtaSecret = config["eta_secret"]

	ChromePath = config["chrome_path"]

	initEs()

	ErrMsgConfigFile = config["err_msg_config_file"]

	if ErrMsgConfigFile != "" {
		viper.SetConfigFile(ErrMsgConfigFile)
		err = viper.ReadInConfig()
		if err != nil {
			if _, ok := err.(viper.ConfigFileNotFoundError); ok {
				fmt.Println(errors.New("config file not found"))
				panic(any("config file not found"))
			} else {
				fmt.Println(errors.New("config file was found but another error was produced err:" + err.Error()))
				panic(any("config file was found but another error was produced err:" + err.Error()))
			}
			return
		}
		ViperConfig = viper.GetViper()
	}
}