package models

import (
	sql2 "database/sql"
	"eta/eta_hub/global"
	"eta/eta_hub/utils"
	"fmt"
	"strings"
)

// AssessmentResearcherVarietyMapping 研究员考核-用户权限关系表
type AssessmentResearcherVarietyMapping struct {
	Id                     int `gorm:"column:id;primaryKey;autoIncrement"`
	AssessmentResearcherId int `description:"研究员ID"`
	AdminId                int `description:"系统用户ID"`
	VarietyId              int `description:"品种ID"`
}

func (m *AssessmentResearcherVarietyMapping) TableName() string {
	return "assessment_researcher_variety_mapping"
}

type AssessmentResearcherVarietyMappingCols struct {
	PrimaryId              string
	AssessmentResearcherId string
	AdminId                string
	VarietyId              string
}

func (m *AssessmentResearcherVarietyMapping) Cols() AssessmentResearcherVarietyMappingCols {
	return AssessmentResearcherVarietyMappingCols{
		PrimaryId:              "id",
		AssessmentResearcherId: "assessment_researcher_id",
		AdminId:                "admin_id",
		VarietyId:              "variety_id",
	}
}

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

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

func (m *AssessmentResearcherVarietyMapping) 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.Id).Error
	return
}

func (m *AssessmentResearcherVarietyMapping) 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 *AssessmentResearcherVarietyMapping) 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 *AssessmentResearcherVarietyMapping) GetItemById(primaryId int) (item *AssessmentResearcherVarietyMapping, 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 *AssessmentResearcherVarietyMapping) GetItemByCondition(condition string, pars []interface{}, orderRule string) (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(&m).Error
	return
}

func (m *AssessmentResearcherVarietyMapping) GetItemsByCondition(condition string, pars []interface{}, fieldArr []string, orderRule string) (items []*AssessmentResearcherVarietyMapping, err error) {
	fields := strings.Join(fieldArr, ",")
	if len(fieldArr) == 0 {
		fields = `*`
	}
	order := ``
	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 *AssessmentResearcherVarietyMapping) GetCountByCondition(condition string, pars []interface{}) (count int, err error) {
	var totalNull sql2.NullInt64
	sql := fmt.Sprintf(`SELECT COUNT(1) FROM %s WHERE 1=1 %s`, m.TableName(), condition)
	err = global.DEFAULT_DB.Raw(sql, pars...).Scan(&totalNull).Error
	if err != nil {
		return
	}
	count = int(totalNull.Int64)

	return
}