init.go 1.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970
  1. package main
  2. import (
  3. "data_clear_sync/config"
  4. "fmt"
  5. "jygit.jydev.jianyu360.cn/data_processing/common_utils/elastic"
  6. "jygit.jydev.jianyu360.cn/data_processing/common_utils/log"
  7. "jygit.jydev.jianyu360.cn/data_processing/common_utils/mongodb"
  8. "os"
  9. )
  10. var (
  11. Mongo, MongoV1, MongoV2 *mongodb.MongodbSim
  12. Es *elastic.Elastic
  13. )
  14. func InitLog() {
  15. logcfg := config.Conf.Log
  16. err := log.InitLog(
  17. log.Path(logcfg.LogPath),
  18. log.Level(logcfg.LogLevel),
  19. log.Compress(logcfg.Compress),
  20. log.MaxSize(logcfg.MaxSize),
  21. log.MaxBackups(logcfg.MaxBackups),
  22. log.MaxAge(logcfg.MaxAge),
  23. log.Format(logcfg.Format),
  24. )
  25. if err != nil {
  26. fmt.Printf("InitLog failed: %v\n", err)
  27. os.Exit(1)
  28. }
  29. }
  30. func InitMgo() {
  31. Mongo = &mongodb.MongodbSim{
  32. MongodbAddr: config.Conf.DB.Mongo.Addr,
  33. DbName: config.Conf.DB.Mongo.Dbname,
  34. Size: config.Conf.DB.Mongo.Size,
  35. UserName: config.Conf.DB.Mongo.User,
  36. Password: config.Conf.DB.Mongo.Password,
  37. }
  38. Mongo.InitPool()
  39. MongoV1 = &mongodb.MongodbSim{
  40. MongodbAddr: config.Conf.DB.Mongo1.Addr,
  41. DbName: config.Conf.DB.Mongo1.Dbname,
  42. Size: config.Conf.DB.Mongo1.Size,
  43. UserName: config.Conf.DB.Mongo1.User,
  44. Password: config.Conf.DB.Mongo1.Password,
  45. }
  46. MongoV1.InitPool()
  47. MongoV2 = &mongodb.MongodbSim{
  48. MongodbAddr: config.Conf.DB.Mongo2.Addr,
  49. DbName: config.Conf.DB.Mongo2.Dbname,
  50. Size: config.Conf.DB.Mongo2.Size,
  51. UserName: config.Conf.DB.Mongo2.User,
  52. Password: config.Conf.DB.Mongo2.Password,
  53. }
  54. MongoV2.InitPool()
  55. }
  56. func InitEs() {
  57. Es = &elastic.Elastic{
  58. S_esurl: config.Conf.DB.Es.Addr,
  59. I_size: config.Conf.DB.Es.Size,
  60. Username: config.Conf.DB.Es.User,
  61. Password: config.Conf.DB.Es.Password,
  62. }
  63. Es.InitElasticSize()
  64. }