eta_bridge/models/eta/admin.go

package eta

import (
	"errors"
	"eta/eta_bridge/global"
	"gorm.io/gorm/clause"
	"time"
)

// Admin 管理员表
type Admin struct {
	AdminId                   int       `gorm:"primaryKey;column:admin_id;type:bigint(20);not null" json:"admin_id"`
	AdminName                 string    `gorm:"uniqueIndex:un;index:name;index:admin_pass;column:admin_name;type:varchar(60);not null" json:"admin_name"`
	AdminAvatar               string    `gorm:"column:admin_avatar;type:varchar(255);not null;default:''" json:"admin_avatar"` // 用户头像
	RealName                  string    `gorm:"column:real_name;type:varchar(60)" json:"real_name"`
	Password                  string    `gorm:"index:password;index:admin_pass;column:password;type:varchar(60);not null" json:"password"`
	LastUpdatedPasswordTime   time.Time `gorm:"column:last_updated_password_time;type:datetime" json:"last_updated_password_time"`
	Enabled                   int       `gorm:"uniqueIndex:un;column:enabled;type:tinyint(1);not null" json:"enabled"` // 1:有效，0:禁用
	Email                     string    `gorm:"column:email;type:varchar(60)" json:"email"`
	LastLoginTime             time.Time `gorm:"column:last_login_time;type:datetime" json:"last_login_time"`                                        // 最近登陆时间
	CreatedTime               time.Time `gorm:"index:created_time;column:created_time;type:datetime;default:CURRENT_TIMESTAMP" json:"created_time"` // 创建时间
	LastUpdatedTime           time.Time `gorm:"index:last_updated_time;column:last_updated_time;type:timestamp;not null;default:CURRENT_TIMESTAMP" json:"last_updated_time"`
	Role                      string    `gorm:"column:role;type:varchar(30);default:saller" json:"role"`                                                  // 用户角色
	Mobile                    string    `gorm:"column:mobile;type:varchar(20)" json:"mobile"`                                                             // 手机号
	RoleType                  int       `gorm:"column:role_type;type:tinyint(4);default:0" json:"role_type"`                                              // 角色类型:1需要录入指标,0:不需要
	RoleId                    int       `gorm:"column:role_id;type:int(11);default:0" json:"role_id"`                                                     // 角色id
	RoleName                  string    `gorm:"column:role_name;type:varchar(100)" json:"role_name"`                                                      // 角色名称
	RoleTypeCode              string    `gorm:"column:role_type_code;type:varchar(20);default:''" json:"role_type_code"`                                  // 角色编码
	DepartmentId              int       `gorm:"column:department_id;type:int(11);default:0" json:"department_id"`                                         // 部门id
	DepartmentName            string    `gorm:"column:department_name;type:varchar(100)" json:"department_name"`                                          // 部门名称
	GroupId                   int       `gorm:"column:group_id;type:int(11);default:0" json:"group_id"`                                                   // 分组id
	GroupName                 string    `gorm:"column:group_name;type:varchar(100)" json:"group_name"`                                                    // 分组名称
	Authority                 int       `gorm:"column:authority;type:tinyint(4);default:0" json:"authority"`                                              // 管理权限，0：无，1：部门负责人，2：小组负责人,或者ficc销售主管，4:ficc销售组长
	Position                  string    `gorm:"column:position;type:varchar(100)" json:"position"`                                                        // 职位
	DisableTime               time.Time `gorm:"column:disable_time;type:datetime" json:"disable_time"`                                                    // 禁用时间
	ChartPermission           uint      `gorm:"column:chart_permission;type:tinyint(9) unsigned;default:0" json:"chart_permission"`                       // 图表指标操作权限，0：只能操作 自己的，1：所有图表可操作
	EdbPermission             uint      `gorm:"column:edb_permission;type:tinyint(9) unsigned;default:0" json:"edb_permission"`                           // 指标库操作权限，0：只能操作 自己的，1：所有指标可操作
	MysteelChemicalPermission uint      `gorm:"column:mysteel_chemical_permission;type:tinyint(9) unsigned;default:0" json:"mysteel_chemical_permission"` // 钢联化工指标操作权限，0：只能操作 自己的，1：所有指标可操作
	OpenId                    string    `gorm:"column:open_id;type:varchar(100);default:''" json:"open_id"`                                               // 弘则部门公众号的openid
	UnionId                   string    `gorm:"column:union_id;type:varchar(100);default:''" json:"union_id"`                                             // 微信公众平台唯一标识
	PredictEdbPermission      int       `gorm:"column:predict_edb_permission;type:tinyint(9)" json:"predict_edb_permission"`                              // 预测指标库操作权限，0：只能操作 自己的，1：所有预测指标可操作
	Province                  string    `gorm:"column:province;type:varchar(255);default:''" json:"province"`                                             // 省
	ProvinceCode              string    `gorm:"column:province_code;type:varchar(100);default:''" json:"province_code"`                                   // 省编码
	City                      string    `gorm:"column:city;type:varchar(255);default:''" json:"city"`                                                     // 市
	CityCode                  string    `gorm:"column:city_code;type:varchar(100);default:''" json:"city_code"`                                           // 市编码
	EmployeeId                string    `gorm:"column:employee_id;type:varchar(64);not null;default:''" json:"employee_id"`                               // 员工工号(钉钉/每刻报销)
	TelAreaCode               string    `gorm:"column:tel_area_code;type:varchar(32);not null;default:'86'" json:"tel_area_code"`                         // 手机号区号
	OutID                     string    `gorm:"column:out_id" json:"out_id"`                                                                              // 外部id
}

func (m *Admin) TableName() string {
	return "admin"
}

// Create 新增用户
func (m *Admin) Create() (err error) {
	err = global.MYSQL["hz_eta"].Create(m).Error
	return
}

// Update 更新用户
func (m *Admin) Update(cols []string) (err error) {
	err = global.MYSQL["hz_eta"].Model(m).Select(cols).Updates(m).Error
	return
}

// GetSysUserById 主键获取系统用户
func GetSysUserById(sysUserId int) (item *Admin, err error) {
	err = global.MYSQL["hz_eta"].Where("admin_id = ?", sysUserId).First(&item).Error
	return
}

// GetSysUserByAdminName 用户名获取系统用户
func GetSysUserByAdminName(adminName string) (item *Admin, err error) {
	err = global.MYSQL["hz_eta"].Where("admin_name = ?", adminName).First(&item).Error
	return
}

// DeleteSysUserByName 根据用户名删除用户
func DeleteSysUserByName(adminName string) (err error) {
	sql := `DELETE FROM admin WHERE admin_name = ? LIMIT 1`
	err = global.MYSQL["hz_eta"].Exec(sql, adminName).Error
	return
}

// GetAllSysUser 获取所有用户
func GetAllSysUser() (items []*Admin, err error) {
	err = global.MYSQL["hz_eta"].Find(&items).Error
	return
}

// GetSysUserByOutId 通过外部id获取系统用户
func GetSysUserByOutId(outId string) (item *Admin, err error) {
	if outId == `` {
		err = errors.New("outId不能为空")
		return
	}
	err = global.MYSQL["hz_eta"].Where("out_id = ?", outId).First(&item).Error
	return
}

func BatchInsertOrUpdateAdmin(admins []Admin) (err error) {
	db := global.MYSQL["hz_eta"]
	OnConflictFunc := clause.OnConflict{
		Columns:   []clause.Column{{Name: "admin_name"}, {Name: "enabled"}},
		DoUpdates: clause.AssignmentColumns([]string{"real_name", "email", "mobile", "department_id", "department_name", "group_id", "group_name", "employee_id"}),
	}
	// 执行批量插入或更新操作
	err = db.Clauses(OnConflictFunc).Create(&admins).Error
	return
}