From 6aa640d7a2e9af5248a2e89353c299daa45085ce Mon Sep 17 00:00:00 2001
From: FLASH2332 <jayaveeran2005@gmail.com>
Date: Thu, 26 Mar 2026 14:48:18 +0530
Subject: [PATCH] fix: handle DB error for agency lookup in routeHandler

---
 internal/restapi/route_handler.go | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/internal/restapi/route_handler.go b/internal/restapi/route_handler.go
index 5d55dd82..d24e689d 100644
--- a/internal/restapi/route_handler.go
+++ b/internal/restapi/route_handler.go
@@ -1,6 +1,8 @@
 package restapi
 
 import (
+	"database/sql"
+	"errors"
 	"net/http"
 
 	"maglev.onebusaway.org/internal/models"
@@ -39,7 +41,12 @@ func (api *RestAPI) routeHandler(w http.ResponseWriter, r *http.Request) {
 	references := models.NewEmptyReferences()
 
 	agency, err := api.GtfsManager.GtfsDB.Queries.GetAgency(ctx, agencyID)
-	if err == nil {
+	if err != nil {
+		if !errors.Is(err, sql.ErrNoRows) {
+			api.serverErrorResponse(w, r, err)
+			return
+		}
+	} else {
 		agencyModel := models.NewAgencyReference(
 			agency.ID,
 			agency.Name,