c9d20afda8
- Add comprehensive validation and error handling for Notification model - Introduce strongly typed enums for notification types and statuses - Implement GORM hooks for pre-create and pre-update validation - Add helper methods for notification retry and sending logic - Improve Schedule model with validation, availability checks, and duration methods - Add constraints and validation for schedule time slots - Enhance model relationships with foreign key references - Implement additional utility methods for schedule management Improves data integrity, adds robust validation, and provides more comprehensive model behaviors for notifications and schedules.
78 lines
2.2 KiB
Go
78 lines
2.2 KiB
Go
package models
|
|
|
|
import (
|
|
"errors"
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// Schedule represents available time slots
|
|
type Schedule struct {
|
|
gorm.Model
|
|
StartTime time.Time `json:"start_time" gorm:"not null;index" validate:"required"`
|
|
EndTime time.Time `json:"end_time" gorm:"not null;index" validate:"required"`
|
|
IsAvailable bool `json:"is_available" gorm:"default:true"`
|
|
MaxBookings int `json:"max_bookings" gorm:"default:1;check:max_bookings > 0" validate:"min=1,max=100"`
|
|
BookedCount int `json:"booked_count" gorm:"default:0;check:booked_count >= 0" validate:"min=0"`
|
|
}
|
|
|
|
// BeforeCreate is a GORM hook that runs before creating a schedule record
|
|
func (s *Schedule) BeforeCreate(tx *gorm.DB) error {
|
|
// Validate that end time is after start time
|
|
if !s.EndTime.After(s.StartTime) {
|
|
return errors.New("end time must be after start time")
|
|
}
|
|
|
|
// Validate minimum duration (15 minutes)
|
|
if s.EndTime.Sub(s.StartTime) < 15*time.Minute {
|
|
return errors.New("schedule slot must be at least 15 minutes long")
|
|
}
|
|
|
|
// Validate that start time is in the future
|
|
if s.StartTime.Before(time.Now()) {
|
|
return errors.New("schedule slot cannot be created in the past")
|
|
}
|
|
|
|
// Set default values
|
|
if s.MaxBookings == 0 {
|
|
s.MaxBookings = 1
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// BeforeUpdate is a GORM hook that runs before updating a schedule record
|
|
func (s *Schedule) BeforeUpdate(tx *gorm.DB) error {
|
|
// Validate that end time is after start time
|
|
if !s.EndTime.After(s.StartTime) {
|
|
return errors.New("end time must be after start time")
|
|
}
|
|
|
|
// Validate that booked count doesn't exceed max bookings
|
|
if s.BookedCount > s.MaxBookings {
|
|
return errors.New("booked count cannot exceed max bookings")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// IsAvailableForBooking checks if the schedule slot has availability
|
|
func (s *Schedule) IsAvailableForBooking() bool {
|
|
return s.IsAvailable && s.BookedCount < s.MaxBookings && s.StartTime.After(time.Now())
|
|
}
|
|
|
|
// GetRemainingSlots returns the number of remaining booking slots
|
|
func (s *Schedule) GetRemainingSlots() int {
|
|
remaining := s.MaxBookings - s.BookedCount
|
|
if remaining < 0 {
|
|
return 0
|
|
}
|
|
return remaining
|
|
}
|
|
|
|
// Duration returns the duration of the schedule slot
|
|
func (s *Schedule) Duration() time.Duration {
|
|
return s.EndTime.Sub(s.StartTime)
|
|
}
|