OAK-8928: Retry reading the graph and binary references in Segment St…

…ore migrator

git-svn-id: https://svn.apache.org/repos/asf/jackrabbit/oak/trunk@1874576 13f79535-47bb-0310-9956-ffa450edef68

oak-segment-azure/src/main/java/org/apache/jackrabbit/oak/segment/azure/tool/SegmentStoreMigrator.java

@@ -193,17 +193,25 @@ private void migrateSegments(SegmentArchiveReader reader, SegmentArchiveWriter w
         }
     }
 
-    private void migrateBinaryRef(SegmentArchiveReader reader, SegmentArchiveWriter writer) throws IOException {
-        Buffer binaryReferences = reader.getBinaryReferences();
+    private void migrateBinaryRef(SegmentArchiveReader reader, SegmentArchiveWriter writer) throws IOException, ExecutionException, InterruptedException {
+        Future<Buffer> future = executor.submit(() -> runWithRetry(() -> reader.getBinaryReferences(), 16, 5));
+        Buffer binaryReferences = future.get();
         if (binaryReferences != null) {
             byte[] array = fetchByteArray(binaryReferences);
             writer.writeBinaryReferences(array);
         }
     }
 
-    private void migrateGraph(SegmentArchiveReader reader, SegmentArchiveWriter writer) throws IOException {
-        if (reader.hasGraph()) {
-            Buffer graph = reader.getGraph();
+    private void migrateGraph(SegmentArchiveReader reader, SegmentArchiveWriter writer) throws IOException, ExecutionException, InterruptedException {
+        Future<Buffer> future = executor.submit(() -> runWithRetry(() -> {
+            if (reader.hasGraph()) {
+                return reader.getGraph();
+            } else {
+                return null;
+            }
+        }, 16, 5));
+        Buffer graph = future.get();
+        if (graph != null) {
             byte[] array = fetchByteArray(graph);
             writer.writeGraph(array);
         }