fix union batch get

Signed-off-by: you06 <[email protected]>
pingcap · Jan 29, 2025 · 79fc917 · 79fc917
1 parent 6eaaf92
commit 79fc917
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pkg/store/driver/txn/batch_getter.go b/pkg/store/driver/txn/batch_getter.go
@@ -164,9 +164,9 @@ func (b tikvBatchBufferSnapshotGetter) Get(ctx context.Context, k []byte) ([]byt
 
 func (b tikvBatchBufferSnapshotGetter) BatchGet(ctx context.Context, keys [][]byte) (map[string][]byte, error) {
 	bufferValues := make(map[string][]byte, len(keys))
-	if b.tidbMiddleCache != nil {
+	if b.tidbSnapshotBuffer != nil {
 		for _, key := range keys {
-			val, err := b.tidbMiddleCache.Get(ctx, key)
+			val, err := b.tidbSnapshotBuffer.Get(ctx, key)
 			if err != nil {
 				if kv.IsErrNotFound(err) {
 					continue