52 lines
1.1 KiB
Go
52 lines
1.1 KiB
Go
package dao
|
|
|
|
import (
|
|
"fmt"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type User struct {
|
|
gorm.Model
|
|
Name string `gorm:"column:name"`
|
|
Age int `gorm:"column:age"`
|
|
Email string `gorm:"column:email"`
|
|
Password string `gorm:"column:password"`
|
|
Gender string `gorm:"column:gender"`
|
|
CreateTime string `gorm:"column:create_time"`
|
|
UpdateTime string `gorm:"column:update_time"`
|
|
}
|
|
|
|
func CreateUser(name string, password, email string) uint {
|
|
user := User{Name: name, Email: email, Password: password}
|
|
DB.Create(&user)
|
|
return user.ID
|
|
}
|
|
|
|
func DeleteUserByID(id int) int {
|
|
DB.Delete(&User{}, id)
|
|
return id
|
|
}
|
|
|
|
func FindUserByID(id int) User {
|
|
var user User
|
|
DB.Debug().First(&user, id)
|
|
return user
|
|
}
|
|
|
|
func FindUserByName(name string) User {
|
|
var user User
|
|
fmt.Println("name:", name)
|
|
DB.Debug().Where("name = ?", name).First(&user)
|
|
return user
|
|
}
|
|
|
|
func FindUserByEmail(email string) User {
|
|
var user User
|
|
DB.Debug().Where("email = ?", email).First(&user)
|
|
return user
|
|
}
|
|
|
|
func UpdateUserByID(id int, name, password, email string) {
|
|
DB.Model(&User{}).Where("id = ?", id).Updates(User{Name: name, Password: password, Email: email})
|
|
}
|