sys_user.go 2.9 KB
Newer Older
1
package sysModel
Mr.奇淼('s avatar
Mr.奇淼( 已提交
2 3 4

import (
	"github.com/jinzhu/gorm"
5
	"github.com/pkg/errors"
Mr.奇淼('s avatar
Mr.奇淼( 已提交
6
	uuid "github.com/satori/go.uuid"
Mr.奇淼('s avatar
Mr.奇淼( 已提交
7
	"main/controller/servers"
Mr.奇淼('s avatar
Mr.奇淼( 已提交
8
	"main/init/qmsql"
Mr.奇淼('s avatar
Mr.奇淼( 已提交
9
	"main/model/modelInterface"
10
	"main/tools"
Mr.奇淼('s avatar
Mr.奇淼( 已提交
11 12
)

13
type SysUser struct {
14
	gorm.Model
15 16 17 18 19 20
	UUID        uuid.UUID    `json:"uuid"`
	Username    string       `json:"userName"`
	Password    string       `json:"-"`
	NickName    string       `json:"nickName" gorm:"default:'QMPlusUser'"`
	HeaderImg   string       `json:"headerImg" gorm:"default:'http://www.henrongyi.top/avatar/lufu.jpg'"`
	Authority   SysAuthority `json:"authority" gorm:"ForeignKey:AuthorityId;AssociationForeignKey:AuthorityId"`
21
	AuthorityId string       `json:"authorityId" gorm:"default:888"`
Mr.奇淼('s avatar
Mr.奇淼( 已提交
22 23 24 25 26
}

//type Propertie struct {
//	gorm.Model
//}
27

28
//注册接口model方法
29 30
func (u *SysUser) Regist() (err error, userInter *SysUser) {
	var user SysUser
31
	//判断用户名是否注册
Mr.奇淼('s avatar
Mr.奇淼( 已提交
32
	findErr := qmsql.DEFAULTDB.Where("username = ?", u.Username).First(&user).Error
33 34 35 36 37
	//err为nil表明读取到了 不能注册
	if findErr == nil {
		return errors.New("用户名已注册"), nil
	} else {
		// 否则 附加uuid 密码md5简单加密 注册
Mr.奇淼('s avatar
Mr.奇淼( 已提交
38
		u.Password = tools.MD5V(u.Password)
39
		u.UUID = uuid.NewV4()
Mr.奇淼('s avatar
Mr.奇淼( 已提交
40
		err = qmsql.DEFAULTDB.Create(u).Error
41
	}
42 43 44
	return err, u
}

45
//修改用户密码
46 47
func (u *SysUser) ChangePassword(newPassword string) (err error, userInter *SysUser) {
	var user SysUser
48
	//后期修改jwt+password模式
Mr.奇淼('s avatar
Mr.奇淼( 已提交
49
	u.Password = tools.MD5V(u.Password)
Mr.奇淼('s avatar
Mr.奇淼( 已提交
50
	err = qmsql.DEFAULTDB.Where("username = ? AND password = ?", u.Username, u.Password).First(&user).Update("password", tools.MD5V(newPassword)).Error
51 52 53
	return err, u
}

54
//用户更新接口
55 56
func (u *SysUser) SetUserAuthority(uuid uuid.UUID, AuthorityId string) (err error) {
	err = qmsql.DEFAULTDB.Where("uuid = ?", uuid).First(&SysUser{}).Update("authority_id", AuthorityId).Error
Mr.奇淼('s avatar
Mr.奇淼( 已提交
57
	return err
58 59
}

60
//用户登录
61 62
func (u *SysUser) Login() (err error, userInter *SysUser) {
	var user SysUser
Mr.奇淼('s avatar
Mr.奇淼( 已提交
63
	u.Password = tools.MD5V(u.Password)
Mr.奇淼('s avatar
Mr.奇淼( 已提交
64
	err = qmsql.DEFAULTDB.Where("username = ? AND password = ?", u.Username, u.Password).First(&user).Error
Mr.奇淼('s avatar
Mr.奇淼( 已提交
65 66 67
	if(err!=nil){
		return err,&user
	}
Mr.奇淼('s avatar
Mr.奇淼( 已提交
68
	err = qmsql.DEFAULTDB.Where("authority_id = ?", user.AuthorityId).First(&user.Authority).Error
69
	return err, &user
70
}
71 72

// 用户头像上传更新地址
73 74
func (u *SysUser) UploadHeaderImg(uuid uuid.UUID, filePath string) (err error, userInter *SysUser) {
	var user SysUser
75
	err = qmsql.DEFAULTDB.Where("uuid = ?", uuid).First(&user).Update("header_img", filePath).First(&user).Error
76 77
	return err, &user
}
Mr.奇淼('s avatar
Mr.奇淼( 已提交
78 79

// 分页获取数据  需要分页实现这个接口即可
80
func (u *SysUser) GetInfoList(info modelInterface.PageInfo) (err error, list interface{}, total int) {
Mr.奇淼('s avatar
Mr.奇淼( 已提交
81 82 83 84 85
	// 封装分页方法 调用即可 传入 当前的结构体和分页信息
	err, db, total := servers.PagingServer(u, info)
	if err != nil {
		return
	} else {
86
		var userList []SysUser
87
		err = db.Preload("Authority").Find(&userList).Error
Mr.奇淼('s avatar
Mr.奇淼( 已提交
88 89 90
		return err, userList, total
	}
}