config.go 9.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363
  1. package utils
  2. import (
  3. "encoding/json"
  4. "fmt"
  5. beego "github.com/beego/beego/v2/adapter"
  6. "github.com/beego/beego/v2/server/web"
  7. "strconv"
  8. )
  9. var (
  10. RunMode string //运行模式
  11. MYSQL_URL string //数据库连接
  12. MYSQL_URL_RDDP string //数据库连接
  13. MYSQL_URL_EDB string
  14. MYSQL_URL_DATA string
  15. MYSQL_URL_GL string
  16. MYSQL_URL_AI string
  17. MYSQL_URL_ETA string
  18. MYSQL_WEEKLY_URL string //用户主库
  19. DbDriverName string // 数据库驱动名称
  20. REDIS_CACHE string //缓存地址
  21. //Rc *cache.Cache //redis缓存
  22. Re error //redis错误
  23. Rc RedisClient //redis缓存
  24. )
  25. var (
  26. LogPath string
  27. LogFile string
  28. BinLogPath string
  29. BinLogFile string
  30. LogMaxDays int //日志最大保留天数
  31. )
  32. var (
  33. APPNAME string //项目中文名称
  34. EmailSendToUsers string // 邮件提醒人员
  35. RefreshEdbInfoEmailSendToUsers string // 刷新完成通知邮件人员
  36. // AlarmMsgUrl 报警服务地址
  37. AlarmMsgUrl string
  38. )
  39. // 经济数据库
  40. var (
  41. EDB_LIB_URL string
  42. EDB_LIB_URL_LIST []EtaLibUrlList
  43. APP_EDB_LIB_NAME_EN string
  44. EDB_LIB_Md5_KEY string
  45. )
  46. type EtaLibUrlList struct {
  47. Url string
  48. }
  49. // 公共数据服务接口参数
  50. var (
  51. APPID string
  52. SECRET string
  53. )
  54. var (
  55. WxAppId string
  56. WxAppSecret string
  57. TemplateId string
  58. RemindTemplateId string
  59. AppointmentRemindTemplateId string //活动预约通知模板id
  60. YbCommonTemplateId string //研报小程序消息模板ID
  61. SendWxTemplateEnable string //是否开启模版消息,1是发送,其他不发送
  62. WxYbAppId string //微信研报小程序
  63. //内部员工公众号(弘则部门)
  64. AdminWxAppId string
  65. AdminWxAppSecret string
  66. )
  67. // 模板消息推送
  68. var (
  69. SendWxTemplateMsgUrl string
  70. )
  71. // BusinessCode 商家编码
  72. var (
  73. BusinessCode string //商户号
  74. WindServerUrl string //wind api 接口服务器地址
  75. LtServerUrl string //路透 api 接口服务器地址
  76. PbServerUrl string //彭博 api 接口服务器地址
  77. )
  78. // ES配置
  79. var (
  80. ES_URL string // ES服务器地址
  81. ES_USERNAME string // ES账号
  82. ES_PASSWORD string // ES密码
  83. )
  84. // ES索引配置
  85. var (
  86. EsReportIndexName string //研报ES索引
  87. EsEnglishReportIndexName string //英文研报ES索引
  88. SmartReportIndexName string //智能研报ES索引
  89. EsExcelIndexName string // 表格ES索引名称
  90. EsDataSourceIndexName string // 数据源ES索引名称
  91. )
  92. // 科大讯飞--语音合成
  93. var (
  94. XfHostUrl string
  95. )
  96. var (
  97. TemplateIdByProduct string //产品运行报告通知-模板ID
  98. )
  99. // 阿里云配置
  100. var (
  101. Bucketname string
  102. Endpoint string
  103. Imghost string
  104. UploadDir string
  105. Upload_Audio_Dir string
  106. AccessKeyId string
  107. AccessKeySecret string
  108. )
  109. var (
  110. // 是否xy
  111. IsXy string
  112. IsPCSG string
  113. )
  114. var (
  115. EtaBridgeUrl string // 桥接服务地址
  116. EtaBridgeAppNameEn string // 桥接服务英文名称-鉴权用
  117. EtaBridgeMd5Key string // 桥接服务Md5密钥-鉴权用
  118. EtaBridgeDesKey string // 桥接服务Des密钥-解密数据用
  119. SyncUserPath string // 桥接服务-用户同步服务地址
  120. SyncUserTime string // 桥接服务-用户同步的时间配置
  121. SyncIndexPath string // 桥接服务-指标同步服务地址
  122. SyncIndexTime string // 桥接服务-指标同步的时间配置
  123. SyncIndexClassifyTime string // 桥接服务-指标分类同步的时间配置
  124. SyncCrmIndexPath string // 桥接服务-crm指标同步服务地址
  125. SyncCrmIndexTime string // 桥接服务-crm指标同步的时间配置
  126. SyncCrmAssetPkgCd string // 桥接服务-crm资产包配置
  127. SyncCrmDataSourceType string // 桥接服务-crm内部来源系统参数配置
  128. SyncCrmIndexNum int // 桥接服务-每次同步crm数据的数量
  129. )
  130. var (
  131. EtaHubUrl string // 公共服务地址
  132. )
  133. var (
  134. PbRefreshFrequency string // 彭博更新频度
  135. PbRefreshOpen string // 彭博刷新开关 0关闭,默认都是开启的
  136. )
  137. var HzDataApi string
  138. var (
  139. RefreshCalculateEdbTaskTime string // 刷新计算指标的任务时间
  140. )
  141. func init() {
  142. tmpRunMode, err := web.AppConfig.String("run_mode")
  143. if err != nil {
  144. panic("配置文件读取run_mode错误 " + err.Error())
  145. }
  146. RunMode = tmpRunMode
  147. if RunMode == "" {
  148. localIp, err := GetLocalIP()
  149. fmt.Println("localIp:", localIp)
  150. if localIp == "10.0.0.123" {
  151. RunMode = "debug"
  152. } else {
  153. RunMode = "release"
  154. }
  155. //RunMode = "release"
  156. configPath := `/home/code/config/eta_task/conf/app.conf`
  157. err = web.LoadAppConfig("ini", configPath)
  158. if err != nil {
  159. fmt.Println("web.LoadAppConfig Err:" + err.Error())
  160. }
  161. }
  162. config, err := web.AppConfig.GetSection(RunMode)
  163. if err != nil {
  164. panic("配置文件读取错误 " + err.Error())
  165. }
  166. beego.Info(RunMode + " 模式")
  167. MYSQL_URL = config["mysql_url"]
  168. MYSQL_URL_RDDP = config["mysql_url_rddp"]
  169. MYSQL_URL_EDB = config["mysql_url_edb"]
  170. MYSQL_URL_DATA = config["mysql_url_data"]
  171. MYSQL_URL_GL = config["mysql_url_gl"]
  172. MYSQL_URL_ETA = config["mysql_url_eta"]
  173. MYSQL_URL_AI = config["mysql_url_ai"]
  174. // 用户主库
  175. MYSQL_WEEKLY_URL = config["mysql_url_weekly"]
  176. // 数据库驱动名称
  177. DbDriverName = config["db_driver_name"]
  178. if DbDriverName == "" {
  179. DbDriverName = "mysql"
  180. }
  181. REDIS_CACHE = config["beego_cache"]
  182. if len(REDIS_CACHE) <= 0 {
  183. panic(any("redis链接参数没有配置"))
  184. }
  185. //Rc, Re = cache.NewCache(REDIS_CACHE) //初始化缓存
  186. //if Re != nil {
  187. // fmt.Println(Re)
  188. // panic(any(Re))
  189. //}
  190. // 初始化缓存
  191. redisClient, err := initRedis(config["redis_type"], config["beego_cache"])
  192. if err != nil {
  193. fmt.Println("redis链接异常:", err)
  194. panic(any(Re))
  195. }
  196. Rc = redisClient
  197. // 项目中文名称
  198. appNameCn, err := web.AppConfig.String("app_name_cn")
  199. if err != nil {
  200. panic(any("配置文件读取app_name_cn错误 " + err.Error()))
  201. }
  202. APPNAME = appNameCn
  203. {
  204. LogPath = config["log_path"]
  205. LogFile = config["log_file"]
  206. BinLogPath = config["binlog_path"]
  207. BinLogFile = config["binlog_file"]
  208. logMaxDaysStr := config["log_max_day"]
  209. LogMaxDays, _ = strconv.Atoi(logMaxDaysStr)
  210. }
  211. // 邮件提醒人员
  212. EmailSendToUsers = config["email_send_to_users"]
  213. // 刷新完成通知邮件人员
  214. RefreshEdbInfoEmailSendToUsers = config["refresh_edb_info_email_send_to_users"]
  215. // 报警服务地址
  216. AlarmMsgUrl = config["alarm_msg_url"]
  217. // 公共指标库相关
  218. {
  219. EDB_LIB_URL = config["edb_lib_url"]
  220. APP_EDB_LIB_NAME_EN = config["app_edb_lib_name_en"]
  221. EDB_LIB_Md5_KEY = config["edb_lib_md5_key"]
  222. if EDB_LIB_URL == "" {
  223. EDB_LIB_URL = "http://127.0.0.1:8300/edbapi/"
  224. } else {
  225. err = json.Unmarshal([]byte(EDB_LIB_URL), &EDB_LIB_URL_LIST)
  226. if err != nil {
  227. fmt.Println("eta_lib_url 配置错误,请检查,Err:" + err.Error())
  228. }
  229. }
  230. }
  231. APPID = config["appid"]
  232. SECRET = config["secret"]
  233. // 商家编码
  234. BusinessCode = config["business_code"]
  235. //服务检测
  236. WindServerUrl = config["wind_server_url"]
  237. LtServerUrl = config["lt_server_url"]
  238. PbServerUrl = config["pb_server_url"]
  239. // ES配置
  240. {
  241. ES_URL = config["es_url"]
  242. ES_USERNAME = config["es_username"]
  243. ES_PASSWORD = config["es_password"]
  244. }
  245. // ES 索引
  246. {
  247. EsReportIndexName = config["es_report_index_name"]
  248. EsEnglishReportIndexName = config["es_english_report_index_name"]
  249. SmartReportIndexName = config["es_smart_report_index_name"]
  250. }
  251. // 科大讯飞
  252. {
  253. XfHostUrl = config["xf_host_url"]
  254. }
  255. // OSS相关
  256. {
  257. Endpoint = config["endpoint"]
  258. Bucketname = config["bucket_name"]
  259. Imghost = config["img_host"]
  260. UploadDir = config["upload_dir"]
  261. Upload_Audio_Dir = config["upload_audio_dir"]
  262. AccessKeyId = config["access_key_id"]
  263. AccessKeySecret = config["access_key_secret"]
  264. }
  265. // 微信模版消息
  266. {
  267. SendWxTemplateEnable = config["send_wx_template_enable"]
  268. }
  269. // 客户信息
  270. {
  271. IsXy = config["is_xy"]
  272. IsPCSG = config["is_pcsg"]
  273. }
  274. // 桥接服务
  275. {
  276. EtaBridgeUrl = config["eta_bridge_url"] // 桥接服务地址
  277. EtaBridgeAppNameEn = config["eta_bridge_app_name_en"] // 桥接服务英文名称-鉴权用
  278. EtaBridgeMd5Key = config["eta_bridge_md5_key"] // 桥接服务Md5密钥-鉴权用
  279. EtaBridgeDesKey = config["eta_bridge_des_key"] // 桥接服务Des密钥-解密数据用
  280. SyncUserPath = config["sync_user_path"] // 桥接服务-用户同步服务地址
  281. SyncUserTime = config["sync_user_time"] // 桥接服务-用户同步的时间配置
  282. SyncIndexPath = config["sync_index_path"] // 桥接服务-指标同步服务地址
  283. SyncIndexTime = config["sync_index_time"] // 桥接服务-指标同步的时间配置
  284. SyncIndexClassifyTime = config["sync_index_classify_time"] // 桥接服务-指标分类同步的时间配置
  285. SyncCrmIndexPath = config["sync_crm_index_path"] // 桥接服务-crm指标同步服务地址
  286. SyncCrmIndexTime = config["sync_crm_index_time"] // 桥接服务-crm指标同步的时间配置
  287. SyncCrmAssetPkgCd = config["sync_crm_asset_pkg_cd"] // 桥接服务-crm资产包配置
  288. SyncCrmDataSourceType = config["sync_crm_data_source_type"] // 桥接服务-crm内部来源系统参数配置
  289. syncCrmIndexNumStr := config["sync_crm_index_num"] // 桥接服务-每次同步crm数据的数量
  290. if syncCrmIndexNumStr != `` {
  291. SyncCrmIndexNum, _ = strconv.Atoi(syncCrmIndexNumStr)
  292. }
  293. if SyncCrmIndexNum <= 0 {
  294. SyncCrmIndexNum = 2000
  295. }
  296. }
  297. // 公共服务
  298. {
  299. EtaHubUrl = config["eta_hub_url"] // 公共服务地址
  300. }
  301. HzDataApi = config["hz_data_api"]
  302. if HzDataApi == "" {
  303. HzDataApi = "https://hzdataapi.hzinsights.com/hzdataapi/"
  304. }
  305. // 彭博刷新频度
  306. PbRefreshFrequency = config["pb_refresh_frequency"]
  307. // 彭博刷新开关
  308. PbRefreshOpen = config["pb_refresh_open"]
  309. //任务刷新时间
  310. RefreshCalculateEdbTaskTime = config["refresh_calculate_edb_task_time"]
  311. // 初始化ES
  312. initEs()
  313. }