33 lines
788 B
Go
33 lines
788 B
Go
package models
|
|
|
|
import (
|
|
"github.com/jinzhu/gorm"
|
|
)
|
|
|
|
type User struct {
|
|
gorm.Model
|
|
Name string `gorm:"unique_index"`
|
|
Email string `gorm:"unique_index"`
|
|
Avatar string
|
|
Pwd string
|
|
Role int `gorm:"default:1"` // 0 管理员 1正常用户
|
|
}
|
|
|
|
func QueryUserByEmailAndPassword(email, password string) (user User, err error) {
|
|
e := db.Model(&User{}).First(&user, "email = ? and pwd = ?", email, password).Error
|
|
return user, e
|
|
}
|
|
|
|
func QueryUserByName(name string) (user User, err error) {
|
|
e := db.Model(&User{}).First(&user, "name = ?", name).Error
|
|
return user, e
|
|
}
|
|
|
|
func QueryUserByEmail(email string) (user User, err error) {
|
|
e := db.Model(&User{}).First(&user, "email = ?", email).Error
|
|
return user, e
|
|
}
|
|
|
|
func SaveUser(user *User) error {
|
|
return db.Create(user).Error
|
|
} |