Bläddra i källkod

HADOOP-16840. AliyunOSS: getFileStatus throws FileNotFoundException in versioning bucket. Contributed by wujinhu.

Weiwei Yang 5 år sedan
förälder
incheckning
6dfe00c71e

+ 12 - 6
hadoop-tools/hadoop-aliyun/src/main/java/org/apache/hadoop/fs/aliyun/oss/AliyunOSSFileSystem.java

@@ -273,12 +273,18 @@ public class AliyunOSSFileSystem extends FileSystem {
     }
     }
     if (meta == null) {
     if (meta == null) {
       ObjectListing listing = store.listObjects(key, 1, null, false);
       ObjectListing listing = store.listObjects(key, 1, null, false);
-      if (CollectionUtils.isNotEmpty(listing.getObjectSummaries()) ||
-          CollectionUtils.isNotEmpty(listing.getCommonPrefixes())) {
-        return new OSSFileStatus(0, true, 1, 0, 0, qualifiedPath, username);
-      } else {
-        throw new FileNotFoundException(path + ": No such file or directory!");
-      }
+      do {
+        if (CollectionUtils.isNotEmpty(listing.getObjectSummaries()) ||
+            CollectionUtils.isNotEmpty(listing.getCommonPrefixes())) {
+          return new OSSFileStatus(0, true, 1, 0, 0, qualifiedPath, username);
+        } else if (listing.isTruncated()) {
+          listing = store.listObjects(key, 1000, listing.getNextMarker(),
+              false);
+        } else {
+          throw new FileNotFoundException(
+              path + ": No such file or directory!");
+        }
+      } while (true);
     } else if (objectRepresentsDirectory(key, meta.getContentLength())) {
     } else if (objectRepresentsDirectory(key, meta.getContentLength())) {
       return new OSSFileStatus(0, true, 1, 0, meta.getLastModified().getTime(),
       return new OSSFileStatus(0, true, 1, 0, meta.getLastModified().getTime(),
           qualifiedPath, username);
           qualifiedPath, username);

+ 23 - 0
hadoop-tools/hadoop-aliyun/src/test/java/org/apache/hadoop/fs/aliyun/oss/TestAliyunOSSFileSystemContract.java

@@ -31,6 +31,7 @@ import org.junit.Test;
 
 
 import java.io.FileNotFoundException;
 import java.io.FileNotFoundException;
 import java.io.IOException;
 import java.io.IOException;
+import java.util.Arrays;
 
 
 import static org.junit.Assert.assertFalse;
 import static org.junit.Assert.assertFalse;
 import static org.junit.Assert.assertTrue;
 import static org.junit.Assert.assertTrue;
@@ -96,6 +97,28 @@ public class TestAliyunOSSFileSystemContract
         UserGroupInformation.getCurrentUser().getShortUserName());
         UserGroupInformation.getCurrentUser().getShortUserName());
   }
   }
 
 
+  @Test
+  public void testGetFileStatusInVersioningBucket() throws Exception {
+    Path file = this.path("/test/hadoop/file");
+    for (int i = 1; i <= 30; ++i) {
+      this.createFile(new Path(file, "sub" + i));
+    }
+    assertTrue("File exists", this.fs.exists(file));
+    FileStatus fs = this.fs.getFileStatus(file);
+    assertEquals(fs.getOwner(),
+        UserGroupInformation.getCurrentUser().getShortUserName());
+    assertEquals(fs.getGroup(),
+        UserGroupInformation.getCurrentUser().getShortUserName());
+
+    AliyunOSSFileSystemStore store = ((AliyunOSSFileSystem)this.fs).getStore();
+    for (int i = 0; i < 29; ++i) {
+      store.deleteObjects(Arrays.asList("test/hadoop/file/sub" + i));
+    }
+
+    // HADOOP-16840, will throw FileNotFoundException without this fix
+    this.fs.getFileStatus(file);
+  }
+
   @Test
   @Test
   public void testDeleteSubdir() throws IOException {
   public void testDeleteSubdir() throws IOException {
     Path parentDir = this.path("/test/hadoop");
     Path parentDir = this.path("/test/hadoop");