Amend 3cc0cea

Somehow I lost a part of this commit. Signed-off-by: Sebastian Wicki <[email protected]> Signed-off-by: Michi Mutsuzaki <[email protected]>
cilium · Apr 15, 2020 · 44fe5df · 44fe5df
1 parent 1c25780
commit 44fe5df
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pkg/server/local_observer.go b/pkg/server/local_observer.go
@@ -240,6 +240,7 @@ func (s *LocalObserverServer) GetFlows(
 }
 
 func getFlows(
+	ctx context.Context,
 	req *observer.GetFlowsRequest,
 	server observer.Observer_GetFlowsServer,
 	obs GRPCServer,
@@ -273,7 +274,7 @@ func getFlows(
 	}
 
 	for ; ; i++ {
-		flow, err := flowsReader.Next(server.Context())
+		flow, err := flowsReader.Next(ctx)
 		if err != nil {
 			if err == io.EOF {
 				return nil