Browse Source

fix:dm_db

more 6 months ago
parent
commit
1f33c9a643
4 changed files with 63 additions and 60 deletions
  1. 2 0
      main.go
  2. 51 58
      models/db.go
  3. 10 0
      models/dm_base.go
  4. 0 2
      utils/config.go

+ 2 - 0
main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"eta_gn/eta_chart_lib/controllers"
+	"eta_gn/eta_chart_lib/models"
 	_ "eta_gn/eta_chart_lib/routers"
 	"eta_gn/eta_chart_lib/services/alarm_msg"
 	"eta_gn/eta_chart_lib/utils"
@@ -20,6 +21,7 @@ func main() {
 		beego.BConfig.WebConfig.StaticDir["/swagger"] = "swagger"
 	}
 
+	models.AfterInitTable()
 	// 异常处理
 	beego.ErrorController(&controllers.ErrorController{})
 

+ 51 - 58
models/db.go

@@ -1,71 +1,64 @@
 package models
 
-import (
-	"eta_gn/eta_chart_lib/models/data_manage"
-	"eta_gn/eta_chart_lib/models/data_manage/future_good"
-	"eta_gn/eta_chart_lib/utils"
-	"time"
+// import (
+// 	_ "github.com/go-sql-driver/mysql"
+// )
 
-	_ "github.com/go-sql-driver/mysql"
+// func init() {
 
-	"github.com/beego/beego/v2/client/orm"
-)
+// 	_ = orm.RegisterDataBase("default", "mysql", utils.MYSQL_URL)
+// 	orm.SetMaxIdleConns("default", 50)
+// 	orm.SetMaxOpenConns("default", 100)
 
-func init() {
+// 	db, _ := orm.GetDB("default")
+// 	db.SetConnMaxLifetime(10 * time.Minute)
 
-	_ = orm.RegisterDataBase("default", "mysql", utils.MYSQL_URL)
-	orm.SetMaxIdleConns("default", 50)
-	orm.SetMaxOpenConns("default", 100)
+// 	_ = orm.RegisterDataBase("data", "mysql", utils.MYSQL_URL_DATA)
+// 	orm.SetMaxIdleConns("data", 50)
+// 	orm.SetMaxOpenConns("data", 100)
 
-	db, _ := orm.GetDB("default")
-	db.SetConnMaxLifetime(10 * time.Minute)
+// 	data_db, _ := orm.GetDB("data")
+// 	data_db.SetConnMaxLifetime(10 * time.Minute)
 
-	_ = orm.RegisterDataBase("data", "mysql", utils.MYSQL_URL_DATA)
-	orm.SetMaxIdleConns("data", 50)
-	orm.SetMaxOpenConns("data", 100)
+// 	orm.Debug = true
+// 	orm.DebugLog = orm.NewLog(utils.Binlog)
 
-	data_db, _ := orm.GetDB("data")
-	data_db.SetConnMaxLifetime(10 * time.Minute)
+// 	//注册对象
+// 	orm.RegisterModel(
+// 		new(data_manage.EdbInfo),
+// 		new(ShareChartRefreshLog),                       //分享图表刷新日志表
+// 		new(ExcelInfo),                                  //excel表格
+// 		new(data_manage.PredictEdbConf),                 //预测指标配置
+// 		new(data_manage.PredictEdbRuleData),             //预测指标配置生成的数据
+// 		new(data_manage.PredictEdbConfCalculateMapping), //预测指标关系表
+// 		new(data_manage.ChartInfoCorrelation),           // 图表相关性信息
+// 		new(FactorEdbSeriesCalculateDataQjjs),           // 因子指标系列-区间计算数据
+// 	)
+// 	// 期货数据库
+// 	initFutureGood()
 
-	orm.Debug = true
-	orm.DebugLog = orm.NewLog(utils.Binlog)
+// 	// 初始化部分数据表变量(直接init会有顺序问题=_=!)
+// 	afterInitTable()
+// }
 
-	//注册对象
-	orm.RegisterModel(
-		new(data_manage.EdbInfo),
-		new(ShareChartRefreshLog),                       //分享图表刷新日志表
-		new(ExcelInfo),                                  //excel表格
-		new(data_manage.PredictEdbConf),                 //预测指标配置
-		new(data_manage.PredictEdbRuleData),             //预测指标配置生成的数据
-		new(data_manage.PredictEdbConfCalculateMapping), //预测指标关系表
-		new(data_manage.ChartInfoCorrelation),           // 图表相关性信息
-		new(FactorEdbSeriesCalculateDataQjjs),           // 因子指标系列-区间计算数据
-	)
-	// 期货数据库
-	initFutureGood()
+// // initFutureGood 注册期货数据 数据表
+// func initFutureGood() {
+// 	//注册对象
+// 	orm.RegisterModel(
+// 		new(future_good.FutureGoodEdbInfo),         //期货指标表
+// 		new(future_good.FutureGoodEdbData),         //期货指标数据表
+// 		new(future_good.ChartInfoFutureGoodProfit), //期货利润图的扩展表
+// 	)
+// }
 
-	// 初始化部分数据表变量(直接init会有顺序问题=_=!)
-	afterInitTable()
-}
+// // afterInitTable
+// // @Description: 初始化表结构的的后置操作
+// // @author: Roc
+// // @datetime 2024-07-01 13:31:09
+// func afterInitTable() {
+// 	// 初始化指标来源配置
+// 	data_manage.InitEdbSourceVar()
 
-// initFutureGood 注册期货数据 数据表
-func initFutureGood() {
-	//注册对象
-	orm.RegisterModel(
-		new(future_good.FutureGoodEdbInfo),         //期货指标表
-		new(future_good.FutureGoodEdbData),         //期货指标数据表
-		new(future_good.ChartInfoFutureGoodProfit), //期货利润图的扩展表
-	)
-}
-
-// afterInitTable
-// @Description: 初始化表结构的的后置操作
-// @author: Roc
-// @datetime 2024-07-01 13:31:09
-func afterInitTable() {
-	// 初始化指标来源配置
-	data_manage.InitEdbSourceVar()
-
-	// 初始化是否启用mongo配置
-	InitUseMongoConf()
-}
+// 	// 初始化是否启用mongo配置
+// 	InitUseMongoConf()
+// }

+ 10 - 0
models/dm_base.go

@@ -1,6 +1,16 @@
 package models
 
+import "eta_gn/eta_chart_lib/models/data_manage"
+
 // 达梦数据库初始化
 func AfterInitTable() {
 	afterInitTable()
 }
+
+func afterInitTable() {
+	// 初始化指标来源配置
+	data_manage.InitEdbSourceVar()
+
+	// 初始化是否启用mongo配置
+	InitUseMongoConf()
+}

+ 0 - 2
utils/config.go

@@ -10,7 +10,6 @@ import (
 	"github.com/beego/beego/v2/server/web"
 	"github.com/qiniu/qmgo"
 	"github.com/shopspring/decimal"
-	"gorm.io/gorm"
 )
 
 var (
@@ -21,7 +20,6 @@ var (
 	MgoUrlData     string // mongodb数据库连接配置
 	DM_URL         string
 	DM_URL_DATA    string
-	DM_MODEL       map[string]*gorm.DB //数据库连接配置
 
 	REDIS_CACHE string //缓存地址
 	//Rc          *cache.Cache //redis缓存