medol-be/app/module/user_roles/repository/user_roles.repository.go

69 lines
1.9 KiB
Go
Raw Normal View History

2024-03-05 19:15:53 +00:00
package repository
import (
"go-humas-be/app/database"
"go-humas-be/app/database/entity"
"go-humas-be/app/module/user_roles/request"
"go-humas-be/utils/paginator"
)
type userRolesRepository struct {
DB *database.Database
}
// UserRolesRepository define interface of IUserRolesRepository
type UserRolesRepository interface {
GetAll(req request.UserRolesQueryRequest) (userRoless []*entity.UserRoles, paging paginator.Pagination, err error)
FindOne(id uint) (userRoles *entity.UserRoles, err error)
Create(userRoles *entity.UserRoles) (err error)
Update(id uint, userRoles *entity.UserRoles) (err error)
Delete(id uint) (err error)
}
func NewUserRolesRepository(db *database.Database) UserRolesRepository {
return &userRolesRepository{
DB: db,
}
}
// implement interface of IUserRolesRepository
func (_i *userRolesRepository) GetAll(req request.UserRolesQueryRequest) (userRoless []*entity.UserRoles, paging paginator.Pagination, err error) {
var count int64
query := _i.DB.DB.Model(&entity.UserRoles{})
query.Count(&count)
req.Pagination.Count = count
req.Pagination = paginator.Paging(req.Pagination)
err = query.Offset(req.Pagination.Offset).Limit(req.Pagination.Limit).Find(&userRoless).Error
if err != nil {
return
}
paging = *req.Pagination
return
}
func (_i *userRolesRepository) FindOne(id uint) (userRoles *entity.UserRoles, err error) {
if err := _i.DB.DB.First(&userRoles, id).Error; err != nil {
return nil, err
}
return userRoles, nil
}
func (_i *userRolesRepository) Create(userRoles *entity.UserRoles) (err error) {
return _i.DB.DB.Create(userRoles).Error
}
func (_i *userRolesRepository) Update(id uint, userRoles *entity.UserRoles) (err error) {
return _i.DB.DB.Model(&entity.UserRoles{}).
Where(&entity.UserRoles{ID: id}).
Updates(userRoles).Error
}
func (_i *userRolesRepository) Delete(id uint) error {
return _i.DB.DB.Delete(&entity.UserRoles{}, id).Error
}