中石化数据库驱动切换

go.mod

@@ -8,6 +8,7 @@ require (
 	github.com/go-redis/redis/v8 v8.11.5
 	github.com/go-sql-driver/mysql v1.7.0
 	github.com/gonum/stat v0.0.0-20181125101827-41a0da705a5b
+	github.com/lib/pq v1.10.5
 	github.com/nosixtools/solarlunar v0.0.0-20211112060703-1b6dea7b4a19
 	github.com/qiniu/qmgo v1.1.8
 	github.com/rdlucklib/rdluck_tools v1.0.3

models/db.go

@@ -4,22 +4,41 @@ import (
 	"eta/eta_chart_lib/models/data_manage"
 	"eta/eta_chart_lib/models/data_manage/future_good"
 	"eta/eta_chart_lib/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() {
 
-	_ = orm.RegisterDataBase("default", "mysql", utils.MYSQL_URL)
+	/*// 高斯数据库的连接字符串
+	connStr := "user=lhr dbname=postgres sslmode=disable password=lhr host=192.168.0.106 port=7654"
+
+	// 使用pq连接高斯数据库
+	db1, err := sql.Open("postgres", connStr)
+	if err != nil {
+		panic(err)
+	}
+	defer db1.Close()
+
+	// 测试连接
+	err = db1.Ping()
+	if err != nil {
+		panic(err)
+	}
+
+	fmt.Println("Connected to the postgres database!")*/
+
+	_ = 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)
 
-	_ = orm.RegisterDataBase("data", "mysql", utils.MYSQL_URL_DATA)
+	//_ = orm.RegisterDataBase("data", "mysql", utils.MYSQL_URL_DATA)
+	_ = orm.RegisterDataBase("data", utils.DBDRIVER_NAME, "user=etadb dbname=eta_index sslmode=disable password=8lAwq4qYbrC7 host=106.14.64.48 port=15400")
 	orm.SetMaxIdleConns("data", 50)
 	orm.SetMaxOpenConns("data", 100)
 

utils/config.go

@@ -17,6 +17,7 @@ var (
 	MYSQL_URL      string //数据库连接
 	MYSQL_URL_DATA string
 	MYSQL_URL_EDB  string
+	DBDRIVER_NAME  string //数据库驱动
 	MgoUrlData     string // mongodb数据库连接配置
 
 	REDIS_CACHE string //缓存地址
@@ -103,7 +104,10 @@ func init() {
 	Md5Key = config["md5_key"]
 
 	MYSQL_URL = config["mysql_url"]
-
+	DBDRIVER_NAME = config["db_driver_name"]
+	if DBDRIVER_NAME == "" {
+		DBDRIVER_NAME = "mysql"
+	}
 	MYSQL_URL_DATA = config["mysql_url_data"]
 	MYSQL_URL_EDB = config["mysql_url_edb"]