eta_index_lib/utils/config.go

package utils

import (
	"encoding/json"
	"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_EDB string
	MYSQL_URL_GL  string

	PYTHON_MYSQL_HOST   string //python数据库链接主机地址
	PYTHON_MYSQL_USER   string //python数据库链接账号
	PYTHON_MYSQL_PASSWD string //python数据库链接密码
	PYTHON_MYSQL_DB     string //python数据库链接数据库名
	PYTHON_PATH         string //python可执行文件地址

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

// DATA_INDEX_NAME 数据指标库ES索引名称
var DATA_INDEX_NAME string

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

	// APP_NAME_CN 项目中文名称
	APP_NAME_CN string

	// APP_NAME_EN 项目英文名称
	APP_NAME_EN string

	// EmailSendToUsers 邮件提醒人员
	EmailSendToUsers string

	// STATIC_DIR 文件目录
	STATIC_DIR string

	// Md5Key 签名秘钥
	Md5Key string

	// AlarmMsgUrl 报警服务地址
	AlarmMsgUrl string
)

// 第三方服务地址配置
var (
	Hz_Ths_Data_Url       string //同花顺，万得接口服务地址（本地机器地址）
	Hz_Pb_Data_Url        string //彭博接口地址
	Hz_LT_Data_Url        string //路透社接口地址
	EDB_DATA_LIMIT        = 10
	Hz_Wind_Data_Url_LIST []WindUrlMap // wind接口服务地址配置（客户本地机器，是个list列表）

	ThsRefreshToken string // 同花顺的刷新token
)

type WindUrlMap struct {
	Url string
	Num int
}

func init() {
	tmpRunMode, err := web.AppConfig.String("run_mode")
	if err != nil {
		panic("配置文件读取run_mode错误 " + err.Error())
	}
	RunMode = tmpRunMode
	fmt.Println("RunMode:", RunMode)
	if RunMode == "" {
		localIp, err := GetLocalIP()
		if localIp == "10.0.0.123" {
			RunMode = "debug"
		} else {
			RunMode = "release"
		}
		configPath := `/home/code/config/hongze_edb_lib/conf/app.conf`
		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("配置文件读取错误 " + err.Error())
	}
	beeLogger.Log.Info(RunMode + " 模式")
	MYSQL_URL = config["mysql_url"]
	MYSQL_URL_EDB = config["mysql_url_edb"]
	MYSQL_URL_GL = config["mysql_url_gl"]

	//python链接
	PYTHON_MYSQL_HOST = config["python_mysql_host"]
	PYTHON_MYSQL_USER = config["python_mysql_user"]
	PYTHON_MYSQL_PASSWD = config["python_mysql_passwd"]
	PYTHON_MYSQL_DB = config["python_mysql_db"]
	PYTHON_PATH = config["python_path"]

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

	// 初始化内部服务配置
	{
		// 项目英文文名称
		APP_NAME_EN = config["app_name_en"]

		// 邮件提醒人员
		EmailSendToUsers = config["email_send_to_users"]

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

		// 签名秘钥
		Md5Key = config["md5_key"]

		// AlarmMsgUrl 报警服务地址
		AlarmMsgUrl = config["alarm_msg_url"]
	}

	// 第三方服务地址配置
	{
		Hz_Ths_Data_Url = config["hz_ths_data_url"]
		Hz_Pb_Data_Url = config["hz_pb_data_url"] //彭博接口地址
		Hz_LT_Data_Url = config["hz_lt_data_url"] //路透社接口地址

		// wind服务地址配置（多组服务器配置）
		windUrlListStr := config["hz_wind_data_url_list"]
		if windUrlListStr != `` {
			windUrlList := make([]WindUrlMap, 0)
			err = json.Unmarshal([]byte(windUrlListStr), &windUrlList)
			if err != nil {
				panic(any("配置文件读取hz_wind_data_url_list错误 " + err.Error()))
			}
			Hz_Wind_Data_Url_LIST = windUrlList
		}

		ThsRefreshToken = config["ths_refresh_token"] //同花顺的刷新token
	}

	// 数据指标库ES索引名称
	DATA_INDEX_NAME = config["data_index_name"]

}

//修改接口文档
//http://8.136.199.33:8300/swagger/