Search for logs using predictor id for realtime api (#1803)

vishalbollu · web-flow · commit fc06dae077da · 2021-01-19T11:11:31.000-05:00
diff --git a/pkg/operator/endpoints/logs.go b/pkg/operator/endpoints/logs.go
@@ -50,6 +50,7 @@ func ReadLogs(w http.ResponseWriter, r *http.Request) {
 	}
 
 	deploymentID := deployedResource.VirtualService.Labels["deploymentID"]
+	predictorID := deployedResource.VirtualService.Labels["predictorID"]
 
 	upgrader := websocket.Upgrader{}
 	socket, err := upgrader.Upgrade(w, r, nil)
@@ -59,5 +60,5 @@ func ReadLogs(w http.ResponseWriter, r *http.Request) {
 	}
 	defer socket.Close()
 
-	operator.StreamLogsFromRandomPod(map[string]string{"apiName": apiName, "deploymentID": deploymentID}, socket)
+	operator.StreamLogsFromRandomPod(map[string]string{"apiName": apiName, "deploymentID": deploymentID, "predictorID": predictorID}, socket)
 }

Original file line number	Diff line number	Diff line change
`@@ -50,6 +50,7 @@ func ReadLogs(w http.ResponseWriter, r *http.Request) {`
`50`	`50`	`}`
`51`	`51`
`52`	`52`	`deploymentID := deployedResource.VirtualService.Labels["deploymentID"]`
	`53`	`+ predictorID := deployedResource.VirtualService.Labels["predictorID"]`
`53`	`54`
`54`	`55`	`upgrader := websocket.Upgrader{}`
`55`	`56`	`socket, err := upgrader.Upgrade(w, r, nil)`
`@@ -59,5 +60,5 @@ func ReadLogs(w http.ResponseWriter, r *http.Request) {`
`59`	`60`	`}`
`60`	`61`	`defer socket.Close()`
`61`	`62`
`62`		`- operator.StreamLogsFromRandomPod(map[string]string{"apiName": apiName, "deploymentID": deploymentID}, socket)`
	`63`	`+ operator.StreamLogsFromRandomPod(map[string]string{"apiName": apiName, "deploymentID": deploymentID, "predictorID": predictorID}, socket)`
`63`	`64`	`}`