From 44fe5df99be34f5dec9895dd8785ade3d5517981 Mon Sep 17 00:00:00 2001 From: Sebastian Wicki Date: Mon, 6 Apr 2020 12:28:03 +0200 Subject: [PATCH] Amend 3cc0cea40bb72e1f7d00878af42c160e8e4660f3 Somehow I lost a part of this commit. Signed-off-by: Sebastian Wicki Signed-off-by: Michi Mutsuzaki --- pkg/server/local_observer.go | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pkg/server/local_observer.go b/pkg/server/local_observer.go index cf8b173b8..f28cabbe2 100644 --- 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