config.go 7.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271
  1. package utils
  2. import (
  3. "encoding/json"
  4. "fmt"
  5. "strconv"
  6. beeLogger "github.com/beego/bee/v2/logger"
  7. "github.com/beego/beego/v2/server/web"
  8. "github.com/qiniu/qmgo"
  9. )
  10. var (
  11. RunMode string //运行模式
  12. MYSQL_URL string //数据库连接
  13. MYSQL_URL_EDB string
  14. MYSQL_URL_GL string
  15. MYSQL_URL_MASTER string
  16. MgoUrlData string // mongodb数据库连接配置
  17. PYTHON_MYSQL_HOST string // python数据库链接主机地址
  18. PYTHON_MYSQL_USER string // python数据库链接账号
  19. PYTHON_MYSQL_PASSWD string // python数据库链接密码
  20. PYTHON_MYSQL_DB string // python数据库链接数据库名
  21. PYTHON_MONGO_HOST string // python链接mongo主机地址
  22. PYTHON_MONGO_DATABASE string // python链接mongo的库名
  23. PYTHON_MONGO_AUTH_MECHANISM string // python认证机制
  24. PYTHON_PATH string // python可执行文件地址
  25. REDIS_CACHE string //缓存地址
  26. //Rc *cache.Cache //redis缓存
  27. Re error //redis错误
  28. Rc RedisClient //redis缓存
  29. MgoDataCli *qmgo.Client // mongodb客户端连接
  30. MgoDataDbName string // mongodb指标数据的库名
  31. )
  32. // DATA_INDEX_NAME 数据指标库ES索引名称
  33. var DATA_INDEX_NAME string
  34. // 基础配置
  35. var (
  36. // APP_NAME_CN 项目中文名称
  37. APP_NAME_CN string
  38. // APP_NAME_EN 项目英文名称
  39. APP_NAME_EN string
  40. // EmailSendToUsers 邮件提醒人员
  41. EmailSendToUsers string
  42. // STATIC_DIR 文件目录
  43. STATIC_DIR string
  44. // Md5Key 签名秘钥
  45. Md5Key string
  46. // AlarmMsgUrl 报警服务地址
  47. AlarmMsgUrl string
  48. EtaBridgeUrl string // 桥接服务地址
  49. EtaBridgeAppNameEn string // 桥接服务英文名称-鉴权用
  50. EtaBridgeMd5Key string // 桥接服务Md5密钥-鉴权用
  51. EtaBridgeDesKey string // 桥接服务Des密钥-解密数据用
  52. )
  53. // 第三方服务地址配置
  54. var (
  55. Hz_Ths_Data_Url string //同花顺,万得接口服务地址(本地机器地址)
  56. Hz_Pb_Data_Url string //彭博接口地址
  57. Hz_LT_Data_Url string //路透社接口地址
  58. EDB_DATA_LIMIT = 10
  59. Hz_Wind_Data_Url_LIST []WindUrlMap // wind接口服务地址配置(客户本地机器,是个list列表)
  60. ThsDataMethod string //同花顺数据获取的方式,app是通过终端;api是通过接口
  61. ThsRefreshToken string // 同花顺的刷新token
  62. MysteelChemicalApiToken string // 钢联化工的api数据token
  63. )
  64. type WindUrlMap struct {
  65. Url string
  66. Num int
  67. }
  68. // ES配置
  69. var (
  70. ES_URL string // ES服务器地址
  71. ES_USERNAME string // ES账号
  72. ES_PASSWORD string // ES密码
  73. )
  74. // 初始化数据用户信息
  75. var (
  76. BusinessCode string //商户号
  77. InitAdminId string
  78. InitAdminName string
  79. )
  80. var (
  81. LogPath string
  82. LogFile string
  83. BinLogPath string
  84. BinLogFile string
  85. ApiLogPath string
  86. ApiLogFile string
  87. LogMaxDays int //日志最大保留天数
  88. )
  89. var (
  90. UseMongo bool // 是否使用mongo
  91. )
  92. var (
  93. SmmDataMethod string // Smm有色指标数据获取方式
  94. )
  95. func init() {
  96. tmpRunMode, err := web.AppConfig.String("run_mode")
  97. if err != nil {
  98. panic("配置文件读取run_mode错误 " + err.Error())
  99. }
  100. RunMode = tmpRunMode
  101. fmt.Println("RunMode:", RunMode)
  102. if RunMode == "" {
  103. localIp, err := GetLocalIP()
  104. if localIp == "10.0.0.123" {
  105. RunMode = "debug"
  106. } else {
  107. RunMode = "release"
  108. }
  109. configPath := `/home/code/config/eta_index_lib/conf/app.conf`
  110. err = web.LoadAppConfig("ini", configPath)
  111. if err != nil {
  112. fmt.Println("web.LoadAppConfig Err:" + err.Error())
  113. }
  114. }
  115. config, err := web.AppConfig.GetSection(RunMode)
  116. if err != nil {
  117. panic("配置文件读取错误 " + err.Error())
  118. }
  119. beeLogger.Log.Info(RunMode + " 模式")
  120. MYSQL_URL = config["mysql_url"]
  121. MYSQL_URL_EDB = config["mysql_url_edb"]
  122. MYSQL_URL_GL = config["mysql_url_gl"]
  123. MYSQL_URL_MASTER = config["mysql_url_master"]
  124. // mongodb数据库连接配置
  125. MgoUrlData = config["mgo_url_data"]
  126. //python链接
  127. PYTHON_MYSQL_HOST = config["python_mysql_host"]
  128. PYTHON_MYSQL_USER = config["python_mysql_user"]
  129. PYTHON_MYSQL_PASSWD = config["python_mysql_passwd"]
  130. PYTHON_MYSQL_DB = config["python_mysql_db"]
  131. PYTHON_MONGO_HOST = config["python_mongo_host"] // python链接mongo主机地址
  132. PYTHON_MONGO_DATABASE = config["python_mongo_db"] // python链接mongo的库名
  133. PYTHON_MONGO_AUTH_MECHANISM = config["python_mongo_auth_mechanism"] // python链接认证机制
  134. PYTHON_PATH = config["python_path"]
  135. REDIS_CACHE = config["beego_cache"]
  136. if len(REDIS_CACHE) <= 0 {
  137. panic("redis链接参数没有配置")
  138. }
  139. //Rc, Re = cache.NewCache(REDIS_CACHE) //初始化缓存
  140. //if Re != nil {
  141. // fmt.Println(Re)
  142. // panic(Re)
  143. //}
  144. // 初始化缓存
  145. redisClient, err := initRedis(config["redis_type"], config["beego_cache"])
  146. if err != nil {
  147. fmt.Println("redis链接异常:", err)
  148. panic(any(Re))
  149. }
  150. Rc = redisClient
  151. // 项目中文名称
  152. appNameCn, err := web.AppConfig.String("app_name_cn")
  153. if err != nil {
  154. panic(any("配置文件读取app_name_cn错误 " + err.Error()))
  155. }
  156. APP_NAME_CN = appNameCn
  157. // 初始化内部服务配置
  158. {
  159. // 项目英文文名称
  160. APP_NAME_EN = config["app_name_en"]
  161. // 邮件提醒人员
  162. EmailSendToUsers = config["email_send_to_users"]
  163. // 静态文件目录
  164. STATIC_DIR = config["static_dir"]
  165. // 签名秘钥
  166. Md5Key = config["md5_key"]
  167. // AlarmMsgUrl 报警服务地址
  168. AlarmMsgUrl = config["alarm_msg_url"]
  169. EtaBridgeUrl = config["eta_bridge_url"] // 桥接服务地址
  170. EtaBridgeAppNameEn = config["eta_bridge_app_name_en"] // 桥接服务英文名称-鉴权用
  171. EtaBridgeMd5Key = config["eta_bridge_md5_key"] // 桥接服务Md5密钥-鉴权用
  172. EtaBridgeDesKey = config["eta_bridge_des_key"] // 桥接服务Des密钥-解密数据用
  173. }
  174. // 第三方服务地址配置
  175. {
  176. Hz_Ths_Data_Url = config["hz_ths_data_url"]
  177. Hz_Pb_Data_Url = config["hz_pb_data_url"] //彭博接口地址
  178. Hz_LT_Data_Url = config["hz_lt_data_url"] //路透社接口地址
  179. // wind服务地址配置(多组服务器配置)
  180. windUrlListStr := config["hz_wind_data_url_list"]
  181. if windUrlListStr != `` {
  182. windUrlList := make([]WindUrlMap, 0)
  183. err = json.Unmarshal([]byte(windUrlListStr), &windUrlList)
  184. if err != nil {
  185. panic(any("配置文件读取hz_wind_data_url_list错误 " + err.Error()))
  186. }
  187. Hz_Wind_Data_Url_LIST = windUrlList
  188. }
  189. ThsRefreshToken = config["ths_refresh_token"] //同花顺的刷新token
  190. ThsDataMethod = config["ths_data_method"] // 同花顺数据获取的方式
  191. // 默认走api接口
  192. if ThsDataMethod == `` {
  193. ThsDataMethod = "api"
  194. }
  195. MysteelChemicalApiToken = config["mysteel_chemical_api_token"]
  196. }
  197. // ES配置
  198. {
  199. ES_URL = config["es_url"]
  200. ES_USERNAME = config["es_username"]
  201. ES_PASSWORD = config["es_password"]
  202. DATA_INDEX_NAME = config["data_index_name"]
  203. }
  204. //日志配置
  205. {
  206. LogPath = config["log_path"]
  207. LogFile = config["log_file"]
  208. BinLogPath = config["binlog_path"]
  209. BinLogFile = config["binlog_file"]
  210. ApiLogPath = config["apilog_path"]
  211. ApiLogFile = config["apilog_file"]
  212. logMaxDaysStr := config["log_max_day"]
  213. LogMaxDays, _ = strconv.Atoi(logMaxDaysStr)
  214. }
  215. // 数据指标库ES索引名称
  216. InitAdminId = config["init_admin_id"]
  217. //初始化数据用户配置
  218. InitAdminName = config["init_admin_name"]
  219. // 初始化商家编码
  220. BusinessCode = config["business_code"]
  221. // 初始化ES
  222. initEs()
  223. }
  224. //修改接口文档
  225. //http://8.136.199.33:8300/swagger/