lightning: fix parquet import always fail (#49391)

close #49368
pingcap · Dec 12, 2023 · 5bd0267 · 5bd0267
1 parent 1b199c2
commit 5bd0267
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/br/pkg/lightning/importer/table_import.go b/br/pkg/lightning/importer/table_import.go
@@ -771,8 +771,9 @@ ChunkLoop:
 
 		if chunk.FileMeta.Type == mydump.SourceTypeParquet {
 			// TODO: use the compressed size of the chunk to conduct memory control
-			_, err = getChunkCompressedSizeForParquet(ctx, chunk, rc.store)
-			return nil, errors.Trace(err)
+			if _, err = getChunkCompressedSizeForParquet(ctx, chunk, rc.store); err != nil {
+				return nil, errors.Trace(err)
+			}
 		}
 
 		restoreWorker := rc.regionWorkers.Apply()