diff --git a/internal/restapi/arrival_and_departure_for_stop_handler.go b/internal/restapi/arrival_and_departure_for_stop_handler.go index 7171bc1c..ddf51d85 100644 --- a/internal/restapi/arrival_and_departure_for_stop_handler.go +++ b/internal/restapi/arrival_and_departure_for_stop_handler.go @@ -172,7 +172,11 @@ func (api *RestAPI) arrivalAndDepartureForStopHandler(w http.ResponseWriter, r * stopAgency, err := api.GtfsManager.GtfsDB.Queries.GetAgency(ctx, stopAgencyID) if err != nil { - api.serverErrorResponse(w, r, err) + if errors.Is(err, sql.ErrNoRows) { + api.sendNotFound(w, r) + } else { + api.serverErrorResponse(w, r, err) + } return } diff --git a/internal/restapi/arrivals_and_departure_for_stop.go b/internal/restapi/arrivals_and_departure_for_stop.go index d96afcd5..599420fb 100644 --- a/internal/restapi/arrivals_and_departure_for_stop.go +++ b/internal/restapi/arrivals_and_departure_for_stop.go @@ -2,6 +2,8 @@ package restapi import ( "context" + "database/sql" + "errors" "log/slog" "net/http" "strconv" @@ -108,7 +110,11 @@ func (api *RestAPI) arrivalsAndDeparturesForStopHandler(w http.ResponseWriter, r agency, err := api.GtfsManager.GtfsDB.Queries.GetAgency(ctx, stopAgencyID) if err != nil { - api.serverErrorResponse(w, r, err) + if errors.Is(err, sql.ErrNoRows) { + api.sendNotFound(w, r) + } else { + api.serverErrorResponse(w, r, err) + } return } diff --git a/internal/restapi/trip_details_handler.go b/internal/restapi/trip_details_handler.go index 384f29e1..aa587a3d 100644 --- a/internal/restapi/trip_details_handler.go +++ b/internal/restapi/trip_details_handler.go @@ -2,6 +2,8 @@ package restapi import ( "context" + "database/sql" + "errors" "fmt" "log/slog" "net/http" @@ -129,7 +131,11 @@ func (api *RestAPI) tripDetailsHandler(w http.ResponseWriter, r *http.Request) { agency, err := api.GtfsManager.GtfsDB.Queries.GetAgency(ctx, route.AgencyID) if err != nil { - api.serverErrorResponse(w, r, err) + if errors.Is(err, sql.ErrNoRows) { + api.sendNotFound(w, r) + } else { + api.serverErrorResponse(w, r, err) + } return } diff --git a/internal/restapi/trip_for_vehicle_handler.go b/internal/restapi/trip_for_vehicle_handler.go index b4b5f778..18e6d413 100644 --- a/internal/restapi/trip_for_vehicle_handler.go +++ b/internal/restapi/trip_for_vehicle_handler.go @@ -45,7 +45,11 @@ func (api *RestAPI) tripForVehicleHandler(w http.ResponseWriter, r *http.Request agency, err := api.GtfsManager.GtfsDB.Queries.GetAgency(ctx, agencyID) if err != nil { - api.serverErrorResponse(w, r, err) + if errors.Is(err, sql.ErrNoRows) { + api.sendNotFound(w, r) + } else { + api.serverErrorResponse(w, r, err) + } return }