chore(tempodb): Adding max size for received messages to 10MB (#4026)

kubeshop · Sep 20, 2024 · 0253f9f · 0253f9f
1 parent 4147cb3
commit 0253f9f
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/agent/tracedb/tempodb.go b/agent/tracedb/tempodb.go
@@ -90,9 +90,12 @@ func grpcGetTraceByID(ctx context.Context, traceID string, conn *grpc.ClientConn
 		return traces.Trace{}, err
 	}
 
-	resp, err := query.FindTraceByID(ctx, &tempopb.TraceByIDRequest{
+	// This is a temporary solution while we figure out how to paginate the traces from Tempo
+	maxSizeOption := grpc.MaxCallRecvMsgSize(1024 * 1024 * 10) // 10MB
+
+	resp, err := query.FindTraceByID(context.Background(), &tempopb.TraceByIDRequest{
 		TraceID: []byte(trID[:]),
-	})
+	}, maxSizeOption)
 	if err != nil {
 		return traces.Trace{}, handleError(err)
 	}