Added missing packing plan handler in heron-tracker (#3783)

apache · Feb 28, 2022 · 1a54477 · 1a54477
1 parent ebd7cea
commit 1a54477
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/heron/tools/tracker/src/python/routers/topologies.py b/heron/tools/tracker/src/python/routers/topologies.py
@@ -33,6 +33,7 @@
     TopologyInfoExecutionState,
     TopologyInfoLogicalPlan,
     TopologyInfoMetadata,
+    TopologyInfoPackingPlan,
     TopologyInfoPhysicalPlan,
     TopologyInfoSchedulerLocation,
 )
@@ -114,6 +115,15 @@ async def get_topology_config(
   topology_info = topology.info
   return topology_info.physical_plan.config
 
+@router.get("/packingplan", response_model=TopologyInfoPackingPlan)
+async def get_topology_packing_plan(
+  cluster: str,
+  environ: str,
+  topology: str,
+  role: Optional[str] = Query(None, deprecated=True),
+):
+  topology = state.tracker.get_topology(cluster, role, environ, topology)
+  return topology.info.packing_plan
 
 @router.get("/physicalplan", response_model=TopologyInfoPhysicalPlan)
 async def get_topology_physical_plan(