56 lines
1.0 KiB
Go
56 lines
1.0 KiB
Go
package repositories
|
|
|
|
import(
|
|
"gorm.io/gorm"
|
|
|
|
"example.com/gin/test/models"
|
|
)
|
|
|
|
type UserRepository interface {
|
|
Create(models.User) (models.User, error)
|
|
GetByEmail(string) (models.User, error)
|
|
GetById(interface{}) (models.User, error)
|
|
}
|
|
|
|
type GORMUserRepository struct {
|
|
DB *gorm.DB
|
|
}
|
|
|
|
func NewGORMUserRepository(db *gorm.DB) UserRepository {
|
|
return &GORMUserRepository{
|
|
DB: db,
|
|
}
|
|
}
|
|
|
|
func (u *GORMUserRepository) Create(user models.User) (models.User, error) {
|
|
result := u.DB.Create(&user)
|
|
|
|
if result.Error != nil {
|
|
return models.User{}, result.Error
|
|
}
|
|
|
|
return user, nil
|
|
}
|
|
|
|
func (u *GORMUserRepository) GetByEmail(email string) (models.User, error) {
|
|
var user models.User
|
|
result := u.DB.First(&user, "email = ?", email)
|
|
|
|
if result.Error != nil {
|
|
return models.User{}, result.Error
|
|
}
|
|
|
|
return user, nil
|
|
}
|
|
|
|
func (u *GORMUserRepository) GetById(id interface{}) (models.User, error) {
|
|
var user models.User
|
|
result := u.DB.First(&user, id)
|
|
|
|
if result.Error != nil {
|
|
return models.User{}, result.Error
|
|
}
|
|
|
|
return user, nil
|
|
}
|