146 lines
3.6 KiB
Go
146 lines
3.6 KiB
Go
package models
|
|
|
|
import (
|
|
orm "go-admin/common/global"
|
|
"go-admin/tools"
|
|
)
|
|
|
|
type Post struct {
|
|
PostId int `gorm:"primary_key;AUTO_INCREMENT" json:"postId"` //岗位编号
|
|
PostName string `gorm:"size:128;" json:"postName"` //岗位名称
|
|
PostCode string `gorm:"size:128;" json:"postCode"` //岗位代码
|
|
Sort int `gorm:"" json:"sort"` //岗位排序
|
|
Status string `gorm:"size:4;" json:"status"` //状态
|
|
Remark string `gorm:"size:255;" json:"remark"` //描述
|
|
CreateBy string `gorm:"size:128;" json:"createBy"`
|
|
UpdateBy string `gorm:"size:128;" json:"updateBy"`
|
|
BaseModel
|
|
|
|
DataScope string `gorm:"-" json:"dataScope"`
|
|
Params string `gorm:"-" json:"params"`
|
|
}
|
|
|
|
func (Post) TableName() string {
|
|
return "sys_post"
|
|
}
|
|
|
|
func (e *Post) Create() (Post, error) {
|
|
var doc Post
|
|
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 *Post) Get() (Post, error) {
|
|
var doc Post
|
|
|
|
table := orm.Eloquent.Table(e.TableName())
|
|
if e.PostId != 0 {
|
|
table = table.Where("post_id = ?", e.PostId)
|
|
}
|
|
if e.PostName != "" {
|
|
table = table.Where("post_name = ?", e.PostName)
|
|
}
|
|
if e.PostCode != "" {
|
|
table = table.Where("post_code = ?", e.PostCode)
|
|
}
|
|
if e.Status != "" {
|
|
table = table.Where("status = ?", e.Status)
|
|
}
|
|
|
|
if err := table.First(&doc).Error; err != nil {
|
|
return doc, err
|
|
}
|
|
return doc, nil
|
|
}
|
|
|
|
func (e *Post) GetList() ([]Post, error) {
|
|
var doc []Post
|
|
|
|
table := orm.Eloquent.Table(e.TableName())
|
|
if e.PostId != 0 {
|
|
table = table.Where("post_id = ?", e.PostId)
|
|
}
|
|
if e.PostName != "" {
|
|
table = table.Where("post_name = ?", e.PostName)
|
|
}
|
|
if e.PostCode != "" {
|
|
table = table.Where("post_code = ?", e.PostCode)
|
|
}
|
|
if e.Status != "" {
|
|
table = table.Where("status = ?", e.Status)
|
|
}
|
|
|
|
if err := table.Find(&doc).Error; err != nil {
|
|
return doc, err
|
|
}
|
|
return doc, nil
|
|
}
|
|
|
|
func (e *Post) GetPage(pageSize int, pageIndex int) ([]Post, int, error) {
|
|
var doc []Post
|
|
|
|
table := orm.Eloquent.Table(e.TableName())
|
|
if e.PostId != 0 {
|
|
table = table.Where("post_id = ?", e.PostId)
|
|
}
|
|
if e.PostName != "" {
|
|
table = table.Where("post_name = ?", e.PostName)
|
|
}
|
|
if e.PostCode != "" {
|
|
table = table.Where("post_code = ?", e.PostCode)
|
|
}
|
|
if e.Status != "" {
|
|
table = table.Where("status = ?", e.Status)
|
|
}
|
|
|
|
// 数据权限控制
|
|
dataPermission := new(DataPermission)
|
|
dataPermission.UserId, _ = tools.StringToInt(e.DataScope)
|
|
table, err := dataPermission.GetDataScope("sys_post", table)
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
var count int64
|
|
|
|
if err := table.Order("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 *Post) Update(id int) (update Post, err error) {
|
|
if err = orm.Eloquent.Table(e.TableName()).First(&update, id).Error; err != nil {
|
|
return
|
|
}
|
|
|
|
//参数1:是要修改的数据
|
|
//参数2:是修改的数据
|
|
if err = orm.Eloquent.Table(e.TableName()).Model(&update).Updates(&e).Error; err != nil {
|
|
return
|
|
}
|
|
return
|
|
}
|
|
|
|
func (e *Post) Delete(id int) (success bool, err error) {
|
|
if err = orm.Eloquent.Table(e.TableName()).Where("post_id = ?", id).Delete(&Post{}).Error; err != nil {
|
|
success = false
|
|
return
|
|
}
|
|
success = true
|
|
return
|
|
}
|
|
|
|
func (e *Post) BatchDelete(id []int) (Result bool, err error) {
|
|
if err = orm.Eloquent.Table(e.TableName()).Where("post_id in (?)", id).Delete(&Post{}).Error; err != nil {
|
|
return
|
|
}
|
|
Result = true
|
|
return
|
|
}
|