lots of stuff, don't truly remember
This commit is contained in:
65
internal/event/repository.go
Normal file
65
internal/event/repository.go
Normal file
@@ -0,0 +1,65 @@
|
||||
package event
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"time"
|
||||
|
||||
"rideaware/pkg/database"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type Repository struct{}
|
||||
|
||||
func NewRepository() *Repository {
|
||||
return &Repository{}
|
||||
}
|
||||
|
||||
func (r *Repository) CreateEvent(event *Event) error {
|
||||
return database.DB.Create(event).Error
|
||||
}
|
||||
|
||||
func (r *Repository) GetEventByID(id, userID uint) (*Event, error) {
|
||||
var event Event
|
||||
if err := database.DB.Where("id = ? AND user_id = ?", id, userID).First(&event).Error; err != nil {
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return nil, errors.New("event not found")
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
return &event, nil
|
||||
}
|
||||
|
||||
func (r *Repository) GetUserEvents(userID uint) ([]Event, error) {
|
||||
var events []Event
|
||||
if err := database.DB.Where("user_id = ?", userID).Order("event_date ASC").Find(&events).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return events, nil
|
||||
}
|
||||
|
||||
func (r *Repository) GetEventsByDateRange(userID uint, start, end time.Time) ([]Event, error) {
|
||||
var events []Event
|
||||
if err := database.DB.Where("user_id = ? AND event_date BETWEEN ? AND ?", userID, start, end).
|
||||
Order("event_date ASC").Find(&events).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return events, nil
|
||||
}
|
||||
|
||||
func (r *Repository) GetUpcomingEvents(userID uint, limit int) ([]Event, error) {
|
||||
var events []Event
|
||||
if err := database.DB.Where("user_id = ? AND event_date >= ?", userID, time.Now()).
|
||||
Order("event_date ASC").Limit(limit).Find(&events).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return events, nil
|
||||
}
|
||||
|
||||
func (r *Repository) UpdateEvent(event *Event) error {
|
||||
return database.DB.Save(event).Error
|
||||
}
|
||||
|
||||
func (r *Repository) DeleteEvent(id, userID uint) error {
|
||||
return database.DB.Where("id = ? AND user_id = ?", id, userID).Delete(&Event{}).Error
|
||||
}
|
||||
Reference in New Issue
Block a user