sys_user.go 4.3 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13
package service

import (
	"errors"
	"gin-vue-admin/global"
	"gin-vue-admin/model"
	"gin-vue-admin/model/request"
	"gin-vue-admin/utils"
	uuid "github.com/satori/go.uuid"
)

// @title    Register
// @description   register, 用户注册
14 15
// @auth                     (2020/04/05  20:22)
// @param     u               model.SysUser
16 17
// @return    err             error
// @return    userInter       *SysUser
Mr.奇淼('s avatar
Mr.奇淼( 已提交
18

19
func Register(u model.SysUser) (err error, userInter model.SysUser) {
20
	var user model.SysUser
21
	// 判断用户名是否注册
22
	notRegister := global.GVA_DB.Where("username = ?", u.Username).First(&user).RecordNotFound()
23
	// notRegister为false表明读取到了 不能注册
24
	if !notRegister {
25
		return errors.New("用户名已注册"), userInter
26 27 28 29
	} else {
		// 否则 附加uuid 密码md5简单加密 注册
		u.Password = utils.MD5V([]byte(u.Password))
		u.UUID = uuid.NewV4()
30
		err = global.GVA_DB.Create(&u).Error
31 32 33 34 35 36
	}
	return err, u
}

// @title    Login
// @description   login, 用户登录
37 38
// @auth                     (2020/04/05  20:22)
// @param     u               *model.SysUser
39 40
// @return    err             error
// @return    userInter       *SysUser
Mr.奇淼('s avatar
Mr.奇淼( 已提交
41

42 43 44 45 46 47 48 49 50 51 52 53 54
func Login(u *model.SysUser) (err error, userInter *model.SysUser) {
	var user model.SysUser
	u.Password = utils.MD5V([]byte(u.Password))
	err = global.GVA_DB.Where("username = ? AND password = ?", u.Username, u.Password).First(&user).Error
	if err != nil {
		return err, &user
	}
	err = global.GVA_DB.Where("authority_id = ?", user.AuthorityId).First(&user.Authority).Error
	return err, &user
}

// @title    ChangePassword
// @description   change the password of a certain user, 修改用户密码
55 56
// @auth                     (2020/04/05  20:22)
// @param     u               *model.SysUser
57 58 59
// @param     newPassword     string
// @return    err             error
// @return    userInter       *SysUser
Mr.奇淼('s avatar
Mr.奇淼( 已提交
60

61
func ChangePassword(u *model.SysUser, newPassword string) (err error, userInter *model.SysUser) {
62
	var user model.SysUser
63
	// TODO:后期修改jwt+password模式
64 65 66 67 68 69 70
	u.Password = utils.MD5V([]byte(u.Password))
	err = global.GVA_DB.Where("username = ? AND password = ?", u.Username, u.Password).First(&user).Update("password", utils.MD5V([]byte(newPassword))).Error
	return err, u
}

// @title    GetInfoList
// @description   get user list by pagination, 分页获取数据
71 72
// @auth                      (2020/04/05  20:22)
// @param     info             request.PageInfo
73 74 75
// @return    err              error
// @return    list             interface{}
// @return    total            int
Mr.奇淼('s avatar
Mr.奇淼( 已提交
76

77 78 79 80
func GetUserInfoList(info request.PageInfo) (err error, list interface{}, total int) {
	limit := info.PageSize
	offset := info.PageSize * (info.Page - 1)
	db := global.GVA_DB
81
	var userList []model.SysUser
Mr.奇淼('s avatar
Mr.奇淼( 已提交
82
	err = db.Find(&userList).Count(&total).Error
83 84
	err = db.Limit(limit).Offset(offset).Preload("Authority").Find(&userList).Error
	return err, userList, total
85 86 87 88
}

// @title    SetUserAuthority
// @description   set the authority of a certain user, 设置一个用户的权限
89
// @auth                     (2020/04/05  20:22)
90 91 92
// @param     uuid            UUID
// @param     authorityId     string
// @return    err             error
Mr.奇淼('s avatar
Mr.奇淼( 已提交
93

94
func SetUserAuthority(uuid uuid.UUID, authorityId string) (err error) {
95 96 97 98
	err = global.GVA_DB.Where("uuid = ?", uuid).First(&model.SysUser{}).Update("authority_id", authorityId).Error
	return err
}

99 100 101 102 103 104 105 106 107 108 109 110 111
// @title    SetUserAuthority
// @description   set the authority of a certain user, 设置一个用户的权限
// @auth                     (2020/04/05  20:22)
// @param     uuid            UUID
// @param     authorityId     string
// @return    err             error

func DeleteUser(id float64) (err error) {
	var user model.SysUser
	err = global.GVA_DB.Where("id = ?", id).Delete(&user).Error
	return err
}

112 113
// @title    UploadHeaderImg
// @description   upload avatar, 用户头像上传更新地址
114
// @auth                     (2020/04/05  20:22)
115 116 117 118
// @param     uuid            UUID
// @param     filePath        string
// @return    err             error
// @return    userInter       *SysUser
Mr.奇淼('s avatar
Mr.奇淼( 已提交
119

120 121 122 123 124
func UploadHeaderImg(uuid uuid.UUID, filePath string) (err error, userInter *model.SysUser) {
	var user model.SysUser
	err = global.GVA_DB.Where("uuid = ?", uuid).First(&user).Update("header_img", filePath).First(&user).Error
	return err, &user
}