fix journey planner query

Travigo · Dec 15, 2024 · 9f84399 · 9f84399
1 parent eebf9de
commit 9f84399
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/pkg/api/routes/planner.go b/pkg/api/routes/planner.go
@@ -38,6 +38,7 @@ func getPlanBetweenStops(c *fiber.Ctx) error {
 	if err != nil {
 		c.SendStatus(fiber.StatusNotFound)
 		return c.JSON(fiber.Map{
+			"field": "origin",
 			"error": err.Error(),
 		})
 	}
@@ -48,6 +49,7 @@ func getPlanBetweenStops(c *fiber.Ctx) error {
 	if err != nil {
 		c.SendStatus(fiber.StatusNotFound)
 		return c.JSON(fiber.Map{
+			"field": "destination",
 			"error": err.Error(),
 		})
 	}

diff --git a/pkg/dataaggregator/source/journeyplanner/journeyplan.go b/pkg/dataaggregator/source/journeyplanner/journeyplan.go
@@ -23,7 +23,7 @@ func (s Source) JourneyPlanQuery(q query.JourneyPlan) (*ctdf.JourneyPlanResults,
 		Stop:          q.OriginStop,
 		Count:         q.Count * 10,
 		StartDateTime: q.StartDateTime,
-		Filter:        &bson.M{"path.destinationstopref": q.DestinationStop.PrimaryIdentifier},
+		Filter:        &bson.M{"path.destinationstopref": bson.M{"$in": append(q.DestinationStop.OtherIdentifiers, q.DestinationStop.PrimaryIdentifier)}},
 	})
 
 	if err != nil {