eta_api/models/data_manage/supply_analysis/variety_edb_info.go

package supply_analysis

import (
	"database/sql"
	"eta/eta_api/global"
	"eta/eta_api/utils"
	"time"

	"gorm.io/gorm"
)

// VarietyEdbInfo variety_edb_info 品种指标表
type VarietyEdbInfo struct {
	VarietyEdbId int       `orm:"column(variety_edb_id);pk" gorm:"primaryKey" `
	VarietyId    int       `description:"品种id"`
	EdbName      string    `description:"品种指标名称"`
	Frequency    string    `description:"频度"`
	StartDate    time.Time `description:"开始日期"`
	EndDate      time.Time `description:"结束日期"`
	LatestValue  float64   `description:"最新值"`
	EdbInfoId    int       `description:"指标id"`
	EdbCode      string    `description:"品种编码"`
	Source       int       `description:"来源，1：影响周度产量；2：周度产量变动；3：影响月度产量；4：月度产量变动"`
	ModifyTime   time.Time `description:"更新时间"`
	CreateTime   time.Time `description:"添加时间"`
}

// BatchCreateVarietyEdbInfo 批量插入
func BatchCreateVarietyEdbInfo(list []*VarietyEdbInfo) (err error) {
	num := len(list)
	if num <= 0 {
		return
	}
	err = global.DbMap[utils.DbNameIndex].CreateInBatches(list, num).Error

	return
}

// GetCountVarietyEdbByVarietyId 根据品种id获取添加到已经生成的指标数量（不管是否加入到ETA指标库）
func GetCountVarietyEdbByVarietyId(varietyId int) (total int, err error) {
	var totalNull sql.NullInt64
	sql := `SELECT count(1) total FROM variety_edb_info AS a WHERE a.variety_id = ?`
	err = global.DbMap[utils.DbNameIndex].Raw(sql, varietyId).Scan(&totalNull).Error
	if !totalNull.Valid {
		total = 0
	} else {
		total = int(totalNull.Int64)
	}
	return
}

// GetVarietyEdbListByVarietyId 根据品种id获取添加到已经生成的指标（不管是否加入到ETA指标库）
func GetVarietyEdbListByVarietyId(varietyId int) (items []*VarietyEdbInfo, err error) {
	sql := `SELECT * FROM variety_edb_info AS a WHERE a.variety_id = ?`
	err = global.DbMap[utils.DbNameIndex].Raw(sql, varietyId).Scan(&items).Error

	return
}

// GetCountVarietyEdbInfoByVarietyId 根据品种id获取添加到ETA指标库的指标数量
func GetCountVarietyEdbInfoByVarietyId(varietyId int) (total int, err error) {
	sql := `SELECT count(1) total FROM variety_edb_info AS a
			JOIN edb_info b on a.edb_info_id=b.edb_info_id
			WHERE a.variety_id = ? and a.edb_info_id >0`
	err = global.DbMap[utils.DbNameIndex].Raw(sql, varietyId).Scan(&total).Error

	return
}

// VarietyEdbInfoItem 品种指标表（展示使用）
type VarietyEdbInfoItem struct {
	VarietyEdbId int                  `orm:"column(variety_edb_id);pk" gorm:"primaryKey" `
	VarietyId    int                  `description:"品种id"`
	EdbName      string               `description:"品种指标名称"`
	Frequency    string               `description:"频度"`
	StartDate    string               `description:"开始日期"`
	EndDate      string               `description:"结束日期"`
	LatestValue  float64              `description:"最新值"`
	EdbInfoId    int                  `description:"指标id"`
	UniqueCode   string               `description:"唯一code"`
	EdbCode      string               `description:"品种编码"`
	ClassifyId   int                  `description:"上级分类id"`
	ClassifyName string               `description:"分类名称"`
	Source       int                  `description:"来源，1：影响周度产量；2：周度产量变动；3：影响月度产量；4：月度产量变动" json:"-"`
	ModifyTime   string               `description:"最近一次更新时间"`
	CreateTime   string               `description:"添加时间"`
	Button       VarietyEdbInfoButton `description:"操作按钮权限"gorm:"-"`
}

func (v *VarietyEdbInfoItem) AfterFind(tx *gorm.DB) (err error) {
	if utils.NeedDateOrTimeFormat(utils.DbDriverName) {
		if v.CreateTime != "" {
			v.CreateTime = utils.GormDateStrToDateTimeStr(v.CreateTime)
		}
		if v.ModifyTime != "" {
			v.ModifyTime = utils.GormDateStrToDateTimeStr(v.ModifyTime)
		}
		if v.StartDate != "" {
			v.StartDate = utils.GormDateStrToDateStr(v.StartDate)
		}
		if v.EndDate != "" {
			v.EndDate = utils.GormDateStrToDateStr(v.EndDate)
		}
	}
	return
}

type VarietyEdbInfoButton struct {
	Copy bool `description:"复制数据权限"`
	Show bool `description:"查看数据权限"`
	Add  bool `description:"添加到指标库"`
}

// GetAllVarietyEdbInfoByVarietyId 根据品种id获取所有的指标
func GetAllVarietyEdbInfoByVarietyId(varietyId int) (items []*VarietyEdbInfoItem, err error) {
	sql := `SELECT * FROM variety_edb_info AS a WHERE a.variety_id = ? ORDER BY a.variety_edb_id ASC `
	err = global.DbMap[utils.DbNameIndex].Raw(sql, varietyId).Find(&items).Error

	return
}

// GetVarietyEdbInfoByVarietyEdbId 根据品种指标id获取的指标
func GetVarietyEdbInfoByVarietyEdbId(varietyId int) (item *VarietyEdbInfo, err error) {
	sql := `SELECT * FROM variety_edb_info AS a WHERE a.variety_edb_id = ? ORDER BY a.variety_edb_id ASC `
	err = global.DbMap[utils.DbNameIndex].Raw(sql, varietyId).First(&item).Error

	return
}

// Update 更基础信息
func (varietyEdbInfo *VarietyEdbInfo) Update(cols []string) (err error) {
	err = global.DbMap[utils.DbNameIndex].Select(cols).Updates(varietyEdbInfo).Error
	return
}