32 lines
1.0 KiB
Go
32 lines
1.0 KiB
Go
package repositories
|
|
|
|
import (
|
|
"GoMembership/internal/database"
|
|
"GoMembership/internal/models"
|
|
)
|
|
|
|
type DriversLicenceInterface interface {
|
|
FindCategoryByName(categoryName string) (models.LicenceCategory, error)
|
|
FindCategoriesByIDs(ids []uint) ([]models.LicenceCategory, error)
|
|
GetAllCategories() ([]models.LicenceCategory, error)
|
|
}
|
|
|
|
type DriversLicenceRepository struct{}
|
|
|
|
func (r *DriversLicenceRepository) GetAllCategories() ([]models.LicenceCategory, error) {
|
|
var categories []models.LicenceCategory
|
|
err := database.DB.Find(&categories).Error
|
|
return categories, err
|
|
}
|
|
func (r *DriversLicenceRepository) FindCategoriesByIDs(ids []uint) ([]models.LicenceCategory, error) {
|
|
var categories []models.LicenceCategory
|
|
err := database.DB.Where("id IN ?", ids).Find(&categories).Error
|
|
return categories, err
|
|
}
|
|
|
|
func (r *DriversLicenceRepository) FindCategoryByName(categoryName string) (models.LicenceCategory, error) {
|
|
var category models.LicenceCategory
|
|
err := database.DB.Where("category = ?", categoryName).First(&category).Error
|
|
return category, err
|
|
}
|