79 lines
1.7 KiB
Go
79 lines
1.7 KiB
Go
package repositories
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"rijig/model"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type UserProfileRepository interface {
|
|
FindByID(userID string) (*model.User, error)
|
|
Update(user *model.User) error
|
|
UpdateAvatar(userID, avatarURL string) error
|
|
|
|
FindAll() ([]model.User, error)
|
|
FindByRoleID(roleID string) ([]model.User, error)
|
|
}
|
|
|
|
type userProfileRepository struct {
|
|
DB *gorm.DB
|
|
}
|
|
|
|
func NewUserProfileRepository(db *gorm.DB) UserProfileRepository {
|
|
return &userProfileRepository{DB: db}
|
|
}
|
|
|
|
func (r *userProfileRepository) FindByID(userID string) (*model.User, error) {
|
|
var user model.User
|
|
err := r.DB.Preload("Role").Where("id = ?", userID).First(&user).Error
|
|
if err != nil {
|
|
if err == gorm.ErrRecordNotFound {
|
|
return nil, fmt.Errorf("user with ID %s not found", userID)
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
if user.Role == nil {
|
|
return nil, fmt.Errorf("role not found for this user")
|
|
}
|
|
|
|
return &user, nil
|
|
}
|
|
|
|
func (r *userProfileRepository) Update(user *model.User) error {
|
|
err := r.DB.Save(user).Error
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *userProfileRepository) UpdateAvatar(userID, avatarURL string) error {
|
|
var user model.User
|
|
err := r.DB.Model(&user).Where("id = ?", userID).Update("avatar", avatarURL).Error
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *userProfileRepository) FindAll() ([]model.User, error) {
|
|
var users []model.User
|
|
err := r.DB.Preload("Role").Find(&users).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return users, nil
|
|
}
|
|
|
|
func (r *userProfileRepository) FindByRoleID(roleID string) ([]model.User, error) {
|
|
var users []model.User
|
|
err := r.DB.Preload("Role").Where("role_id = ?", roleID).Find(&users).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return users, nil
|
|
}
|