init.go 2.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495
  1. package main
  2. import (
  3. "field_purchasing/config"
  4. "fmt"
  5. "jygit.jydev.jianyu360.cn/data_processing/common_utils/log"
  6. "jygit.jydev.jianyu360.cn/data_processing/common_utils/mongodb"
  7. "jygit.jydev.jianyu360.cn/data_processing/common_utils/mysqldb"
  8. "os"
  9. )
  10. var (
  11. MongoTool *mongodb.MongodbSim
  12. Mysql *mysqldb.Mysql
  13. )
  14. // InitLog @Description
  15. // @Author J 2022/7/26 15:30
  16. func InitLog() {
  17. logcfg := config.Conf.Log
  18. err := log.InitLog(
  19. log.Path(logcfg.LogPath),
  20. log.Level(logcfg.LogLevel),
  21. log.Compress(logcfg.Compress),
  22. log.MaxSize(logcfg.MaxSize),
  23. log.MaxBackups(logcfg.MaxBackups),
  24. log.MaxAge(logcfg.MaxAge),
  25. log.Format(logcfg.Format),
  26. )
  27. if err != nil {
  28. fmt.Printf("InitLog failed: %v\n", err)
  29. os.Exit(1)
  30. }
  31. }
  32. func InitMgo() {
  33. MongoTool = &mongodb.MongodbSim{
  34. MongodbAddr: config.Conf.DB.Mongo.Addr,
  35. Size: config.Conf.DB.Mongo.Size,
  36. DbName: config.Conf.DB.Mongo.Dbname,
  37. UserName: config.Conf.DB.Mongo.User,
  38. Password: config.Conf.DB.Mongo.Password,
  39. }
  40. MongoTool.InitPool()
  41. }
  42. func InitMysql() {
  43. dbcfg := config.Conf.DB.Mysql
  44. Mysql = &mysqldb.Mysql{
  45. Address: dbcfg.Addr,
  46. DBName: dbcfg.DbnameBasic,
  47. UserName: dbcfg.User,
  48. PassWord: dbcfg.Password,
  49. }
  50. Mysql.Init()
  51. }
  52. //
  53. //func InitField() {
  54. // info := Mysql.Find("code_area", nil, "", "", -1, -1)
  55. // for _, m := range *info {
  56. // var key string
  57. // for i, v := range []string{"area", "city", "district"} {
  58. // if i == 0 && util.ObjToString(m[v]) != "" {
  59. // key = util.ObjToString(m[v])
  60. // } else if util.ObjToString(m[v]) != "" {
  61. // key += "," + util.ObjToString(m[v])
  62. // }
  63. // }
  64. // AreaCode[key] = util.ObjToString(m["code"])
  65. // }
  66. // log.Info("InitField", zap.Int("AreaCode", len(AreaCode)))
  67. //}
  68. //
  69. //func InitPoCode() {
  70. // data_class := Mysql.Find("d_yl_productclass_code", nil, "", "", -1, -1)
  71. //
  72. // for _, v := range *data_class {
  73. // name := util.ObjToString(v["name"])
  74. // code := util.ObjToString(v["code"])
  75. // pcode := util.ObjToString(v["pcode"])
  76. // level := util.IntAll(v["level"])
  77. // info := map[string]string{}
  78. // for key, value := range PclassCode[pcode] {
  79. // info[key] = value
  80. // }
  81. // new_key := "class_" + fmt.Sprintf("%d", level)
  82. // info[new_key] = name
  83. // PclassCode[code] = info
  84. //
  85. // }
  86. //
  87. //}