连接高斯数据库

go.mod

@@ -9,6 +9,7 @@ require (
 	github.com/dgrijalva/jwt-go v3.2.0+incompatible
 	github.com/go-redis/redis/v8 v8.11.5
 	github.com/go-sql-driver/mysql v1.7.0
+	github.com/lib/pq v1.10.5
 	github.com/minio/minio-go/v7 v7.0.69
 	github.com/rdlucklib/rdluck_tools v1.0.3
 	gopkg.in/gomail.v2 v2.0.0-20160411212932-81ebce5c23df

models/db.go

@@ -2,22 +2,22 @@ package models
 
 import (
 	"eta/eta_report/utils"
+	"github.com/beego/beego/v2/client/orm"
 	_ "github.com/go-sql-driver/mysql"
+	_ "github.com/lib/pq"
 	"time"
-
-	"github.com/beego/beego/v2/client/orm"
 )
 
 func init() {
 	// eta_report
-	_ = orm.RegisterDataBase("default", "mysql", utils.MYSQL_URL)
+	_ = orm.RegisterDataBase("default", utils.DBDRIVER_NAME, utils.MYSQL_URL)
 	orm.SetMaxIdleConns("default", 50)
 	orm.SetMaxOpenConns("default", 100)
 	db, _ := orm.GetDB("default")
 	db.SetConnMaxLifetime(10 * time.Minute)
 
 	// eta_master
-	_ = orm.RegisterDataBase("eta", "mysql", utils.MYSQL_URL_ETA)
+	_ = orm.RegisterDataBase("eta", utils.DBDRIVER_NAME, utils.MYSQL_URL_ETA)
 	orm.SetMaxIdleConns("eta", 50)
 	orm.SetMaxOpenConns("eta", 100)
 	etaDb, _ := orm.GetDB("eta")
@@ -25,7 +25,7 @@ func init() {
 
 	// crm_master
 	if utils.MYSQL_WEEKLY_URL != `` && utils.BusinessCode == utils.BusinessCodeRelease {
-		_ = orm.RegisterDataBase("weekly", "mysql", utils.MYSQL_WEEKLY_URL)
+		_ = orm.RegisterDataBase("weekly", utils.DBDRIVER_NAME, utils.MYSQL_WEEKLY_URL)
 		orm.SetMaxIdleConns("weekly", 50)
 		orm.SetMaxOpenConns("weekly", 100)
 		weeklyDb, _ := orm.GetDB("weekly")

utils/config.go

@@ -11,8 +11,8 @@ var (
 	MYSQL_URL        string //数据库连接
 	MYSQL_URL_ETA    string
 	MYSQL_WEEKLY_URL string //CRM主库
-
-	REDIS_CACHE string //缓存地址
+	DBDRIVER_NAME    string //数据库驱动
+	REDIS_CACHE      string //缓存地址
 	//Rc          *cache.Cache //redis缓存
 	Re error       //redis错误
 	Rc RedisClient //redis缓存
@@ -131,7 +131,10 @@ func init() {
 	MYSQL_URL = config["mysql_url"]
 	MYSQL_URL_ETA = config["mysql_url_eta"]
 	MYSQL_WEEKLY_URL = config["mysql_url_weekly"]
-
+	DBDRIVER_NAME = config["db_driver_name"]
+	if DBDRIVER_NAME == "" {
+		DBDRIVER_NAME = "mysql"
+	}
 	REDIS_CACHE = config["beego_cache"]
 	if len(REDIS_CACHE) <= 0 {
 		panic(any("redis链接参数没有配置"))