eta_api/models/data_manage/supply_analysis/variety.go

package supply_analysis

import (
	"database/sql"
	"eta/eta_api/global"
	"eta/eta_api/utils"
	"fmt"
	"gorm.io/gorm"
	"time"
)

// Variety variety 品种表
type Variety struct {
	VarietyId                 int       `orm:"column(variety_id);pk" gorm:"primaryKey" `
	VarietyName               string    `description:"品种名称"`
	LastUpdateSysUserId       int       `description:"最后更新人id"`
	LastUpdateSysUserRealName string    `description:"最后更新人名称"`
	ProductionDay             int       `description:"生产天数"`
	SysUserId                 int       `description:"创建人id"`
	SysUserRealName           string    `description:"创建人姓名"`
	ModifyTime                time.Time `description:"修改时间"`
	CreateTime                time.Time `description:"创建时间"`
}

// GetVarietyById 根据品种id获取品种详情
func GetVarietyById(id int) (item *Variety, err error) {
	sql := `SELECT * FROM variety WHERE variety_id = ?`
	err = global.DbMap[utils.DbNameIndex].Raw(sql, id).First(&item).Error

	return
}

// GetVarietyByName 根据品种名称获取品种详情
func GetVarietyByName(name string) (item *Variety, err error) {
	sql := `SELECT * FROM variety WHERE variety_name = ?`
	err = global.DbMap[utils.DbNameIndex].Raw(sql, name).First(&item).Error
	return
}

// AddVariety 添加品种
func AddVariety(item *Variety) (lastId int64, err error) {
	err = global.DbMap[utils.DbNameIndex].Create(item).Error
	return
}

// CreateVariety 添加品种
func CreateVariety(item *Variety, adminIdList []int) (err error) {
	to := global.DbMap[utils.DbNameIndex].Begin()

	defer func() {
		if err != nil {
			_ = to.Rollback()
		} else {
			_ = to.Commit()
		}
	}()
	err = to.Create(item).Error
	if err != nil {
		return
	}

	varietyAdminPermissionList := make([]*VarietyAdminPermission, 0)
	for _, adminId := range adminIdList {
		varietyAdminPermissionList = append(varietyAdminPermissionList, &VarietyAdminPermission{
			//Id:              0,
			VarietyId:  item.VarietyId,
			SysUserId:  adminId,
			CreateTime: time.Now(),
		})
	}
	if len(varietyAdminPermissionList) > 0 {
		err = to.CreateInBatches(varietyAdminPermissionList, utils.MultiAddNum).Error
	}

	return
}

// EditVariety 编辑品种
func EditVariety(item *Variety, adminIdList []int) (err error) {
	to := global.DbMap[utils.DbNameIndex].Begin()

	defer func() {
		if err != nil {
			_ = to.Rollback()
		} else {
			_ = to.Commit()
		}
	}()

	err = to.Select("VarietyName", "LastUpdateSysUserId", "LastUpdateSysUserRealName", "ModifyTime").Updates(item).Error
	if err != nil {
		return
	}

	// 删除历史的权限配置
	sql := `DELETE FROM variety_admin_permission where variety_id = ? `
	err = to.Exec(sql, item.VarietyId).Error
	if err != nil {
		return
	}

	// 添加新的权限配置
	varietyAdminPermissionList := make([]*VarietyAdminPermission, 0)
	for _, adminId := range adminIdList {
		varietyAdminPermissionList = append(varietyAdminPermissionList, &VarietyAdminPermission{
			//Id:              0,
			VarietyId:  item.VarietyId,
			SysUserId:  adminId,
			CreateTime: time.Now(),
		})
	}
	if len(varietyAdminPermissionList) > 0 {
		err = to.CreateInBatches(varietyAdminPermissionList, utils.MultiAddNum).Error
	}

	return
}

// VarietyItem 列表页返回的数据结构
type VarietyItem struct {
	VarietyId                 int           `orm:"column(variety_id);pk" gorm:"primaryKey" `
	VarietyName               string        `description:"品种名称"`
	ProductionDay             int           `description:"生产天数"`
	LastUpdateSysUserId       int           `description:"最后更新人id"`
	LastUpdateSysUserRealName string        `description:"最后更新人名称"`
	PermissionUserId          string        `description:"有操作权限的用户id"`
	ModifyTime                string        `description:"修改时间"`
	CreateTime                string        `description:"创建时间"`
	Button                    VarietyButton `gorm:"-" description:"操作按钮权限"`
}

func (m *VarietyItem) AfterFind(db *gorm.DB) (err error) {
	m.CreateTime = utils.GormDateStrToDateTimeStr(m.CreateTime)
	m.ModifyTime = utils.GormDateStrToDateTimeStr(m.ModifyTime)
	return
}

type VarietyButton struct {
	Edit    bool `description:"操作权限"`
	Delete  bool `description:"删除权限"`
	Analyse bool `description:"分析权限"`
}

// GetListBySuperAdminPage 不区分是否有分析权限的获取分页数据
func (item Variety) GetListBySuperAdminPage(condition string, pars []interface{}, startSize, pageSize int) (total int, items []*VarietyItem, err error) {
	//baseSql := ` FROM ( SELECT a.*, GROUP_CONCAT(DISTINCT b.sys_user_id ORDER BY b.sys_user_id ASC SEPARATOR ',') AS permission_user_id FROM variety a
	//			LEFT JOIN variety_admin_permission b on a.variety_id=b.variety_id
	//			LEFT JOIN variety_edb_info c on a.variety_id=c.variety_id WHERE 1=1 `

	baseSql := fmt.Sprintf(` FROM ( SELECT a.*, 
%s AS permission_user_id FROM variety a 
				LEFT JOIN variety_admin_permission b on a.variety_id=b.variety_id 
				LEFT JOIN variety_edb_info c on a.variety_id=c.variety_id WHERE 1=1 `, utils.GroupUnitFunc(utils.DbDriverName, "b.sys_user_id", ",", true, "b.sys_user_id", "ASC", true))

	if condition != "" {
		baseSql += condition
	}
	baseSql += ` GROUP BY a.variety_id ) d `
	// 数据总数
	totalSql := `SELECT COUNT(1) total ` + baseSql
	var totalNull sql.NullInt64
	err = global.DbMap[utils.DbNameIndex].Raw(totalSql, pars...).Scan(&totalNull).Error
	if err != nil {
		return
	}

	if !totalNull.Valid {
		total = 0
	} else {
		total = int(totalNull.Int64)
	}

	// 列表页数据
	listSql := `SELECT * ` + baseSql + ` ORDER BY modify_time DESC,variety_id DESC LIMIT ?,?`
	pars = append(pars, startSize, pageSize)
	err = global.DbMap[utils.DbNameIndex].Raw(listSql, pars...).Find(&items).Error
	return
}

// GetListByPage 获取分页数据
func (item Variety) GetListByPage(condition string, pars []interface{}, startSize, pageSize int) (total int, items []*VarietyItem, err error) {
	//baseSql := ` FROM ( SELECT a.*, GROUP_CONCAT(DISTINCT b.sys_user_id ORDER BY b.sys_user_id ASC SEPARATOR ',') AS permission_user_id FROM variety a
	//			JOIN variety_admin_permission b on a.variety_id=b.variety_id
	//			JOIN variety_edb_info c on a.variety_id=c.variety_id WHERE 1=1 `

	baseSql := fmt.Sprintf(` FROM ( SELECT a.*, %s AS permission_user_id FROM variety a 
				JOIN variety_admin_permission b on a.variety_id=b.variety_id 
				JOIN variety_edb_info c on a.variety_id=c.variety_id WHERE 1=1 `, utils.GroupUnitFunc(utils.DbDriverName, "b.sys_user_id", ",", true, "b.sys_user_id", "ASC", true))

	if condition != "" {
		baseSql += condition
	}
	baseSql += ` GROUP BY a.variety_id ) d `
	// 数据总数
	totalSql := `SELECT COUNT(1) total ` + baseSql
	var totalNull sql.NullInt64
	err = global.DbMap[utils.DbNameIndex].Raw(totalSql, pars...).Scan(&totalNull).Error
	if err != nil {
		return
	}
	if !totalNull.Valid {
		total = 0
	} else {
		total = int(totalNull.Int64)
	}
	// 列表页数据
	listSql := `SELECT * ` + baseSql + ` ORDER BY modify_time DESC,variety_id DESC LIMIT ?,?`
	pars = append(pars, startSize, pageSize)
	err = global.DbMap[utils.DbNameIndex].Raw(listSql, pars...).Find(&items).Error
	return
}

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

// Delete 删除
func (variety *Variety) Delete() (err error) {
	sql := ` DELETE FROM variety  WHERE variety_id = ? `
	err = global.DbMap[utils.DbNameIndex].Exec(sql, variety.VarietyId).Error
	return
}

// GetVarietyMaxSort 获取图表分类下最大的排序数
func GetVarietyMaxSort(parentId int) (sort int, err error) {
	var sortNull sql.NullInt64
	sql := `SELECT Max(sort) AS sort FROM future_good_chart_classify WHERE parent_id=? AND is_delete=0 `
	err = global.DbMap[utils.DbNameIndex].Raw(sql, parentId).Scan(&sortNull).Error
	if !sortNull.Valid {
		sort = 0
	} else {
		sort = int(sortNull.Int64)
	}
	return
}

type VarietyView struct {
	VarietyId   int    `orm:"column(future_good_chart_classify_id);pk"`
	VarietyName string `description:"分类名称"`
	ParentId    int    `description:"父级id"`
}

func GetVarietyViewById(classifyId int) (item *VarietyView, err error) {
	sql := `SELECT * FROM future_good_chart_classify WHERE future_good_chart_classify_id=? AND is_delete=0 `
	err = global.DbMap[utils.DbNameIndex].Raw(sql, classifyId).First(&item).Error
	return
}