Browse Source

change gorm

Roc 1 month ago
parent
commit
bb89993a15

+ 1 - 1
controllers/base_from_sci99.go

@@ -108,7 +108,7 @@ func (this *Sci99Controller) Refresh() {
 	err = models.RefreshEdbDataFromSci99(req.EdbInfoId, req.EdbCode, req.StartDate)
 	if err != nil && err.Error() != utils.ErrNoRow() {
 		br.Msg = "刷新指标信息失败!"
-		br.ErrMsg = "刷新指标信息失败 RefreshEdbDataFromIcpi,Err:" + err.Error()
+		br.ErrMsg = "刷新指标信息失败 RefreshEdbDataFromSci99,Err:" + err.Error()
 		return
 	}
 	// 更新指标最大最小值

+ 8 - 0
models/base_from_icpi.go

@@ -4,6 +4,7 @@ import (
 	"eta/eta_index_lib/global"
 	"eta/eta_index_lib/utils"
 	"fmt"
+	"gorm.io/gorm"
 	"strconv"
 	"strings"
 	"time"
@@ -34,6 +35,13 @@ type BaseFromIcpiData struct {
 	ModifyTime          time.Time `description:"修改时间"`
 }
 
+// AfterFind 在该模型上设置钩子函数,把日期转成正确的string,所以查询函数只能用Find函数,First或者Scan是不会触发该函数的来获取数据
+func (m *BaseFromIcpiData) AfterFind(db *gorm.DB) (err error) {
+	m.DataTime = utils.GormDateStrToDateStr(m.DataTime)
+
+	return
+}
+
 type BaseFromIcpiClassify struct {
 	BaseFromIcpiClassifyId int `gorm:"column:base_from_icpi_classify_id;primaryKey"`
 	//BaseFromIcpiClassifyId int       `orm:"column(base_from_icpi_classify_id);pk"`

+ 1 - 0
models/base_from_sci99.go

@@ -76,6 +76,7 @@ func GetBaseFromSci99DataDataByCondition(condition string, pars []interface{}) (
 	sql += ` ORDER BY data_time DESC `
 	//_, err = o.Raw(sql, pars).QueryRows(&item)
 	err = global.DEFAULT_DB.Raw(sql, pars...).Find(&item).Error
+
 	return
 }
 

+ 10 - 10
models/edb_terminal.go

@@ -9,16 +9,16 @@ import (
 // EdbTerminal 指标终端
 type EdbTerminal struct {
 	TerminalId   int       `gorm:"column:terminal_id;type:int(9) UNSIGNED;primaryKey;not null;"`
-	Source       int       `gorm:"column:source;type:int(9) UNSIGNED;comment:指标来源类型;default:0;"`                    // 指标来源类型
-	Name         string    `gorm:"column:name;type:varchar(255);comment:终端别名;"`                                       // 终端别名
-	TerminalCode string    `gorm:"column:terminal_code;type:varchar(255);comment:终端编码,用于配置在机器上;"`             // 终端编码,用于配置在机器上
-	ServerUrl    string    `gorm:"column:server_url;type:varchar(255);comment:终端地址;"`                                 // 终端地址
-	Num          int       `gorm:"column:num;type:int(9) UNSIGNED;comment:终端最大指标数;default:0;"`                     // 终端最大指标数
+	Source       int       `gorm:"column:source;type:int(9) UNSIGNED;comment:指标来源类型;default:0;"`                  // 指标来源类型
+	Name         string    `gorm:"column:name;type:varchar(255);comment:终端别名;"`                                   // 终端别名
+	TerminalCode string    `gorm:"column:terminal_code;type:varchar(255);comment:终端编码,用于配置在机器上;"`                 // 终端编码,用于配置在机器上
+	ServerUrl    string    `gorm:"column:server_url;type:varchar(255);comment:终端地址;"`                             // 终端地址
+	Num          int       `gorm:"column:num;type:int(9) UNSIGNED;comment:终端最大指标数;default:0;"`                    // 终端最大指标数
 	ModifyTime   time.Time `gorm:"column:modify_time;type:timestamp;comment:最近一次修改时间;default:CURRENT_TIMESTAMP;"` // 最近一次修改时间
-	CreateTime   time.Time `gorm:"column:create_time;type:datetime;comment:创建时间;default:NULL;"`                       // 创建时间
-	Status       int       `gorm:"column:status;type:tinyint(4);comment:状态,1启用,2禁用;not null;default:1;"`            // 状态,1启用,2禁用
-	Value        string    `gorm:"column:value;type:text;comment:终端相关的凭证;"`                                        // 终端相关的凭证
-	DirPath      string    `gorm:"column:dir_path;type:varchar(255);comment:终端存放的文件夹路径;default:NULL;"`          // 终端存放的文件夹路径
+	CreateTime   time.Time `gorm:"column:create_time;type:datetime;comment:创建时间;default:NULL;"`                   // 创建时间
+	Status       int       `gorm:"column:status;type:tinyint(4);comment:状态,1启用,2禁用;not null;default:1;"`          // 状态,1启用,2禁用
+	Value        string    `gorm:"column:value;type:text;comment:终端相关的凭证;"`                                       // 终端相关的凭证
+	DirPath      string    `gorm:"column:dir_path;type:varchar(255);comment:终端存放的文件夹路径;default:NULL;"`            // 终端存放的文件夹路径
 }
 
 // GetEdbTerminalListBySource 根据指标来源类型获取所有的终端列表
@@ -52,7 +52,7 @@ func GetEdbTerminalByCode(terminalCode string) (item *EdbTerminal, err error) {
 
 // GetEdbCountGroupByTerminal 获取终端code分组总数
 func GetEdbCountGroupByTerminal(source int) (list []TerminalCodeCountGroup, err error) {
-	sql := `select terminal_code,count(1) total from edb_info where source = ? AND no_update=0 AND terminal_code != "" group by terminal_code; `
+	sql := `select terminal_code,count(1) total from edb_info where source = ? AND no_update=0 AND terminal_code != '' group by terminal_code; `
 	err = global.DEFAULT_DB.Raw(sql, source).Find(&list).Error
 
 	return

+ 1 - 1
models/predict_edb_conf.go

@@ -41,7 +41,7 @@ type EditPredictEdbInfoReq struct {
 }
 
 type PredictEdbConf struct {
-	ConfigId         int       `orm:"column(config_id);pk" description:"规则id"`
+	ConfigId         int       `gorm:"primaryKey;autoIncrement;column:config_id" description:"规则id"`
 	PredictEdbInfoId int       `orm:"column(predict_edb_info_id)" description:"预测指标id"`
 	SourceEdbInfoId  int       `description:"来源指标id"`
 	RuleType         int       `description:"预测规则,1:最新,2:固定值,3:同比,4:同差,5:环比,6:环差,7:N期移动均值,8:N期段线性外推值"`