package supply_analysis import ( "eta/eta_api/global" "eta/eta_api/utils" "gorm.io/gorm" "time" ) // VarietyPlant variety_plant 品种装置表 type VarietyPlant struct { VarietyPlantId int `orm:"column(variety_plant_id);pk" gorm:"primaryKey" ` VarietyId int `description:"品种id"` Province string `description:"所属省份"` City string `description:"所属城市"` FactoryName string `description:"工厂名称"` PlantName string `description:"装置/产线名称"` MaintenanceDate time.Time `description:"检修日期"` ResumptionDate time.Time `description:"复产日期"` AnnualCapacity float64 `description:"年产能"` Coefficient float64 `description:"降负系数"` AverageDailyCapacityReduction float64 `description:"日均产量减量"` IsStop int `description:"是否停产，0：未停产；1：停产；默认未停产"` Sort int `description:"排序字段，越小越靠前"` SysUserId int `description:"添加人id"` SysUserRealName string `description:"添加人真实姓名"` ModifyTime time.Time `description:"最近一次更新时间"` CreateTime time.Time `description:"添加时间"` } // AddVarietyPlant 添加variety_plant装置 func AddVarietyPlant(item *VarietyPlant) (err error) { err = global.DbMap[utils.DbNameIndex].Create(item).Error return } // MultiAddVarietyPlant 批量添加variety_plant装置 func MultiAddVarietyPlant(items []*VarietyPlant) (err error) { err = global.DbMap[utils.DbNameIndex].CreateInBatches(items, utils.MultiAddNum).Error return } // Update 更新基础信息 func (varietyPlant *VarietyPlant) Update(cols []string) (err error) { err = global.DbMap[utils.DbNameIndex].Select(cols).Updates(varietyPlant).Error return } // GetVarietyPlantById 根据品种装置id获取品种装置详情 func GetVarietyPlantById(id int) (item *VarietyPlant, err error) { sql := `SELECT * FROM variety_plant WHERE variety_plant_id = ?` err = global.DbMap[utils.DbNameIndex].Raw(sql, id).First(&item).Error return } // GetVarietyPlantByIdList 根据品种装置id列表获取品种装置详情列表 func GetVarietyPlantByIdList(idList []int) (items []*VarietyPlant, err error) { num := len(idList) if num <= 0 { return } sql := `SELECT * FROM variety_plant WHERE variety_plant_id in (` + utils.GetOrmInReplace(num) + `)` err = global.DbMap[utils.DbNameIndex].Raw(sql, idList).Scan(&items).Error return } // VarietyPlantItem 品种装置数据（展示使用） type VarietyPlantItem struct { VarietyPlantId int `orm:"column(variety_plant_id);pk" gorm:"primaryKey" ` VarietyId int `description:"品种id"` Province string `description:"所属省份"` City string `description:"所属城市"` FactoryName string `description:"工厂名称"` PlantName string `description:"装置/产线名称"` MaintenanceDate string `description:"检修日期"` ResumptionDate string `description:"复产日期"` AnnualCapacity float64 `description:"年产能"` Coefficient float64 `description:"降负系数"` AverageDailyCapacityReduction float64 `description:"日均产量减量"` Day int `description:"时长"` CapacityReduction float64 `description:"期间减量"` IsStop int `description:"是否停产，0：未停产；1：停产；默认未停产"` Sort int `description:"排序字段，越小越靠前"` SysUserId int `description:"添加人id"` SysUserRealName string `description:"添加人真实姓名"` ModifyTime string `description:"最近一次更新时间"` CreateTime string `description:"添加时间"` Button VarietyPlantButton `gorm:"-" description:"操作按钮权限"` } func (v *VarietyPlantItem) 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.MaintenanceDate != "" { v.MaintenanceDate = utils.GormDateStrToDateStr(v.MaintenanceDate) } if v.ResumptionDate != "" { v.ResumptionDate = utils.GormDateStrToDateStr(v.ResumptionDate) } } return } type VarietyPlantButton struct { Edit bool `description:"操作权限"` Delete bool `description:"删除权限"` } // GetAllVarietyPlantByVarietyId 根据品种id获取所有的装置 func GetAllVarietyPlantByVarietyId(varietyId int) (items []*VarietyPlantItem, err error) { sql := `SELECT * FROM variety_plant a WHERE variety_id = ? ORDER BY sort ASC, variety_plant_id ASC ` err = global.DbMap[utils.DbNameIndex].Raw(sql, varietyId).Scan(&items).Error return } // GetCountVarietyPlantByVarietyId 根据品种id获取所有的装置数量 func GetCountVarietyPlantByVarietyId(varietyId int) (total int, err error) { sql := `SELECT count(1) total FROM variety_plant a WHERE variety_id = ? ` err = global.DbMap[utils.DbNameIndex].Raw(sql, varietyId).Scan(&total).Error return } // DeleteVarietyPlantById 根据装置id删除 func DeleteVarietyPlantById(id int) (err error) { sql := `DELETE FROM variety_plant WHERE variety_plant_id = ?` err = global.DbMap[utils.DbNameIndex].Exec(sql, id).Error return } // UpdateEdbInfoSortByPrevId 根据上一层级id更新排序 func UpdateEdbInfoSortByPrevId(varietyId, prevVarietyId, nowSort int, updateSort string) (err error) { sql := `UPDATE variety_plant set sort = ` + updateSort + ` WHERE variety_id=? AND ` if prevVarietyId > 0 { sql += ` ( sort > ? or ( sort=? AND variety_plant_id > ?)) ` } err = global.DbMap[utils.DbNameIndex].Exec(sql, varietyId, nowSort, nowSort, prevVarietyId).Error return } // UpdateEdbInfoSortByNextId 根据下一层级id更新排序 func UpdateEdbInfoSortByNextId(varietyId, prevVarietyId, nowSort int, updateSort string) (err error) { sql := `UPDATE variety_plant set sort = ` + updateSort + ` WHERE variety_id=? AND ` if prevVarietyId > 0 { sql += ` ( sort < ? or ( sort=? AND variety_plant_id < ? )) ` } err = global.DbMap[utils.DbNameIndex].Exec(sql, varietyId, nowSort, nowSort, prevVarietyId).Error return }