eta_api/models/assessment_variety.go

package models

import (
	"eta/eta_api/global"
	"eta/eta_api/utils"
	"fmt"
	"github.com/rdlucklib/rdluck_tools/paging"
	"strings"
	"time"
)

const (
	AssessmentVarietyDisabled = 0
	AssessmentVarietyEnabled  = 1
)

// AssessmentVariety 研究员考核品种表
type AssessmentVariety struct {
	AssessmentVarietyId int       `gorm:"column:assessment_variety_id;primaryKey;autoIncrement"`
	VarietyName         string    `description:"品种名称"`
	VarietyCode         string    `description:"品种编码"`
	MonthlyFluctuate    float64   `description:"月度波动阈值"`
	WeeklyFluctuate     float64   `description:"周度波动阈值"`
	Enabled             int       `description:"状态：0-禁用；1-正常"`
	CreateTime          time.Time `description:"创建时间"`
	ModifyTime          time.Time `description:"修改时间"`
}

func (m *AssessmentVariety) TableName() string {
	return "assessment_variety"
}

type AssessmentVarietyCols struct {
	PrimaryId        string
	VarietyName      string
	VarietyCode      string
	MonthlyFluctuate string
	WeeklyFluctuate  string
	Enabled          string
	CreateTime       string
	ModifyTime       string
}

func (m *AssessmentVariety) Cols() AssessmentVarietyCols {
	return AssessmentVarietyCols{
		PrimaryId:        "assessment_variety_id",
		VarietyName:      "variety_name",
		VarietyCode:      "variety_code",
		MonthlyFluctuate: "monthly_fluctuate",
		WeeklyFluctuate:  "weekly_fluctuate",
		Enabled:          "enabled",
		CreateTime:       "create_time",
		ModifyTime:       "modify_time",
	}
}

func (m *AssessmentVariety) Create() (err error) {
	err = global.DEFAULT_DB.Create(m).Error
	return
}

func (m *AssessmentVariety) CreateMulti(items []*AssessmentVariety) (err error) {
	if len(items) == 0 {
		return
	}
	err = global.DEFAULT_DB.CreateInBatches(items, utils.MultiAddNum).Error
	return
}

func (m *AssessmentVariety) Update(cols []string) (err error) {
	err = global.DEFAULT_DB.Select(cols).Updates(m).Error
	return
}

func (m *AssessmentVariety) Remove() (err error) {
	sql := fmt.Sprintf(`DELETE FROM %s WHERE %s = ? LIMIT 1`, m.TableName(), m.Cols().PrimaryId)
	err = global.DEFAULT_DB.Exec(sql, m.AssessmentVarietyId).Error
	return
}

func (m *AssessmentVariety) MultiRemove(ids []int) (err error) {
	if len(ids) == 0 {
		return
	}
	sql := fmt.Sprintf(`DELETE FROM %s WHERE %s IN (%s)`, m.TableName(), m.Cols().PrimaryId, utils.GetOrmInReplace(len(ids)))
	err = global.DEFAULT_DB.Exec(sql, ids).Error
	return
}

func (m *AssessmentVariety) RemoveByCondition(condition string, pars []interface{}) (err error) {
	if condition == "" {
		return
	}
	sql := fmt.Sprintf(`DELETE FROM %s WHERE %s`, m.TableName(), condition)
	err = global.DEFAULT_DB.Exec(sql, pars...).Error
	return
}

func (m *AssessmentVariety) GetItemById(primaryId int) (item *AssessmentVariety, err error) {
	sql := fmt.Sprintf(`SELECT * FROM %s WHERE %s = ? LIMIT 1`, m.TableName(), m.Cols().PrimaryId)
	err = global.DEFAULT_DB.Raw(sql, primaryId).First(&item).Error
	return
}

func (m *AssessmentVariety) GetItemByCondition(condition string, pars []interface{}, orderRule string) (item *AssessmentVariety, err error) {
	order := ``
	if orderRule != "" {
		order = ` ORDER BY ` + orderRule
	}
	sql := fmt.Sprintf(`SELECT * FROM %s WHERE 1=1 %s %s LIMIT 1`, m.TableName(), condition, order)
	err = global.DEFAULT_DB.Raw(sql, pars...).First(&item).Error
	return
}

func (m *AssessmentVariety) GetCountByCondition(condition string, pars []interface{}) (count int, err error) {
	sql := fmt.Sprintf(`SELECT COUNT(1) FROM %s WHERE 1=1 %s`, m.TableName(), condition)
	err = global.DEFAULT_DB.Raw(sql, pars...).Scan(&count).Error
	return
}

func (m *AssessmentVariety) GetItemsByCondition(condition string, pars []interface{}, fieldArr []string, orderRule string) (items []*AssessmentVariety, err error) {
	fields := strings.Join(fieldArr, ",")
	if len(fieldArr) == 0 {
		fields = `*`
	}
	order := fmt.Sprintf(`ORDER BY %s DESC`, m.Cols().CreateTime)
	if orderRule != "" {
		order = ` ORDER BY ` + orderRule
	}
	sql := fmt.Sprintf(`SELECT %s FROM %s WHERE 1=1 %s %s`, fields, m.TableName(), condition, order)
	err = global.DEFAULT_DB.Raw(sql, pars...).Find(&items).Error
	return
}

func (m *AssessmentVariety) GetPageItemsByCondition(condition string, pars []interface{}, fieldArr []string, orderRule string, startSize, pageSize int) (items []*AssessmentVariety, err error) {
	fields := strings.Join(fieldArr, ",")
	if len(fieldArr) == 0 {
		fields = `*`
	}
	order := fmt.Sprintf(`ORDER BY %s DESC`, m.Cols().CreateTime)
	if orderRule != "" {
		order = ` ORDER BY ` + orderRule
	}
	sql := fmt.Sprintf(`SELECT %s FROM %s WHERE 1=1 %s %s LIMIT ?,?`, fields, m.TableName(), condition, order)
	pars = append(pars, startSize, pageSize)
	err = global.DEFAULT_DB.Raw(sql, pars...).Find(&items).Error
	return
}

// AssessmentVarietyAddReq 新增品种请求
type AssessmentVarietyAddReq struct {
	VarietyName             string  `description:"品种名称"`
	VarietyCode             string  `description:"品种编码"`
	MonthlyFluctuate        float64 `description:"月度波动阈值"`
	WeeklyFluctuate         float64 `description:"周度波动阈值"`
	AssessmentResearcherIds []int   `description:"研究员ID"`
}

// AssessmentVarietyEditReq 编辑品种请求
type AssessmentVarietyEditReq struct {
	AssessmentVarietyId int `description:"品种ID"`
	AssessmentVarietyAddReq
}

// AssessmentVarietyRemoveReq 删除品种请求
type AssessmentVarietyRemoveReq struct {
	AssessmentVarietyId int `description:"研究员ID"`
}

// AssessmentVarietyPageListReq 品种列表
type AssessmentVarietyPageListReq struct {
	PageSize             int    `form:"PageSize"`
	CurrentIndex         int    `form:"CurrentIndex"`
	AssessmentVarietyIds string `form:"AssessmentVarietyIds" description:"品种IDs"`
	OnlyEnabled          bool   `form:"OnlyEnabled" description:"是否仅显示启用：true-是"`
}

// AssessmentVarietyPageListResp 研究员列表响应
type AssessmentVarietyPageListResp struct {
	List   []*AssessmentVarietyDetail
	Paging *paging.PagingItem `description:"分页数据"`
}

type AssessmentVarietyDetail struct {
	AssessmentVarietyId int                          `description:"研究员ID"`
	VarietyName         string                       `description:"品种名称"`
	VarietyCode         string                       `description:"品种编码"`
	MonthlyFluctuate    float64                      `description:"月度波动阈值"`
	WeeklyFluctuate     float64                      `description:"周度波动阈值"`
	Enabled             int                          `description:"状态：0-禁用；1-正常"`
	RelationResearcher  []AssessmentResearcherDetail `description:"关联研究员信息"`
}

func (m *AssessmentVariety) Format2Detail() (item *AssessmentVarietyDetail) {
	item = new(AssessmentVarietyDetail)
	item.AssessmentVarietyId = m.AssessmentVarietyId
	item.VarietyName = m.VarietyName
	item.VarietyCode = m.VarietyCode
	item.MonthlyFluctuate = m.MonthlyFluctuate
	item.WeeklyFluctuate = m.WeeklyFluctuate
	item.Enabled = m.Enabled
	return
}