|
@@ -2,9 +2,10 @@ package dbModel
|
|
|
|
|
|
import (
|
|
|
"github.com/jinzhu/gorm"
|
|
|
+ "github.com/pkg/errors"
|
|
|
uuid "github.com/satori/go.uuid"
|
|
|
"main/init/mysql"
|
|
|
- "main/model/modelInterface"
|
|
|
+ "main/tools"
|
|
|
)
|
|
|
|
|
|
type User struct {
|
|
@@ -21,26 +22,43 @@ type User struct {
|
|
|
|
|
|
|
|
|
|
|
|
-func NewUser(user User) *User {
|
|
|
- return &User{UserName: user.UserName, PassWord: user.PassWord, NickName: user.NickName, HeaderImg: user.HeaderImg}
|
|
|
-}
|
|
|
|
|
|
-func (u *User) Create() (err error, user modelInterface.CURD) {
|
|
|
- err = mysql.DEFAULTDB.Create(u).Error
|
|
|
+
|
|
|
+func (u *User) Regist() (err error, userInter *User) {
|
|
|
+ var user User
|
|
|
+
|
|
|
+ findErr := mysql.DEFAULTDB.Where("user_name = ?", u.UserName).First(&user).Error
|
|
|
+
|
|
|
+ if findErr == nil {
|
|
|
+ return errors.New("用户名已注册"), nil
|
|
|
+ } else {
|
|
|
+
|
|
|
+ u.PassWord = tools.MD5V(u.PassWord)
|
|
|
+ u.UUID = uuid.NewV4()
|
|
|
+ err = mysql.DEFAULTDB.Create(u).Error
|
|
|
+ }
|
|
|
return err, u
|
|
|
}
|
|
|
|
|
|
-func (u *User) Delete() (err error, user modelInterface.CURD) {
|
|
|
- err = mysql.DEFAULTDB.Create(u).Error
|
|
|
+
|
|
|
+func (u *User) ChangePassWord(newPassWord string) (err error, userInter *User) {
|
|
|
+ var user User
|
|
|
+
|
|
|
+ u.PassWord = tools.MD5V(u.PassWord)
|
|
|
+ err = mysql.DEFAULTDB.Where("user_name = ? AND pass_word = ?", u.UserName, u.PassWord).First(&user).Update("pass_word", tools.MD5V(newPassWord)).Error
|
|
|
return err, u
|
|
|
}
|
|
|
|
|
|
-func (u *User) Updata() (err error, user modelInterface.CURD) {
|
|
|
+
|
|
|
+func (u *User) UpdataUser() (err error, userInter *User) {
|
|
|
err = mysql.DEFAULTDB.Create(u).Error
|
|
|
return err, u
|
|
|
}
|
|
|
|
|
|
-func (u *User) Read() (err error, user modelInterface.CURD) {
|
|
|
- err = mysql.DEFAULTDB.Create(u).Error
|
|
|
- return err, u
|
|
|
+
|
|
|
+func (u *User) Login() (err error, userInter *User) {
|
|
|
+ var user User
|
|
|
+ u.PassWord = tools.MD5V(u.PassWord)
|
|
|
+ err = mysql.DEFAULTDB.Where("user_name = ? AND pass_word = ?", u.UserName, u.PassWord).First(&user).Error
|
|
|
+ return err, &user
|
|
|
}
|