[#5019] feat: (hadoop-catalog): Add a framework to support multi-storage in a pluginized manner for fileset catalog

.gitignore

@@ -53,3 +53,7 @@ include clients/client-python/.gitignore
 **/metastore_db
 **/spark-warehouse
 derby.log
+
+web/node_modules
+web/dist
+web/.next

catalogs/catalog-hadoop/src/main/java/org/apache/gravitino/catalog/hadoop/FileSystemProvider.java

@@ -0,0 +1,53 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *  http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+package org.apache.gravitino.catalog.hadoop;
+
+import java.io.IOException;
+import org.apache.hadoop.conf.Configuration;
+import org.apache.hadoop.fs.FileSystem;
+import org.apache.hadoop.fs.Path;
+
+public interface FileSystemProvider {
+
+  /**
+   * Get the FileSystem instance according to the configuration and the path.
+   *
+   * <p>Compared to the FileSystem.get method, this method allows the provider to create a
+   * FileSystem instance with a specific configuration and path and do further initialization if
+   * needed.
+   *
+   * <p>For example, we can check endpoint configurations for S3AFileSystem, or set the default one.
+   *
+   * @param configuration The configuration.
+   * @param path The path.
+   * @return The FileSystem instance.
+   * @throws IOException If the FileSystem instance cannot be created.
+   */
+  FileSystem getFileSystem(Configuration configuration, Path path) throws IOException;
+
+  /**
+   * Get the scheme of this FileSystem provider.
+   *
+   * @return The scheme of this FileSystem provider.
+   */
+  default String getScheme() {
+    return "file";
+  }
+}

catalogs/catalog-hadoop/src/main/java/org/apache/gravitino/catalog/hadoop/HadoopCatalogOperations.java

@@ -44,6 +44,8 @@
 import org.apache.gravitino.audit.CallerContext;
 import org.apache.gravitino.audit.FilesetAuditConstants;
 import org.apache.gravitino.audit.FilesetDataOperation;
+import org.apache.gravitino.catalog.hadoop.fs.HDFSFileSystemProvider;
+import org.apache.gravitino.catalog.hadoop.fs.LocalFileSystemProvider;
 import org.apache.gravitino.connector.CatalogInfo;
 import org.apache.gravitino.connector.CatalogOperations;
 import org.apache.gravitino.connector.HasPropertyMetadata;
@@ -77,6 +79,7 @@ public class HadoopCatalogOperations implements CatalogOperations, SupportsSchem
   private static final String SLASH = "/";
 
   private static final Logger LOG = LoggerFactory.getLogger(HadoopCatalogOperations.class);
+  public static final Map<String, FileSystemProvider> FILE_SYSTEM_PROVIDERS = Maps.newHashMap();
 
   private final EntityStore store;
 
@@ -90,6 +93,14 @@ public class HadoopCatalogOperations implements CatalogOperations, SupportsSchem
 
   private CatalogInfo catalogInfo;
 
+  static {
+    FileSystemProvider localFileSystemProvider = new LocalFileSystemProvider();
+    FileSystemProvider hdfsFileSystemProvider = new HDFSFileSystemProvider();
+
+    FILE_SYSTEM_PROVIDERS.put(localFileSystemProvider.getScheme(), localFileSystemProvider);
+    FILE_SYSTEM_PROVIDERS.put(hdfsFileSystemProvider.getScheme(), hdfsFileSystemProvider);
+  }
+
   HadoopCatalogOperations(EntityStore store) {
     this.store = store;
   }
@@ -119,32 +130,57 @@ public void initialize(
       Map<String, String> config, CatalogInfo info, HasPropertyMetadata propertiesMetadata)
       throws RuntimeException {
     this.propertiesMetadata = propertiesMetadata;
+    this.catalogInfo = info;
+
     // Initialize Hadoop Configuration.
     this.conf = config;
-    this.hadoopConf = new Configuration();
-    this.catalogInfo = info;
+
+    hadoopConf = new Configuration();
     Map<String, String> bypassConfigs =
-        config.entrySet().stream()
+        conf.entrySet().stream()
             .filter(e -> e.getKey().startsWith(CATALOG_BYPASS_PREFIX))
             .collect(
                 Collectors.toMap(
                     e -> e.getKey().substring(CATALOG_BYPASS_PREFIX.length()),
                     Map.Entry::getValue));
     bypassConfigs.forEach(hadoopConf::set);
 
+    initPluginFileSystem(conf);
+
     String catalogLocation =
         (String)
             propertiesMetadata
                 .catalogPropertiesMetadata()
                 .getOrDefault(config, HadoopCatalogPropertiesMetadata.LOCATION);
-    conf.forEach(hadoopConf::set);
-
     this.catalogStorageLocation =
         StringUtils.isNotBlank(catalogLocation)
             ? Optional.of(catalogLocation).map(Path::new)
             : Optional.empty();
   }
 
+  private void initPluginFileSystem(Map<String, String> config) {
+    String fileSystemProviders =
+        (String)
+            propertiesMetadata
+                .catalogPropertiesMetadata()
+                .getOrDefault(config, HadoopCatalogPropertiesMetadata.FILESYSTEM_PROVIDER);
+
+    if (StringUtils.isNotBlank(fileSystemProviders)) {
+      String[] providers = fileSystemProviders.split(",");
+      for (String provider : providers) {
+        try {
+          FileSystemProvider fileSystemProvider =
+              (FileSystemProvider)
+                  Class.forName(provider.trim()).getDeclaredConstructor().newInstance();
+          FILE_SYSTEM_PROVIDERS.put(fileSystemProvider.getScheme(), fileSystemProvider);
+        } catch (Exception e) {
+          throw new GravitinoRuntimeException(
+              e, "Failed to initialize file system provider: %s", provider);
+        }
+      }
+    }
+  }
+
   @Override
   public NameIdentifier[] listFilesets(Namespace namespace) throws NoSuchSchemaException {
     try {
@@ -236,7 +272,8 @@ public Fileset createFileset(
     try {
       // formalize the path to avoid path without scheme, uri, authority, etc.
       filesetPath = formalizePath(filesetPath, hadoopConf);
-      FileSystem fs = filesetPath.getFileSystem(hadoopConf);
+
+      FileSystem fs = getFileSystem(filesetPath, hadoopConf);
       if (!fs.exists(filesetPath)) {
         if (!fs.mkdirs(filesetPath)) {
           throw new RuntimeException(
@@ -339,7 +376,7 @@ public boolean dropFileset(NameIdentifier ident) {
 
       // For managed fileset, we should delete the related files.
       if (filesetEntity.filesetType() == Fileset.Type.MANAGED) {
-        FileSystem fs = filesetPath.getFileSystem(hadoopConf);
+        FileSystem fs = getFileSystem(filesetPath, hadoopConf);
         if (fs.exists(filesetPath)) {
           if (!fs.delete(filesetPath, true)) {
             LOG.warn("Failed to delete fileset {} location {}", ident, filesetPath);
@@ -459,7 +496,7 @@ public Schema createSchema(NameIdentifier ident, String comment, Map<String, Str
     Path schemaPath = getSchemaPath(ident.name(), properties);
     if (schemaPath != null) {
       try {
-        FileSystem fs = schemaPath.getFileSystem(hadoopConf);
+        FileSystem fs = getFileSystem(schemaPath, hadoopConf);
         if (!fs.exists(schemaPath)) {
           if (!fs.mkdirs(schemaPath)) {
             // Fail the operation when failed to create the schema path.
@@ -577,8 +614,7 @@ public boolean dropSchema(NameIdentifier ident, boolean cascade) throws NonEmpty
       if (schemaPath == null) {
         return false;
       }
-
-      FileSystem fs = schemaPath.getFileSystem(hadoopConf);
+      FileSystem fs = getFileSystem(schemaPath, hadoopConf);
       // Nothing to delete if the schema path does not exist.
       if (!fs.exists(schemaPath)) {
         return false;
@@ -731,7 +767,7 @@ private boolean hasCallerContext() {
   private boolean checkSingleFile(Fileset fileset) {
     try {
       Path locationPath = new Path(fileset.storageLocation());
-      return locationPath.getFileSystem(hadoopConf).getFileStatus(locationPath).isFile();
+      return getFileSystem(locationPath, hadoopConf).getFileStatus(locationPath).isFile();
     } catch (FileNotFoundException e) {
       // We should always return false here, same with the logic in `FileSystem.isFile(Path f)`.
       return false;
@@ -742,4 +778,27 @@ private boolean checkSingleFile(Fileset fileset) {
           fileset.name());
     }
   }
+
+  static FileSystem getFileSystem(Path path, Configuration conf) throws IOException {
+    if (path == null) {
+      String defaultFS = conf.get("fs.defaultFS");
+      if (defaultFS != null) {
+        path = new Path(defaultFS);
+      }
+    }
+
+    String scheme;
+    if (path == null || path.toUri().getScheme() == null) {
+      scheme = "file";
+    } else {
+      scheme = path.toUri().getScheme();
+    }
+
+    FileSystemProvider provider = FILE_SYSTEM_PROVIDERS.get(scheme);
+    if (provider == null) {
+      throw new IllegalArgumentException("Unsupported scheme: " + scheme);
+    }
+
+    return provider.getFileSystem(conf, path);
+  }
 }

catalogs/catalog-hadoop/src/main/java/org/apache/gravitino/catalog/hadoop/HadoopCatalogPropertiesMetadata.java

@@ -34,6 +34,13 @@ public class HadoopCatalogPropertiesMetadata extends BaseCatalogPropertiesMetada
   // If not, users have to specify the storage location in the Schema or Fileset level.
   public static final String LOCATION = "location";
 
+  /**
+   * The implementation class name of the {@link FileSystemProvider} to be used by the catalog.
+   * Gravitino supports LocalFileSystem and HDFS by default. Users can implement their own by
+   * extending {@link FileSystemProvider} and specify the class name here.
+   */
+  public static final String FILESYSTEM_PROVIDER = "filesystem.providers";
+
   private static final Map<String, PropertyEntry<?>> HADOOP_CATALOG_PROPERTY_ENTRIES =
       ImmutableMap.<String, PropertyEntry<?>>builder()
           .put(
@@ -44,6 +51,14 @@ public class HadoopCatalogPropertiesMetadata extends BaseCatalogPropertiesMetada
                   false /* immutable */,
                   null,
                   false /* hidden */))
+          .put(
+              FILESYSTEM_PROVIDER,
+              PropertyEntry.stringOptionalPropertyEntry(
+                  FILESYSTEM_PROVIDER,
+                  "The file system provider class name",
+                  false /* immutable */,
+                  null,
+                  false /* hidden */))
           // The following two are about authentication.
           .putAll(KerberosConfig.KERBEROS_PROPERTY_ENTRIES)
           .putAll(AuthenticationConfig.AUTHENTICATION_PROPERTY_ENTRIES)

catalogs/catalog-hadoop/src/main/java/org/apache/gravitino/catalog/hadoop/fs/HDFSFileSystemProvider.java

@@ -0,0 +1,80 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *  http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+package org.apache.gravitino.catalog.hadoop.fs;
+
+import java.io.IOException;
+import java.net.URI;
+import org.apache.commons.lang3.StringUtils;
+import org.apache.gravitino.catalog.hadoop.FileSystemProvider;
+import org.apache.hadoop.conf.Configuration;
+import org.apache.hadoop.fs.FileSystem;
+import org.apache.hadoop.fs.Path;
+
+public class HDFSFileSystemProvider implements FileSystemProvider {
+
+  @Override
+  public FileSystem getFileSystem(Configuration configuration, Path path) throws IOException {
+    Path fileSystemPath = path;
+    if (fileSystemPath == null) {
+      String pathString = configuration.get("fs.defaultFS");
+      if (StringUtils.isNotBlank(pathString)) {
+        fileSystemPath = new Path(pathString);
+      }
+    }
+
+    if (fileSystemPath == null) {
+      throw new IllegalArgumentException("The path should be specified.");
+    }
+
+    URI uri = fileSystemPath.toUri();
+    if (uri.getScheme() != null && !uri.getScheme().equals("hdfs")) {
+      throw new IllegalArgumentException("The path should be a HDFS path.");
+    }
+
+    // Should we call DistributedFileSystem to create file system instance explicitly? If we
+    // explicitly create a HDFS file system here, we can't reuse the file system cache in the
+    // FileSystem class.
+    String impl = configuration.get("fs.hdfs.impl");
+    if (impl == null) {
+      configuration.set("fs.hdfs.impl", "org.apache.hadoop.hdfs.DistributedFileSystem");
+    } else {
+      if (!impl.equals("org.apache.hadoop.hdfs.DistributedFileSystem")) {
+        throw new IllegalArgumentException(
+            "The HDFS file system implementation class should be 'org.apache.hadoop.hdfs.DistributedFileSystem'.");
+      }
+    }
+
+    try {
+      if (HDFSFileSystemProvider.class.getClassLoader().loadClass(configuration.get("fs.hdfs.impl"))
+          == null) {
+        throw new IllegalArgumentException(
+            "The HDFS file system implementation class is not found.");
+      }
+    } catch (ClassNotFoundException e) {
+      throw new IllegalArgumentException("The HDFS file system implementation class is not found.");
+    }
+
+    return FileSystem.get(uri, configuration);
+  }
+
+  @Override
+  public String getScheme() {
+    return "hdfs";
+  }
+}

catalogs/catalog-hadoop/src/main/java/org/apache/gravitino/catalog/hadoop/fs/LocalFileSystemProvider.java

@@ -0,0 +1,46 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *  http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+package org.apache.gravitino.catalog.hadoop.fs;
+
+import java.io.IOException;
+import org.apache.commons.lang3.StringUtils;
+import org.apache.gravitino.catalog.hadoop.FileSystemProvider;
+import org.apache.hadoop.conf.Configuration;
+import org.apache.hadoop.fs.FileSystem;
+import org.apache.hadoop.fs.Path;
+
+public class LocalFileSystemProvider implements FileSystemProvider {
+
+  @Override
+  public FileSystem getFileSystem(Configuration configuration, Path path) throws IOException {
+    Path fileSystemPath = path;
+    if (fileSystemPath == null) {
+      String pathString = configuration.get("fs.defaultFS");
+      if (StringUtils.isNotBlank(pathString)) {
+        fileSystemPath = new Path(pathString);
+      }
+    }
+
+    if (fileSystemPath == null) {
+      fileSystemPath = new Path("file:///");
+    }
+
+    return FileSystem.get(fileSystemPath.toUri(), configuration);
+  }
+}