user.go 2.9 KB
Newer Older
1
package dbModel
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 User struct {
Mr.奇淼('s avatar
Mr.奇淼( 已提交
14 15 16 17 18 19 20 21
	gorm.Model  `json:"-"`
	UUID        uuid.UUID `json:"uuid"`
	UserName    string    `json:"userName"`
	PassWord    string    `json:"passWord"`
	NickName    string    `json:"nickName" gorm:"default:'QMPlusUser'"`
	HeaderImg   string    `json:"headerImg" gorm:"default:'http://www.henrongyi.top/avatar/lufu.jpg'"`
	Authority   Authority `json:"authority" form:"ForeignKey:AuthorityId;AssociationForeignKey:AuthorityId"`
	AuthorityId float64   `json:"authorityId" gorm:"default:888"`
Mr.奇淼('s avatar
Mr.奇淼( 已提交
22 23 24 25 26 27 28
	//Propertie                //	多余属性自行添加
	//PropertieId uint  // 自动关联 Propertie 的Id 附加属性过多 建议创建一对一关系
}

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

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

47 48 49 50 51
//修改用户密码
func (u *User) ChangePassWord(newPassWord string) (err error, userInter *User) {
	var user User
	//后期修改jwt+password模式
	u.PassWord = tools.MD5V(u.PassWord)
Mr.奇淼('s avatar
Mr.奇淼( 已提交
52
	err = qmsql.DEFAULTDB.Where("user_name = ? AND pass_word = ?", u.UserName, u.PassWord).First(&user).Update("pass_word", tools.MD5V(newPassWord)).Error
53 54 55
	return err, u
}

56 57
//用户更新接口
func (u *User) UpdataUser() (err error, userInter *User) {
Mr.奇淼('s avatar
Mr.奇淼( 已提交
58
	err = qmsql.DEFAULTDB.Create(u).Error
59 60 61
	return err, u
}

62 63 64 65
//用户登录
func (u *User) Login() (err error, userInter *User) {
	var user User
	u.PassWord = tools.MD5V(u.PassWord)
Mr.奇淼('s avatar
Mr.奇淼( 已提交
66
	err = qmsql.DEFAULTDB.Where("user_name = ? AND pass_word = ?", u.UserName, u.PassWord).First(&user).Error
Mr.奇淼('s avatar
Mr.奇淼( 已提交
67
	err = qmsql.DEFAULTDB.Model(&user).Related(&user.Authority).Error
68
	return err, &user
69
}
70 71 72 73 74 75 76

// 用户头像上传更新地址
func (u *User) UploadHeaderImg(userName string, filePath string) (err error, userInter *User) {
	var user User
	err = qmsql.DEFAULTDB.Where("user_name = ?", userName).First(&user).Update("header_img", filePath).First(&user).Error
	return err, &user
}
Mr.奇淼('s avatar
Mr.奇淼( 已提交
77 78 79 80 81 82 83 84 85 86 87 88 89

// 分页获取数据  需要分页实现这个接口即可
func (u *User) GetInfoList(info modelInterface.PageInfo) (err error, list interface{}, total int) {
	// 封装分页方法 调用即可 传入 当前的结构体和分页信息
	err, db, total := servers.PagingServer(u, info)
	if err != nil {
		return
	} else {
		var userList []User
		err = db.Find(&userList).Error
		return err, userList, total
	}
}