98 lines
2.7 KiB
Go
98 lines
2.7 KiB
Go
package repositories
|
|
|
|
import (
|
|
"GoMembership/internal/database"
|
|
|
|
"gorm.io/gorm"
|
|
|
|
"GoMembership/internal/models"
|
|
)
|
|
|
|
type SubscriptionsRepositoryInterface interface {
|
|
CreateSubscription(subscription *models.Subscription) (uint, error)
|
|
UpdateSubscription(subscription *models.Subscription) (*models.Subscription, error)
|
|
GetSubscriptionNames() ([]string, error)
|
|
GetSubscriptions(where map[string]interface{}) (*[]models.Subscription, error)
|
|
// GetUsersBySubscription(id uint) (*[]models.Subscription, error)
|
|
DeleteSubscription(id *uint) error
|
|
}
|
|
|
|
type SubscriptionsRepository struct{}
|
|
|
|
func (sr *SubscriptionsRepository) CreateSubscription(subscription *models.Subscription) (uint, error) {
|
|
|
|
result := database.DB.Create(subscription)
|
|
if result.Error != nil {
|
|
return 0, result.Error
|
|
}
|
|
return subscription.ID, nil
|
|
}
|
|
|
|
func (sr *SubscriptionsRepository) UpdateSubscription(subscription *models.Subscription) (*models.Subscription, error) {
|
|
|
|
result := database.DB.Model(&models.Subscription{ID: subscription.ID}).Updates(subscription)
|
|
if result.Error != nil {
|
|
return nil, result.Error
|
|
}
|
|
return subscription, nil
|
|
}
|
|
|
|
func (sr *SubscriptionsRepository) DeleteSubscription(id *uint) error {
|
|
|
|
result := database.DB.Delete(&models.Subscription{}, id)
|
|
if result.Error != nil {
|
|
return result.Error
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func GetSubscriptionByName(modelname *string) (*models.Subscription, error) {
|
|
var model models.Subscription
|
|
result := database.DB.Where("name = ?", modelname).First(&model)
|
|
if result.Error != nil {
|
|
return nil, result.Error
|
|
}
|
|
return &model, nil
|
|
}
|
|
|
|
func (sr *SubscriptionsRepository) GetSubscriptionNames() ([]string, error) {
|
|
var names []string
|
|
if err := database.DB.Model(&models.Subscription{}).Pluck("name", &names).Error; err != nil {
|
|
return []string{}, err
|
|
}
|
|
return names, nil
|
|
}
|
|
|
|
func (sr *SubscriptionsRepository) GetSubscriptions(where map[string]interface{}) (*[]models.Subscription, error) {
|
|
var subscriptions []models.Subscription
|
|
result := database.DB.Where(where).Find(&subscriptions)
|
|
if result.Error != nil {
|
|
if result.Error == gorm.ErrRecordNotFound {
|
|
return nil, gorm.ErrRecordNotFound
|
|
}
|
|
return nil, result.Error
|
|
}
|
|
return &subscriptions, nil
|
|
}
|
|
|
|
func GetUsersBySubscription(subscriptionID uint) (*[]models.User, error) {
|
|
var users []models.User
|
|
|
|
err := database.DB.Preload("Membership").
|
|
Preload("Membership.Subscription").
|
|
Preload("BankAccount").
|
|
Preload("Licence").
|
|
Preload("Licence.Categories").
|
|
Joins("JOIN memberships ON users.id = memberships.user_id").
|
|
Joins("JOIN subscriptions ON memberships.subscription_id = subscriptions.id").
|
|
Where("subscriptions.id = ?", subscriptionID).
|
|
Find(&users).Error
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &users, nil
|
|
|
|
}
|