63 lines
1.4 KiB
Go
63 lines
1.4 KiB
Go
|
package db
|
||
|
|
|
||
|
|
import (
|
||
|
|
"log"
|
||
|
|
|
||
|
|
"golang.org/x/crypto/bcrypt"
|
||
|
|
"gorm.io/gorm"
|
||
|
|
"gorm.io/gorm/clause"
|
||
|
|
)
|
||
|
|
|
||
|
|
type User struct {
|
||
|
|
ID uint `gorm:"primaryKey"`
|
||
|
|
FirstName string
|
||
|
|
LastName string
|
||
|
|
Email string `gorm:"unique"`
|
||
|
|
Password string
|
||
|
|
SignupMethod string
|
||
|
|
EmailVerifiedAt int64
|
||
|
|
CreatedAt int64 `gorm:"autoCreateTime"`
|
||
|
|
UpdatedAt int64 `gorm:"autoUpdateTime"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func (user *User) BeforeSave(tx *gorm.DB) error {
|
||
|
|
// Modify current operation through tx.Statement, e.g:
|
||
|
|
if pw, err := bcrypt.GenerateFromPassword([]byte(user.Password), 0); err == nil {
|
||
|
|
tx.Statement.SetColumn("Password", pw)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// AddUser function to add user
|
||
|
|
func (mgr *manager) AddUser(user User) (User, error) {
|
||
|
|
result := mgr.db.Clauses(clause.OnConflict{DoNothing: true}).Create(&user)
|
||
|
|
if result.Error != nil {
|
||
|
|
log.Println(result.Error)
|
||
|
|
return user, result.Error
|
||
|
|
}
|
||
|
|
return user, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// GetUsers function to get all users
|
||
|
|
func (mgr *manager) GetUsers() ([]User, error) {
|
||
|
|
var users []User
|
||
|
|
result := mgr.db.Find(&users)
|
||
|
|
if result.Error != nil {
|
||
|
|
log.Println(result.Error)
|
||
|
|
return users, result.Error
|
||
|
|
}
|
||
|
|
return users, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (mgr *manager) GetUserByEmail(email string) (User, error) {
|
||
|
|
var user User
|
||
|
|
result := mgr.db.Where("email = ?", email).First(&user)
|
||
|
|
|
||
|
|
if result.Error != nil {
|
||
|
|
return user, result.Error
|
||
|
|
}
|
||
|
|
|
||
|
|
return user, nil
|
||
|
|
}
|