分类树查询修复

models/data_manage/base_from_oilchem.go

@@ -3,6 +3,7 @@ package data_manage
 import (
 	"eta/eta_api/global"
 	"eta/eta_api/utils"
+	"gorm.io/gorm"
 
 	"github.com/rdlucklib/rdluck_tools/paging"
 )
@@ -30,6 +31,18 @@ type BaseFromOilchemIndex struct {
 	ModifyTime             string // 修改时间
 }
 
+func (baseFromOilchemIndex *BaseFromOilchemIndex) AfterFind(db *gorm.DB) (err error) {
+	if utils.NeedDateOrTimeFormat(utils.DbDriverName) {
+		if baseFromOilchemIndex.ModifyTime != "" {
+			baseFromOilchemIndex.ModifyTime = utils.GormDateStrToDateTimeStr(baseFromOilchemIndex.ModifyTime)
+		}
+		if baseFromOilchemIndex.CreateTime != "" {
+			baseFromOilchemIndex.CreateTime = utils.GormDateStrToDateTimeStr(baseFromOilchemIndex.CreateTime)
+		}
+	}
+	return
+}
+
 type BaseFromOilchemData struct {
 	BaseFromOilchemDataId  int    // 数据表ID
 	BaseFromOilchemIndexId int    // 指标ID
@@ -40,6 +53,20 @@ type BaseFromOilchemData struct {
 	ModifyTime             string
 }
 
+func (baseFromOilchemData *BaseFromOilchemData) AfterFind(db *gorm.DB) (err error) {
+	if utils.NeedDateOrTimeFormat(utils.DbDriverName) {
+		if baseFromOilchemData.ModifyTime != "" {
+			baseFromOilchemData.ModifyTime = utils.GormDateStrToDateTimeStr(baseFromOilchemData.ModifyTime)
+		}
+		if baseFromOilchemData.CreateTime != "" {
+			baseFromOilchemData.CreateTime = utils.GormDateStrToDateTimeStr(baseFromOilchemData.CreateTime)
+		}
+		if baseFromOilchemData.DataTime != "" {
+			baseFromOilchemData.DataTime = utils.GormDateStrToDateTimeStr(baseFromOilchemData.DataTime)
+		}
+	}
+	return
+}
 func GetOilchemClassifyList() (list []*BaseFromOilchemClassify, err error) {
 	o := global.DbMap[utils.DbNameIndex]
 	sql := "SELECT * FROM base_from_oilchem_classify ORDER BY sort ASC"
@@ -200,6 +227,24 @@ type BaseFromOilchemIndexView struct {
 	Value                  string `description:"值"`
 }
 
+func (baseFromOilchemIndexView *BaseFromOilchemIndexView) AfterFind(db *gorm.DB) (err error) {
+	if utils.NeedDateOrTimeFormat(utils.DbDriverName) {
+		if baseFromOilchemIndexView.DataTime != "" {
+			baseFromOilchemIndexView.DataTime = utils.GormDateStrToDateStr(baseFromOilchemIndexView.DataTime)
+		}
+		if baseFromOilchemIndexView.StartDate != "" {
+			baseFromOilchemIndexView.StartDate = utils.GormDateStrToDateStr(baseFromOilchemIndexView.StartDate)
+		}
+		if baseFromOilchemIndexView.EndDate != "" {
+			baseFromOilchemIndexView.EndDate = utils.GormDateStrToDateStr(baseFromOilchemIndexView.EndDate)
+		}
+		if baseFromOilchemIndexView.ModifyTime != "" {
+			baseFromOilchemIndexView.ModifyTime = utils.GormDateStrToDateTimeStr(baseFromOilchemIndexView.ModifyTime)
+		}
+	}
+	return
+}
+
 // ExportOilchemExcelReq 导出隆众资讯excel指标
 type ExportOilchemExcelReq struct {
 	KeyWord       string   `description:"关键字, 指标编码或指标ID"`