dm.go 3.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100
  1. package global
  2. import (
  3. _ "dm"
  4. dm "dmgorm2"
  5. "eta_gn/eta_api/utils"
  6. "fmt"
  7. "gorm.io/gorm"
  8. "gorm.io/gorm/logger"
  9. "io"
  10. "log"
  11. "os"
  12. "time"
  13. _ "github.com/go-sql-driver/mysql"
  14. )
  15. // 达梦数据库初始化
  16. func InitDm() {
  17. dmSqlMap := make(map[string]*gorm.DB)
  18. //开启日志
  19. logWriter := io.MultiWriter(utils.Binlog) //binlog日志,记录到文件中去
  20. if utils.RunMode == `debug` { // 测试环境,默认输出在控制台,不需要的话,可以注释if里面下面的这行
  21. logWriter = io.MultiWriter(utils.Binlog, os.Stdout)
  22. }
  23. newLogger := logger.New(log.New(logWriter, "\r\n", log.LstdFlags), logger.Config{
  24. SlowThreshold: 200 * time.Millisecond, //慢sql :200ms
  25. LogLevel: logger.Info, //记录的日志类型,info代表所有信息都记录
  26. IgnoreRecordNotFoundError: true, //是否忽略找不到数据错误信息(只是日志记录记录成err还是普通的输出的区别,并不影响业务代码中的:找不到数据行error)
  27. Colorful: true, //是否颜色输出
  28. })
  29. // 默认库
  30. connectDm(utils.MYSQL_URL, "default", newLogger, dmSqlMap, true)
  31. // 报告库
  32. connectDm(utils.MYSQL_URL_RDDP, "rddp", newLogger, dmSqlMap, false)
  33. // 手工数据库
  34. connectDm(utils.MYSQL_URL_EDB, "edb", newLogger, dmSqlMap, false)
  35. // 指标库
  36. connectDm(utils.MYSQL_URL_DATA, "data", newLogger, dmSqlMap, false)
  37. // 钢联库
  38. connectDm(utils.MYSQL_URL_GL, "gl", newLogger, dmSqlMap, false)
  39. // 用户主库
  40. if utils.MYSQL_WEEKLY_URL != `` && (utils.BusinessCode == utils.BusinessCodeRelease || utils.BusinessCode == utils.BusinessCodeSandbox || utils.BusinessCode == utils.BusinessCodeDebug) {
  41. // 用户主库
  42. connectDm(utils.MYSQL_WEEKLY_URL, "weekly", newLogger, dmSqlMap, false)
  43. }
  44. if utils.MYSQL_AI_URL != "" {
  45. // AI库
  46. connectDm(utils.MYSQL_AI_URL, "ai", newLogger, dmSqlMap, false)
  47. }
  48. //全局赋值数据库链接
  49. DmSQL = dmSqlMap
  50. }
  51. // 达梦数据库连接
  52. func connectDm(dsn, aliasName string, newLogger logger.Interface, dmSqlMap map[string]*gorm.DB, isDefault bool) {
  53. if dsn == `` {
  54. return
  55. }
  56. //dsn := "dm://SYSDBA:SYSDBA001@8.136.199.33:30236?schema=\"eta_master\""
  57. //dsn := "dm://ETA:Han6258199118%60@localhost:5236?schema=ECOLOGY_TARGET"
  58. db, err := gorm.Open(dm.Open(dsn), &gorm.Config{
  59. Logger: newLogger,
  60. })
  61. if err != nil {
  62. //global.LOG.Errorf("mysql 启动异常,数据库:default;Err:", err)
  63. panic(fmt.Errorf("mysql 启动异常,数据库:%s;Err:%s", aliasName, err))
  64. }
  65. //创建连接池
  66. sqlDB, err := db.DB()
  67. if err != nil {
  68. //global.LOG.Errorf("mysql 创建连接池失败,数据库:default;Err:", err)
  69. panic(fmt.Errorf("mysql 创建连接池失败,数据库:%s;Err:%s", aliasName, err))
  70. }
  71. dmSqlMap[aliasName] = db
  72. //默认数据库连接
  73. if isDefault {
  74. DEFAULT_DmSQL = db
  75. }
  76. // SetMaxIdleConns 设置空闲连接池中连接的最大数量
  77. sqlDB.SetMaxIdleConns(50)
  78. // SetMaxOpenConns 设置打开数据库连接的最大数量。
  79. sqlDB.SetMaxOpenConns(100)
  80. // SetConnMaxLifetime 设置了连接可复用的最大时间。
  81. sqlDB.SetConnMaxLifetime(10 * time.Minute)
  82. }