init.go 2.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788
  1. package main
  2. import (
  3. "app.yhyue.com/data_processing/common_utils/elastic"
  4. "app.yhyue.com/data_processing/common_utils/log"
  5. "app.yhyue.com/data_processing/common_utils/mongodb"
  6. "app.yhyue.com/data_processing/common_utils/mysqldb"
  7. "fieldproject_common/config"
  8. "fmt"
  9. "os"
  10. )
  11. // InitLog @Description
  12. // @Author J 2022/7/26 15:30
  13. func InitLog() {
  14. logcfg := config.Conf.Log
  15. err := log.InitLog(
  16. log.Path(logcfg.LogPath),
  17. log.Level(logcfg.LogLevel),
  18. log.Compress(logcfg.Compress),
  19. log.MaxSize(logcfg.MaxSize),
  20. log.MaxBackups(logcfg.MaxBackups),
  21. log.MaxAge(logcfg.MaxAge),
  22. log.Format(logcfg.Format),
  23. )
  24. if err != nil {
  25. fmt.Printf("InitLog failed: %v\n", err)
  26. os.Exit(1)
  27. }
  28. }
  29. func InitMgo() {
  30. MongoTool = &mongodb.MongodbSim{
  31. MongodbAddr: config.Conf.DB.Mongo.Addr,
  32. Size: config.Conf.DB.Mongo.Size,
  33. DbName: config.Conf.DB.Mongo.Dbname,
  34. UserName: config.Conf.DB.Mongo.User,
  35. Password: config.Conf.DB.Mongo.Password,
  36. }
  37. MongoTool.InitPool()
  38. MongoTool1 = &mongodb.MongodbSim{
  39. MongodbAddr: config.Conf.DB.Mongo1.Addr,
  40. Size: config.Conf.DB.Mongo1.Size,
  41. DbName: config.Conf.DB.Mongo1.Dbname,
  42. UserName: config.Conf.DB.Mongo1.User,
  43. Password: config.Conf.DB.Mongo1.Password,
  44. }
  45. MongoTool1.InitPool()
  46. MongoTool2 = &mongodb.MongodbSim{
  47. MongodbAddr: config.Conf.DB.Mongo2.Addr,
  48. Size: config.Conf.DB.Mongo2.Size,
  49. DbName: config.Conf.DB.Mongo2.Dbname,
  50. UserName: config.Conf.DB.Mongo2.User,
  51. Password: config.Conf.DB.Mongo2.Password,
  52. }
  53. MongoTool2.InitPool()
  54. }
  55. func InitEs() {
  56. Es = &elastic.Elastic{
  57. S_esurl: config.Conf.DB.Es.Addr,
  58. I_size: config.Conf.DB.Es.Size,
  59. }
  60. Es.InitElasticSize()
  61. }
  62. func InitMysql() {
  63. dbcfg := config.Conf.DB.Mysql
  64. MysqlB = &mysqldb.Mysql{
  65. Address: dbcfg.Addr,
  66. DBName: dbcfg.DbnameBasic,
  67. UserName: dbcfg.User,
  68. PassWord: dbcfg.Password,
  69. }
  70. MysqlB.Init()
  71. MysqlM = &mysqldb.Mysql{
  72. Address: dbcfg.Addr,
  73. DBName: dbcfg.DbnameMedical,
  74. UserName: dbcfg.User,
  75. PassWord: dbcfg.Password,
  76. }
  77. MysqlM.Init()
  78. }