diff --git a/internal/booking/service.go b/internal/booking/service.go index 06d0a2b..61ede45 100644 --- a/internal/booking/service.go +++ b/internal/booking/service.go @@ -37,6 +37,12 @@ func (bs Service) Create(From time.Time, To time.Time, Name string, PhoneNumber return b } +func (bs Service) One(id int) *Booking { + b := &Booking{Id: id} + bs.db.Preload("Items").First(b) + return b +} + type Line struct { From time.Time To time.Time diff --git a/internal/server/handler_booking.go b/internal/server/handler_booking.go index 640d4db..aa3d93a 100644 --- a/internal/server/handler_booking.go +++ b/internal/server/handler_booking.go @@ -79,8 +79,8 @@ func handleBookingPage(bs *booking.Service) echo.HandlerFunc { if err != nil { return err } - b := &booking.Booking{Id: id} - bs.db.Preload("Items").First(b) + + b := bs.One(id) bvm := &views.BookingViewModel{ Id: fmt.Sprintf("%04s", strconv.Itoa(b.Id)), @@ -111,7 +111,7 @@ func handleBookingPage(bs *booking.Service) echo.HandlerFunc { PaymentMethods: constants.PaymentMethods, } component := views.BookingById(bvm) - return s.renderTempl(c, http.StatusOK, component) + return renderTempl(c, http.StatusOK, component) } }