Handle --vreplication_experimental_flags options properly

Signed-off-by: Matt Lord <[email protected]>
vitessio · Dec 24, 2024 · 28adbec · 28adbec
1 parent 059d01a
commit 28adbec
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 13 deletions.
diff --git a/go/vt/vttablet/tabletmanager/vreplication/table_plan_partial.go b/go/vt/vttablet/tabletmanager/vreplication/table_plan_partial.go
@@ -26,7 +26,6 @@ import (
 	"vitess.io/vitess/go/vt/log"
 	"vitess.io/vitess/go/vt/sqlparser"
 	"vitess.io/vitess/go/vt/vterrors"
-	vttablet "vitess.io/vitess/go/vt/vttablet/common"
 )
 
 // isBitSet returns true if the bit at index is set
@@ -47,13 +46,11 @@ func setBit(data []byte, index int, value bool) {
 }
 
 func (tp *TablePlan) isPartial(rowChange *binlogdatapb.RowChange) bool {
-	if (tp.WorkflowConfig.ExperimentalFlags /**/ & /**/ vttablet.VReplicationExperimentalFlagAllowNoBlobBinlogRowImage) == 0 ||
-		rowChange.DataColumns == nil ||
-		rowChange.DataColumns.Count == 0 {
-
+	if rowChange == nil {
 		return false
 	}
-	return true
+	return (rowChange.DataColumns != nil && rowChange.DataColumns.Count > 0) ||
+		(rowChange.JsonPartialValues != nil && rowChange.JsonPartialValues.Count > 0)
 }
 
 func (tpb *tablePlanBuilder) generatePartialValuesPart(buf *sqlparser.TrackedBuffer, bvf *bindvarFormatter, dataColumns *binlogdatapb.RowChange_Bitmap) *sqlparser.ParsedQuery {

diff --git a/go/vt/vttablet/tabletmanager/vreplication/vplayer_flaky_test.go b/go/vt/vttablet/tabletmanager/vreplication/vplayer_flaky_test.go
@@ -1762,11 +1762,21 @@ func TestPlayerPartialImages(t *testing.T) {
 			execStatements(t, []string{tc.input})
 			var want qh.ExpectationSequencer
 			if tc.error != "" {
-				want = qh.Expect(
-					"rollback",
-				).Then(qh.Immediately(
-					fmt.Sprintf("/update _vt.vreplication set message=.*%s.*", tc.error),
-				))
+				if vttablet.DefaultVReplicationConfig.ExperimentalFlags&vttablet.VReplicationExperimentalFlagVPlayerBatching == 0 {
+					want = qh.Expect(
+						"begin",
+						"delete from dst where id=3",
+						"rollback",
+					).Then(qh.Immediately(
+						fmt.Sprintf("/update _vt.vreplication set message=.*%s.*", tc.error),
+					))
+				} else {
+					want = qh.Expect(
+						"rollback",
+					).Then(qh.Immediately(
+						fmt.Sprintf("/update _vt.vreplication set message=.*%s.*", tc.error),
+					))
+				}
 				expectDBClientQueries(t, want)
 			} else {
 				want = qh.Expect(

diff --git a/go/vt/vttablet/tabletserver/vstreamer/vstreamer.go b/go/vt/vttablet/tabletserver/vstreamer/vstreamer.go
@@ -1031,8 +1031,8 @@ func (vs *vstreamer) processRowEvent(vevents []*binlogdatapb.VEvent, plan *strea
 		}
 		if afterOK {
 			rowChange.After = sqltypes.RowToProto3(afterValues)
-			if (vs.config.ExperimentalFlags /**/ & /**/ vttablet.VReplicationExperimentalFlagAllowNoBlobBinlogRowImage != 0) &&
-				(partial || row.JSONPartialValues.Count() > 0) {
+			if ((vs.config.ExperimentalFlags /**/ & /**/ vttablet.VReplicationExperimentalFlagAllowNoBlobBinlogRowImage != 0) &&
+				partial) || (row.JSONPartialValues.Count() > 0) {
 
 				rowChange.DataColumns = &binlogdatapb.RowChange_Bitmap{
 					Count: int64(rows.DataColumns.Count()),