Skip to content

Commit 731799a

Browse files
Fix vstream call for new signature in vdiff framework
Signed-off-by: Rohit Nayak <rohit@planetscale.com>
1 parent d516f8b commit 731799a

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

go/vt/vttablet/tabletmanager/vdiff/framework_test.go

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -213,7 +213,7 @@ func shortCircuitTestAfterQuery(query string, dbClient *binlogplayer.MockDBClien
213213
dbClient.ExpectRequest("insert into _vt.vdiff_log(vdiff_id, message) values (1, 'Error: Short circuiting test')", singleRowAffected, nil)
214214
}
215215

216-
//--------------------------------------
216+
// --------------------------------------
217217
// Topos and tablets
218218

219219
// fakeTabletConn implement TabletConn interface. We only care about the
@@ -249,7 +249,7 @@ func (ftc *fakeTabletConn) VStream(ctx context.Context, request *binlogdatapb.VS
249249
if vstreamHook != nil {
250250
vstreamHook(ctx)
251251
}
252-
return vdiffenv.vse.Stream(ctx, request.Position, request.TableLastPKs, request.Filter, throttlerapp.VStreamerName, send)
252+
return vdiffenv.vse.Stream(ctx, request.Position, request.TableLastPKs, request.Filter, throttlerapp.VStreamerName, send, nil)
253253
}
254254

255255
// vstreamRowsHook allows you to do work just before calling VStreamRows.
@@ -283,7 +283,7 @@ func (ftc *fakeTabletConn) Close(ctx context.Context) error {
283283
return nil
284284
}
285285

286-
//--------------------------------------
286+
// --------------------------------------
287287
// Binlog Client to TabletManager
288288

289289
// fakeBinlogClient satisfies binlogplayer.Client.
@@ -344,7 +344,7 @@ func (bts *btStream) Recv() (*binlogdatapb.BinlogTransaction, error) {
344344
return nil, bts.ctx.Err()
345345
}
346346

347-
//--------------------------------------
347+
// --------------------------------------
348348
// DBCLient wrapper
349349

350350
func realDBClientFactory() binlogplayer.DBClient {
@@ -427,7 +427,7 @@ func (dbc *realDBClient) SupportsCapability(capability capabilities.FlavorCapabi
427427
return dbc.conn.SupportsCapability(capability)
428428
}
429429

430-
//----------------------------------------------
430+
// ----------------------------------------------
431431
// fakeTMClient
432432

433433
type fakeTMClient struct {

0 commit comments

Comments
 (0)