device-manage-server/device-manage/app/admin/models/dictdata.go

package models

import (
	"errors"

	orm "device-manage/common/global"
	"device-manage/tools"
)

type DictData struct {
	DictCode  int    `gorm:"primary_key;auto_increment;" json:"dictCode" example:"1"` //字典编码
	DictSort  int    `gorm:"" json:"dictSort"`                                        //显示顺序
	DictLabel string `gorm:"size:128;" json:"dictLabel"`                              //数据标签
	DictValue string `gorm:"size:255;" json:"dictValue"`                              //数据键值
	DictType  string `gorm:"size:64;" json:"dictType"`                                //字典类型
	CssClass  string `gorm:"size:128;" json:"cssClass"`                               //
	ListClass string `gorm:"size:128;" json:"listClass"`                              //
	IsDefault string `gorm:"size:8;" json:"isDefault"`                                //
	Status    string `gorm:"size:4;" json:"status"`                                   //状态
	Default   string `gorm:"size:8;" json:"default"`                                  //
	CreateBy  string `gorm:"size:64;" json:"createBy"`                                //
	UpdateBy  string `gorm:"size:64;" json:"updateBy"`                                //
	Remark    string `gorm:"size:255;" json:"remark"`                                 //备注
	BaseModel

	Params    string `gorm:"-" json:"params"`
	DataScope string `gorm:"-" json:"dataScope"`
}

func (DictData) TableName() string {
	return "sys_dict_data"
}

func (e *DictData) Create() (DictData, error) {
	var doc DictData

	var i int64
	if err := orm.Eloquent.Table(e.TableName()).
		Where("dict_type = ?", e.DictType).
		Where("dict_label=? or (dict_label=? and dict_value = ?)", e.DictLabel, e.DictLabel, e.DictValue).
		Count(&i).Error; err != nil {
		return doc, err
	}
	if i > 0 {
		return doc, errors.New("字典标签或者字典键值已经存在！")
	}

	result := orm.Eloquent.Table(e.TableName()).Create(&e)
	if result.Error != nil {
		err := result.Error
		return doc, err
	}
	doc = *e
	return doc, nil
}

func (e *DictData) GetByCode() (DictData, error) {
	var doc DictData

	table := orm.Eloquent.Table(e.TableName())
	if e.DictCode != 0 {
		table = table.Where("dict_code = ?", e.DictCode)
	}
	if e.DictLabel != "" {
		table = table.Where("dict_label = ?", e.DictLabel)
	}
	if e.DictType != "" {
		table = table.Where("dict_type = ?", e.DictType)
	}

	if err := table.First(&doc).Error; err != nil {
		return doc, err
	}
	return doc, nil
}

func (e *DictData) Get() ([]DictData, error) {
	var doc []DictData

	table := orm.Eloquent.Table(e.TableName())
	if e.DictCode != 0 {
		table = table.Where("dict_code = ?", e.DictCode)
	}
	if e.DictLabel != "" {
		table = table.Where("dict_label = ?", e.DictLabel)
	}
	if e.DictType != "" {
		table = table.Where("dict_type = ?", e.DictType)
	}

	if err := table.Order("dict_sort").Find(&doc).Error; err != nil {
		return doc, err
	}
	return doc, nil
}

func (e *DictData) GetPage(pageSize int, pageIndex int) ([]DictData, int, error) {
	var doc []DictData

	table := orm.Eloquent.Table(e.TableName())
	if e.DictCode != 0 {
		table = table.Where("dict_code = ?", e.DictCode)
	}
	if e.DictType != "" {
		table = table.Where("dict_type = ?", e.DictType)
	}
	if e.DictLabel != "" {
		table = table.Where("dict_label = ?", e.DictLabel)
	}
	if e.Status != "" {
		table = table.Where("status = ?", e.Status)
	}

	// 数据权限控制
	dataPermission := new(DataPermission)
	dataPermission.UserId, _ = tools.StringToInt(e.DataScope)
	table, err := dataPermission.GetDataScope("sys_dict_data", table)
	if err != nil {
		return nil, 0, err
	}
	var count int64

	if err := table.Order("dict_sort").Offset((pageIndex - 1) * pageSize).Limit(pageSize).Find(&doc).Offset(-1).Limit(-1).Count(&count).Error; err != nil {
		return nil, 0, err
	}
	//table.Where("`deleted_at` IS NULL").Count(&count)
	return doc, int(count), nil
}

func (e *DictData) Update(id int) (update DictData, err error) {
	if err = orm.Eloquent.Table(e.TableName()).Where("dict_code = ?", id).First(&update).Error; err != nil {
		return
	}

	if e.DictLabel != "" && e.DictLabel != update.DictLabel {
		return update, errors.New("标签不允许修改！")
	}

	if e.DictValue != "" && e.DictValue != update.DictValue {
		return update, errors.New("键值不允许修改！")
	}

	//参数1:是要修改的数据
	//参数2:是修改的数据
	if err = orm.Eloquent.Table(e.TableName()).Model(&update).Updates(&e).Error; err != nil {
		return
	}
	return
}

func (e *DictData) Delete(id int) (success bool, err error) {
	if err = orm.Eloquent.Table(e.TableName()).Where("dict_code = ?", id).Delete(&DictData{}).Error; err != nil {
		success = false
		return
	}
	success = true
	return
}

func (e *DictData) BatchDelete(id []int) (Result bool, err error) {
	if err = orm.Eloquent.Table(e.TableName()).Where("dict_code in (?)", id).Delete(&DictData{}).Error; err != nil {
		return
	}
	Result = true
	return
}