eta_api/utils/config.go

package utils

import (
	"fmt"
	beeLogger "github.com/beego/bee/v2/logger"
	"github.com/beego/beego/v2/server/web"
	"github.com/rdlucklib/rdluck_tools/cache"
)

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

	REDIS_CACHE string       //缓存地址
	Rc          *cache.Cache //redis缓存
	Re          error        //redis错误
)

var SystemType string // 系统类型； hz：弘则；trial:试用平台；custom：客户

// 基础配置
var (
	STATIC_DIR       string
	DesKey           string // 接口返回加密KEY
	APPNAME          string //项目中文名称
	EmailSendToUsers string // 邮件提醒人员
)

// 公共api内部服务调用
var (
	// EDB_LIB_URL 公共指标库
	EDB_LIB_URL string

	// SendWxTemplateMsgUrl 模板消息推送
	SendWxTemplateMsgUrl string

	// HandleVideoDecibelUrl 处理音频分贝的url
	HandleVideoDecibelUrl string

	// Ppt2ImageUrl ppt转图片服务的地址
	Ppt2ImageUrl string

	// HongzeOpenApiHost 开放openapi的地址
	HongzeOpenApiHost string

	// EnglishShareUrl 英文研报群发邮件分享地址
	EnglishShareUrl string

	// EnPolicyReportUrl 英文策略报告
	EnPolicyReportUrl string
)

// 微信配置信息
var (
	WxId                string //微信原始ID
	WxAppId             string
	WxAppSecret         string
	TemplateIdByProduct string //产品运行报告通知-模板ID

	WxYbAppId string // 研报小程序appid
)

// ES索引配置
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语义分析文档索引名
)

// 阿里云配置
var (
	Bucketname       string
	Endpoint         string
	Imghost          string
	UploadDir        string
	Upload_Audio_Dir string
	AccessKeyId      string
	AccessKeySecret  string
)

// 科大讯飞--语音合成
var (
	XfSTATUS_FIRST_FRAME    = 0 //第一帧标识
	XfSTATUS_CONTINUE_FRAME = 1 //中间帧标识
	XfSTATUS_LAST_FRAME     = 2 //最后一帧标识
	XfHost                  = "tts-api.xfyun.cn"
	XfMaxFontSize           = 8000
	XfAPPID                 string
	XfAPIKey                string
	XfAPISecret             string
	XfHostUrl               string
	XfOrigin                string
	XfVcn                   string //发言人
)

// LibreOfficePath LibreOfficePath的地址
var LibreOfficePath string

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

	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"]

	REDIS_CACHE = config["beego_cache"]
	if len(REDIS_CACHE) <= 0 {
		panic(any("redis链接参数没有配置"))
	}
	Rc, Re = cache.NewCache(REDIS_CACHE) //初始化缓存
	if Re != nil {
		fmt.Println(Re)
		panic(any(Re))
	}

	// 系统类型
	systemType, err := web.AppConfig.String("system_type")
	if err != nil {
		panic(any("配置文件读取system_type错误 " + err.Error()))
	}
	SystemType = systemType

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

	// 初始化内部服务配置

	// 接口返回加密KEY
	DesKey = config["des_key"]
	// 邮件提醒人员
	EmailSendToUsers = config["email_send_to_users"]

	// ppt 转图片服务地址
	Ppt2ImageUrl = config["ppt2_image_url"]

	// 静态文件目录
	STATIC_DIR = config["static_dir"]

	EDB_LIB_URL = config["edb_lib_url"]

	// 发送微信模板消息地址
	SendWxTemplateMsgUrl = config["send_wx_template_msg_url"]

	// 语音分贝处理的地址
	HandleVideoDecibelUrl = config["handle_video_decibel_url"]

	// 弘则open api的地址
	HongzeOpenApiHost = config["hongze_open_api_host"]

	// 英文研报分享的地址
	EnglishShareUrl = config["english_share_url"]

	// 策略报告地址
	EnPolicyReportUrl = config["en_policy_report_url"]

	// 微信相关
	{
		WxYbAppId = config["wx_yb_appid"] //弘则研报小程序
		WxAppId = config["wx_appid"]
		WxAppSecret = config["wx_app_secret"]
		WxId = config["wx_id"]
		//模板消息
		TemplateIdByProduct = config["template_id_by_product"]
	}

	// OSS相关
	{
		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"]
	}

	// 科大讯飞
	{

		XfAPPID = config["xf_appid"]
		XfAPIKey = config["xf_api_key"]
		XfAPISecret = config["xf_api_secret"]
		XfHostUrl = config["xf_host_url"]
		XfOrigin = config["xf_origin"]
		XfVcn = config["xf_vcn"]
	}

	// ES 索引
	{
		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"]
	}

	LibreOfficePath = config["libre_office_path"]
	return
}