ferry/models/tools/systables.go

package tools

import (
	"ferry/global/orm"
	"ferry/models"
)

type SysTables struct {
	TableId             int          `gorm:"primary_key;auto_increment" json:"tableId"`             //表编码
	TBName              string       `gorm:"column:table_name;type:varchar(255);" json:"tableName"` //表名称
	TableComment        string       `gorm:"type:varchar(255);" json:"tableComment"`                //表备注
	ClassName           string       `gorm:"type:varchar(255);" json:"className"`                   //类名
	TplCategory         string       `gorm:"type:varchar(255);" json:"tplCategory"`
	PackageName         string       `gorm:"type:varchar(255);" json:"packageName"` //包名
	ModuleName          string       `gorm:"type:varchar(255);" json:"moduleName"`  //模块名
	BusinessName        string       `gorm:"type:varchar(255);" json:"businessName"`
	FunctionName        string       `gorm:"type:varchar(255);" json:"functionName"`   //功能名称
	FunctionAuthor      string       `gorm:"type:varchar(255);" json:"functionAuthor"` //功能作者
	PkColumn            string       `gorm:"type:varchar(255);" json:"pkColumn"`
	PkGoField           string       `gorm:"type:varchar(255);" json:"pkGoField"`
	PkJsonField         string       `gorm:"type:varchar(255);" json:"pkJsonField"`
	Options             string       `gorm:"type:varchar(255);" json:"options"`
	TreeCode            string       `gorm:"type:varchar(255);" json:"treeCode"`
	TreeParentCode      string       `gorm:"type:varchar(255);" json:"treeParentCode"`
	TreeName            string       `gorm:"type:varchar(255);" json:"treeName"`
	Tree                bool         `gorm:"type:char(1);" json:"tree"`
	Crud                bool         `gorm:"type:char(1);" json:"crud"`
	Remark              string       `gorm:"type:varchar(255);" json:"remark"`
	IsLogicalDelete     string       `gorm:"type:char(1);" json:"isLogicalDelete"`
	LogicalDelete       bool         `gorm:"type:char(1);" json:"logicalDelete"`
	LogicalDeleteColumn string       `gorm:"type:varchar(128);" json:"logicalDeleteColumn"`
	CreateBy            string       `gorm:"type:varchar(128);" json:"createBy"`
	UpdateBy            string       `gorm:"type:varchar(128);" json:"updateBy"`
	DataScope           string       `gorm:"-" json:"dataScope"`
	Params              Params       `gorm:"-" json:"params"`
	Columns             []SysColumns `gorm:"-" json:"columns"`

	models.BaseModel
}

func (SysTables) TableName() string {
	return "sys_tables"
}

type Params struct {
	TreeCode       string `gorm:"-" json:"treeCode"`
	TreeParentCode string `gorm:"-" json:"treeParentCode"`
	TreeName       string `gorm:"-" json:"treeName"`
}

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

	table := orm.Eloquent.Select("*").Table("sys_tables")

	if e.TBName != "" {
		table = table.Where("table_name = ?", e.TBName)
	}
	if e.TableComment != "" {
		table = table.Where("table_comment = ?", e.TableComment)
	}

	var count int

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

func (e *SysTables) Get() (SysTables, error) {
	var doc SysTables
	var err error
	table := orm.Eloquent.Select("*").Table("sys_tables")

	if e.TBName != "" {
		table = table.Where("table_name = ?", e.TBName)
	}
	if e.TableId != 0 {
		table = table.Where("table_id = ?", e.TableId)
	}
	if e.TableComment != "" {
		table = table.Where("table_comment = ?", e.TableComment)
	}

	if err := table.First(&doc).Error; err != nil {
		return doc, err
	}
	var col SysColumns
	col.TableId = doc.TableId
	if doc.Columns, err = col.GetList(); err != nil {
		return doc, err
	}

	return doc, nil
}

func (e *SysTables) Create() (SysTables, error) {
	var doc SysTables
	result := orm.Eloquent.Table("sys_tables").Create(&e)
	if result.Error != nil {
		err := result.Error
		return doc, err
	}
	doc = *e
	for i := 0; i < len(e.Columns); i++ {
		e.Columns[i].TableId = doc.TableId

		e.Columns[i].Create()
	}

	return doc, nil
}

func (e *SysTables) Update() (update SysTables, err error) {
	if err = orm.Eloquent.Table("sys_tables").First(&update, e.TableId).Error; err != nil {
		return
	}

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

	for i := 0; i < len(e.Columns); i++ {
		_, _ = e.Columns[i].Update()
	}
	return
}

func (e *SysTables) Delete() (success bool, err error) {
	if err = orm.Eloquent.Table("sys_tables").Delete(SysTables{}, "table_id = ?", e.TableId).Error; err != nil {
		success = false
		return
	}
	if err = orm.Eloquent.Table("sys_columns").Delete(SysColumns{}, "table_id = ?", e.TableId).Error; err != nil {
		success = false
		return
	}
	success = true
	return
}

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