config.go 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476
  1. package utils
  2. import (
  3. "fmt"
  4. beeLogger "github.com/beego/bee/v2/logger"
  5. "github.com/beego/beego/v2/server/web"
  6. "github.com/qiniu/qmgo"
  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_LOG_URL string
  17. MYSQL_WEEKLY_URL string //用户主库
  18. MgoUrlData string // mongodb数据库连接配置
  19. REDIS_CACHE string //缓存地址
  20. //Rc *cache.Cache //redis缓存
  21. Re error //redis错误
  22. Rc RedisClient //redis缓存
  23. MgoDataCli *qmgo.Client // mongodb客户端连接
  24. MgoDataDbName string // mongodb指标数据的库名
  25. )
  26. // 基础配置
  27. var (
  28. STATIC_DIR string
  29. DesKey string // 接口返回加密KEY
  30. APPNAME string //项目中文名称
  31. EmailSendToUsers string // 邮件提醒人员
  32. ResourceProxyUrl string // 代理资源地址
  33. RESOURCE_DIR string
  34. )
  35. // 公共api内部服务调用
  36. var (
  37. // EDB_LIB_URL 公共指标库
  38. EDB_LIB_URL string
  39. APP_EDB_LIB_NAME_EN string
  40. EDB_LIB_Md5_KEY string
  41. // PublicChartHost 公共图库的host
  42. PublicChartHost string
  43. // SendWxTemplateMsgUrl 模板消息推送
  44. SendWxTemplateMsgUrl string
  45. // SendTemplateMsgAuthorization 模板推送秘钥
  46. SendTemplateMsgAuthorization string
  47. // HandleVideoDecibelUrl 处理音频分贝的url
  48. HandleVideoDecibelUrl string
  49. // EnglishShareUrl 英文研报群发邮件分享地址
  50. EnglishShareUrl string
  51. // EnPolicyReportUrl 英文策略报告
  52. EnPolicyReportUrl string
  53. // GoogleTranslateUrl 谷歌翻译服务
  54. GoogleTranslateUrl string
  55. // AlarmMsgUrl 报警服务地址
  56. AlarmMsgUrl string
  57. EtaBridgeUrl string // 桥接服务地址
  58. EtaBridgeAppNameEn string // 桥接服务英文名称-鉴权用
  59. EtaBridgeMd5Key string // 桥接服务Md5密钥-鉴权用
  60. EtaBridgeDesKey string // 桥接服务Des密钥-解密数据用
  61. EtaBridgeLoginUrl string // 第三方登录鉴权接口地址
  62. EtaBridgeLogoutUrl string // 第三方登出接口地址
  63. )
  64. // TemplateIdByProduct 微信配置信息
  65. var (
  66. TemplateIdByProduct string //产品运行报告通知-模板ID
  67. WxYbAppId string // 研报小程序appid
  68. WxYbAppSecret string // 研报小程序秘钥
  69. )
  70. // ES配置
  71. var (
  72. ES_URL string // ES服务器地址
  73. ES_USERNAME string // ES账号
  74. ES_PASSWORD string // ES密码
  75. )
  76. // ES索引配置
  77. var (
  78. DATA_INDEX_NAME string //数据指标库索引
  79. CHART_INDEX_NAME string //研究图库索引
  80. EsReportIndexName string //研报ES索引
  81. EsEnglishReportIndexName string //英文研报ES索引
  82. MY_CHART_INDEX_NAME string //研究图库(MY ETA)索引
  83. SmartReportIndexName string //智能研报ES索引
  84. )
  85. // 科大讯飞--语音合成
  86. var (
  87. XfHostUrl string
  88. )
  89. // 阿里云配置
  90. var (
  91. Bucketname string
  92. Endpoint string
  93. Imghost string
  94. UploadDir string
  95. Upload_Audio_Dir string
  96. AccessKeyId string
  97. AccessKeySecret string
  98. )
  99. // 阿里云oss前端上传用
  100. var (
  101. AliStsScheme string
  102. RegionId string
  103. RoleArn string
  104. RoleSessionName string
  105. RAMAccessKeyId string
  106. RAMAccessKeySecret string
  107. STSTokenCacheKey string
  108. )
  109. // 阿里云邮箱配置
  110. var (
  111. AliyunEmailAccountName string
  112. AliyunEmailAccessKeyId string
  113. AliyunEmailAccessKeySecret string
  114. )
  115. // 腾讯云邮箱配置
  116. var (
  117. TencentSDKSecretId string // 腾讯云主账号SecretId
  118. TencentSDKSecretKey string // 腾讯云主账号SecretKey
  119. TencentEmailFromEmailAddress string // 腾讯云邮件发信地址
  120. TencentEmailTemplateID uint64 // 云邮件模板ID
  121. )
  122. // 日志配置
  123. var (
  124. LogPath string //调用过程中的日志存放地址
  125. LogFile string
  126. LogDataPath string //调用过程中图表相关的日志存放地址
  127. LogDataFile string
  128. BinLogPath string //数据库相关的日志存放地址
  129. BinLogFile string
  130. ApiLogPath string //接口请求地址和接口返回值日志存放地址
  131. ApiLogFile string
  132. LogMaxDays int //日志最大保留天数
  133. )
  134. // CrmEtaServerUrl CRM-ETA服务地址
  135. var (
  136. CrmEtaServerUrl string
  137. CrmEtaServerDes3Key string
  138. )
  139. // BusinessCode 商家编码
  140. var BusinessCode string
  141. // MinIo配置
  142. var (
  143. MinIoBucketname string
  144. MinIoEndpoint string
  145. MinIoBackEndpoint string
  146. MinIoImghost string
  147. MinIoUploadDir string
  148. MinIoUpload_Audio_Dir string
  149. MinIoAccessKeyId string
  150. MinIoAccessKeySecret string
  151. MinIoUseSSL string
  152. MinIoPort string
  153. MinIoRegion string
  154. )
  155. // 对象存储客户端
  156. var (
  157. ObjectStorageClient string // 目前有oss minio,默认oss
  158. )
  159. // S3配置
  160. var (
  161. S3Endpoint string
  162. S3BucketName string
  163. S3UploadDir string
  164. S3AccessKeyId string
  165. S3AccessKeySecret string
  166. S3Host string
  167. S3Region string
  168. S3ForceStyle string
  169. S3EndpointPort string
  170. S3Protocol string
  171. )
  172. // chrome配置
  173. var (
  174. ChromePath string // chrome浏览器地址
  175. )
  176. var (
  177. UseMongo bool // 是否使用mongo
  178. )
  179. // eta_forum_hub ETA社区桥接服务地址
  180. var (
  181. ETA_FORUM_HUB_URL string
  182. ETA_FORUM_HUB_NAME_EN string
  183. ETA_FORUM_HUB_MD5_KEY string
  184. )
  185. func init() {
  186. tmpRunMode, err := web.AppConfig.String("run_mode")
  187. if err != nil {
  188. panic("配置文件读取run_mode错误 " + err.Error())
  189. }
  190. RunMode = tmpRunMode
  191. fmt.Println("RunMode:", RunMode)
  192. if RunMode == "" {
  193. localIp, err := GetLocalIP()
  194. fmt.Println("localIp:", localIp)
  195. if localIp == "10.0.0.123" {
  196. RunMode = "debug"
  197. } else {
  198. RunMode = "release"
  199. }
  200. fmt.Println("RunMode:", RunMode)
  201. configPath := `/home/code/config/eta_mobile/conf/app.conf`
  202. fmt.Println("configPath:", configPath)
  203. err = web.LoadAppConfig("ini", configPath)
  204. if err != nil {
  205. fmt.Println("web.LoadAppConfig Err:" + err.Error())
  206. }
  207. }
  208. config, err := web.AppConfig.GetSection(RunMode)
  209. if err != nil {
  210. panic("配置文件读取错误 " + err.Error())
  211. }
  212. beeLogger.Log.Info(RunMode + " 模式")
  213. MYSQL_URL = config["mysql_url"]
  214. MYSQL_URL_RDDP = config["mysql_url_rddp"]
  215. MYSQL_URL_EDB = config["mysql_url_edb"]
  216. MYSQL_URL_DATA = config["mysql_url_data"]
  217. MYSQL_URL_GL = config["mysql_url_gl"]
  218. MYSQL_LOG_URL = config["mysql_url_log"]
  219. MYSQL_WEEKLY_URL = config["mysql_url_weekly"]
  220. // mongodb数据库连接配置
  221. MgoUrlData = config["mgo_url_data"]
  222. REDIS_CACHE = config["beego_cache"]
  223. if len(REDIS_CACHE) <= 0 {
  224. panic("redis链接参数没有配置")
  225. }
  226. //Rc, Re = cache.NewCache(REDIS_CACHE) //初始化缓存
  227. //if Re != nil {
  228. // fmt.Println(Re)
  229. // panic(Re)
  230. //}
  231. // 初始化缓存
  232. redisClient, err := initRedis(config["redis_type"], config["beego_cache"])
  233. if err != nil {
  234. fmt.Println("redis链接异常:", err)
  235. panic(any(Re))
  236. }
  237. Rc = redisClient
  238. // 项目中文名称
  239. appNameCn, err := web.AppConfig.String("app_name_cn")
  240. if err != nil {
  241. panic(any("配置文件读取app_name_cn错误 " + err.Error()))
  242. }
  243. APPNAME = appNameCn
  244. // 初始化内部服务配置
  245. // 接口返回加密KEY
  246. DesKey = config["des_key"]
  247. // 邮件提醒人员
  248. EmailSendToUsers = config["email_send_to_users"]
  249. // 静态文件目录
  250. STATIC_DIR = config["static_dir"]
  251. // 代理资源地址
  252. ResourceProxyUrl = config["resource_proxy_url"]
  253. RESOURCE_DIR = config["resource_dir"]
  254. if RESOURCE_DIR == "" {
  255. RESOURCE_DIR = "eta/"
  256. }
  257. // 系统内部服务地址
  258. {
  259. // 公共指标库相关
  260. EDB_LIB_URL = config["edb_lib_url"]
  261. APP_EDB_LIB_NAME_EN = config["app_edb_lib_name_en"]
  262. EDB_LIB_Md5_KEY = config["edb_lib_md5_key"]
  263. // 图表项目域名
  264. PublicChartHost = config["public_chart_host"]
  265. // 发送微信模板消息地址
  266. SendWxTemplateMsgUrl = config["send_wx_template_msg_url"]
  267. // 发送微信模板消息地址
  268. SendTemplateMsgAuthorization = config["send_template_msg_authorization"]
  269. // 语音分贝处理的地址
  270. HandleVideoDecibelUrl = config["handle_video_decibel_url"]
  271. // 英文研报分享的地址
  272. EnglishShareUrl = config["english_share_url"]
  273. // 策略报告地址
  274. EnPolicyReportUrl = config["en_policy_report_url"]
  275. // 谷歌翻译服务地址
  276. GoogleTranslateUrl = config["google_translate_url"]
  277. // 报警服务地址
  278. AlarmMsgUrl = config["alarm_msg_url"]
  279. EtaBridgeUrl = config["eta_bridge_url"] // 桥接服务地址
  280. EtaBridgeAppNameEn = config["eta_bridge_app_name_en"] // 桥接服务英文名称-鉴权用
  281. EtaBridgeMd5Key = config["eta_bridge_md5_key"] // 桥接服务Md5密钥-鉴权用
  282. EtaBridgeDesKey = config["eta_bridge_des_key"] // 桥接服务Des密钥-解密数据用
  283. EtaBridgeLoginUrl = config["eta_bridge_login_url"] // eta桥接服务-登录接口地址
  284. EtaBridgeLogoutUrl = config["eta_bridge_logout_url"] // eta桥接服务-登出接口地址
  285. }
  286. //日志配置
  287. {
  288. LogPath = config["log_path"]
  289. LogFile = config["log_file"]
  290. LogDataPath = config["log_data_path"]
  291. LogDataFile = config["log_data_file"]
  292. BinLogPath = config["binlog_path"]
  293. BinLogFile = config["binlog_file"]
  294. ApiLogPath = config["apilog_path"]
  295. ApiLogFile = config["apilog_file"]
  296. logMaxDaysStr := config["log_max_day"]
  297. LogMaxDays, _ = strconv.Atoi(logMaxDaysStr)
  298. }
  299. {
  300. CrmEtaServerUrl = config["crm_eta_server_url"]
  301. CrmEtaServerDes3Key = config["crm_eta_server_des3_key"]
  302. }
  303. // ES配置
  304. {
  305. ES_URL = config["es_url"]
  306. ES_USERNAME = config["es_username"]
  307. ES_PASSWORD = config["es_password"]
  308. }
  309. // ES 索引
  310. {
  311. DATA_INDEX_NAME = config["data_index_name"]
  312. CHART_INDEX_NAME = config["chart_index_name"]
  313. MY_CHART_INDEX_NAME = config["my_chart_index_name"]
  314. EsReportIndexName = config["es_report_index_name"]
  315. EsEnglishReportIndexName = config["es_english_report_index_name"]
  316. SmartReportIndexName = config["es_smart_report_index_name"]
  317. }
  318. // 微信相关
  319. {
  320. //模板消息
  321. TemplateIdByProduct = config["template_id_by_product"]
  322. WxYbAppId = config["wx_yb_appid"] //弘则研报小程序
  323. WxYbAppSecret = config["wx_yb_app_secret"] // 研报小程序秘钥
  324. }
  325. // 对象存储客户端
  326. ObjectStorageClient = config["object_storage_client"]
  327. // OSS相关
  328. {
  329. Endpoint = config["endpoint"]
  330. Bucketname = config["bucket_name"]
  331. Imghost = config["img_host"]
  332. UploadDir = config["upload_dir"]
  333. Upload_Audio_Dir = config["upload_audio_dir"]
  334. AccessKeyId = config["access_key_id"]
  335. AccessKeySecret = config["access_key_secret"]
  336. }
  337. // OSS相关(前端使用)
  338. {
  339. AliStsScheme = config["ali_sts_scheme"]
  340. RegionId = config["region_id"]
  341. RoleArn = config["role_arn"]
  342. RoleSessionName = config["role_session_name"]
  343. RAMAccessKeyId = config["ram_access_key_id"]
  344. RAMAccessKeySecret = config["ram_access_key_secret"]
  345. STSTokenCacheKey = config["sts_token_cache_key"]
  346. }
  347. // 阿里云邮箱配置
  348. {
  349. AliyunEmailAccountName = config["aliyun_email_account_name"]
  350. AliyunEmailAccessKeyId = config["aliyun_email_access_key_id"]
  351. AliyunEmailAccessKeySecret = config["aliyun_email_access_key_secret"]
  352. }
  353. // 腾讯云邮箱配置
  354. {
  355. TencentSDKSecretId = config["tencent_sdk_secret_id"] // 腾讯云主账号SecretId
  356. TencentSDKSecretKey = config["tencent_sdk_secret_key"] // 腾讯云主账号SecretKey
  357. TencentEmailFromEmailAddress = config["tencent_email_from_email_address"] // 腾讯云邮件发信地址
  358. tencentEmailTemplateIDtr := config["tencent_email_template_id"]
  359. if tencentEmailTemplateIDtr != `` {
  360. tmpTencentEmailTemplateID, err := strconv.Atoi(tencentEmailTemplateIDtr) //弘则研报来源
  361. if err != nil {
  362. panic(any("配置文件读取tencent_email_template_id错误 " + err.Error()))
  363. }
  364. TencentEmailTemplateID = uint64(tmpTencentEmailTemplateID)
  365. }
  366. }
  367. // 科大讯飞
  368. {
  369. XfHostUrl = config["xf_host_url"]
  370. }
  371. // MinIo相关
  372. {
  373. MinIoEndpoint = config["minio_endpoint"]
  374. MinIoBackEndpoint = config["minio_back_endpoint"]
  375. MinIoBucketname = config["minio_bucket_name"]
  376. MinIoImghost = config["minio_img_host"]
  377. MinIoUploadDir = config["minio_upload_dir"]
  378. MinIoUpload_Audio_Dir = config["minio_upload_audio_dir"]
  379. MinIoAccessKeyId = config["minio_access_key_id"]
  380. MinIoAccessKeySecret = config["minio_access_key_secret"]
  381. MinIoUseSSL = config["minio_use_ssl"]
  382. MinIoPort = config["minio_port"]
  383. MinIoRegion = config["minio_region"]
  384. }
  385. // S3-OSS相关
  386. {
  387. S3Endpoint = config["s3_endpoint"]
  388. S3BucketName = config["s3_bucket_name"]
  389. S3Host = config["s3_host"]
  390. S3AccessKeyId = config["s3_access_key_id"]
  391. S3AccessKeySecret = config["s3_access_key_secret"]
  392. S3UploadDir = config["s3_upload_dir"]
  393. S3Region = config["s3_region"]
  394. S3ForceStyle = config["s3_force_style"]
  395. S3EndpointPort = config["s3_endpoint_port"]
  396. S3Protocol = config["s3_protocol"]
  397. }
  398. // 商家编码
  399. BusinessCode = config["business_code"]
  400. // chrome配置
  401. ChromePath = config["chrome_path"]
  402. // eta_forum_hub ETA社区桥接服务地址
  403. {
  404. ETA_FORUM_HUB_URL = config["eta_forum_hub_url"]
  405. ETA_FORUM_HUB_NAME_EN = config["eta_forum_hub_name_en"]
  406. ETA_FORUM_HUB_MD5_KEY = config["eta_forum_hub_md5_key"]
  407. }
  408. // 初始化ES
  409. initEs()
  410. }