turn off raw packets when consolidation is on

Signed-off-by: Harshit Gangal <[email protected]>
vitessio · Nov 15, 2024 · 881851f · 881851f
1 parent 05b80b8
commit 881851f
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/go/vt/vttablet/tabletserver/query_executor.go b/go/vt/vttablet/tabletserver/query_executor.go
@@ -102,12 +102,12 @@ func allocStreamResult() *sqltypes.Result {
 	return streamResultPool.Get().(*sqltypes.Result)
 }
 
-func (qre *QueryExecutor) shouldConsolidate() bool {
-	// TODO (harshit): This is a temporary implementation to test the feature.
-	// This should ideally work with consolidator.
-	if qre.options.GetRawMysqlPackets() {
-		return false
-	}
+func (qre *QueryExecutor) shouldConsolidate() (consolidate bool) {
+	defer func() {
+		if consolidate && qre.options != nil {
+			qre.options.RawMysqlPackets = false
+		}
+	}()
 	co := qre.options.GetConsolidator()
 	switch co {
 	case querypb.ExecuteOptions_CONSOLIDATOR_DISABLED: