From c03eb6f6645b20fae0fe0d26b4b0679d41dd9dd1 Mon Sep 17 00:00:00 2001 From: Zhen Ye Date: Fri, 13 Sep 2024 16:41:08 +0800 Subject: [PATCH] fix: streaming node consume blocks if recv message is too large (#36151) issue: #36081 Signed-off-by: chyezh --- .../streamingnode/client/handler/consumer/consumer_impl.go | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/internal/streamingnode/client/handler/consumer/consumer_impl.go b/internal/streamingnode/client/handler/consumer/consumer_impl.go index a879c39a29..2ba136c659 100644 --- a/internal/streamingnode/client/handler/consumer/consumer_impl.go +++ b/internal/streamingnode/client/handler/consumer/consumer_impl.go @@ -3,6 +3,7 @@ package consumer import ( "context" "io" + "math" "github.com/cockroachdb/errors" "go.uber.org/zap" @@ -45,7 +46,9 @@ func CreateConsumer( } // TODO: configurable or auto adjust grpc.MaxCallRecvMsgSize - streamClient, err := handlerClient.Consume(ctx, grpc.MaxCallRecvMsgSize(8388608)) + // The messages are always managed by milvus cluster, so the size of message shouldn't be controlled here + // to avoid infinitely blocks. + streamClient, err := handlerClient.Consume(ctx, grpc.MaxCallRecvMsgSize(math.MaxInt32)) if err != nil { return nil, err }