55 lines
2.1 KiB
Go
55 lines
2.1 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/gochat/gochat/internal/model"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// NotificationSubscriptionRepo provides data access for NotificationSubscription.
|
|
type NotificationSubscriptionRepo struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewNotificationSubscriptionRepo(db *gorm.DB) *NotificationSubscriptionRepo {
|
|
return &NotificationSubscriptionRepo{db: db}
|
|
}
|
|
|
|
// Create inserts a new notification subscription.
|
|
func (r *NotificationSubscriptionRepo) Create(ctx context.Context, sub *model.NotificationSubscription) error {
|
|
return r.db.WithContext(ctx).Create(sub).Error
|
|
}
|
|
|
|
// FindByID retrieves a subscription by ID.
|
|
func (r *NotificationSubscriptionRepo) FindByID(ctx context.Context, id uint) (*model.NotificationSubscription, error) {
|
|
var sub model.NotificationSubscription
|
|
err := r.db.WithContext(ctx).First(&sub, id).Error
|
|
return &sub, err
|
|
}
|
|
|
|
// FindByUserID retrieves all subscriptions for a user.
|
|
func (r *NotificationSubscriptionRepo) FindByUserID(ctx context.Context, userID uint) ([]model.NotificationSubscription, error) {
|
|
var subs []model.NotificationSubscription
|
|
err := r.db.WithContext(ctx).Where("user_id = ?", userID).Find(&subs).Error
|
|
return subs, err
|
|
}
|
|
|
|
// FindByIdentifier retrieves a subscription by its unique identifier.
|
|
func (r *NotificationSubscriptionRepo) FindByIdentifier(ctx context.Context, identifier string) (*model.NotificationSubscription, error) {
|
|
var sub model.NotificationSubscription
|
|
err := r.db.WithContext(ctx).Where("identifier = ?", identifier).First(&sub).Error
|
|
return &sub, err
|
|
}
|
|
|
|
// Delete removes a subscription by ID.
|
|
func (r *NotificationSubscriptionRepo) Delete(ctx context.Context, id uint) error {
|
|
return r.db.WithContext(ctx).Delete(&model.NotificationSubscription{}, id).Error
|
|
}
|
|
|
|
// DeleteByUserIDAndIdentifier removes a specific subscription by user and identifier.
|
|
func (r *NotificationSubscriptionRepo) DeleteByUserIDAndIdentifier(ctx context.Context, userID uint, identifier string) error {
|
|
return r.db.WithContext(ctx).
|
|
Where("user_id = ? AND identifier = ?", userID, identifier).
|
|
Delete(&model.NotificationSubscription{}).Error
|
|
} |