forked from apache/seatunnel
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add OSS/S3 to cluster-mode type apache#4621
- Loading branch information
1 parent
c8a5d52
commit 34a827e
Showing
7 changed files
with
353 additions
and
4 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
66 changes: 66 additions & 0 deletions
66
...main/java/org/apache/seatunnel/engine/imap/storage/file/config/AbstractConfiguration.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,66 @@ | ||
| /* | ||
| * 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.seatunnel.engine.imap.storage.file.config; | ||
|
|
||
| import org.apache.seatunnel.engine.imap.storage.api.exception.IMapStorageException; | ||
|
|
||
| import org.apache.hadoop.conf.Configuration; | ||
|
|
||
| import java.util.Map; | ||
|
|
||
| public abstract class AbstractConfiguration { | ||
|
|
||
| protected static final String HDFS_IMPL_KEY = "impl"; | ||
|
|
||
| /** | ||
| * check the configuration keys | ||
| * | ||
| * @param config configuration | ||
| * @param keys keys | ||
| */ | ||
| void checkConfiguration(Map<String, Object> config, String... keys) { | ||
| for (String key : keys) { | ||
| if (!config.containsKey(key) || null == config.get(key)) { | ||
| throw new IllegalArgumentException(key + " is required"); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| public abstract Configuration buildConfiguration(Map<String, Object> config) | ||
| throws IMapStorageException; | ||
|
|
||
| /** | ||
| * set extra options for configuration | ||
| * | ||
| * @param hadoopConf | ||
| * @param config | ||
| * @param prefix | ||
| */ | ||
| void setExtraConfiguration( | ||
| Configuration hadoopConf, Map<String, Object> config, String prefix) { | ||
| config.forEach( | ||
| (k, v) -> { | ||
| if (k.startsWith(prefix)) { | ||
| hadoopConf.set(k, String.valueOf(v)); | ||
| } | ||
| }); | ||
| } | ||
| } |
46 changes: 46 additions & 0 deletions
46
...src/main/java/org/apache/seatunnel/engine/imap/storage/file/config/FileConfiguration.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -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.seatunnel.engine.imap.storage.file.config; | ||
|
|
||
| public enum FileConfiguration { | ||
| HDFS("hdfs", new HdfsConfiguration()), | ||
| S3("s3", new S3Configuration()), | ||
| OSS("oss", new OssConfiguration()); | ||
|
|
||
| /** file system type */ | ||
| private final String name; | ||
|
|
||
| /** file system configuration */ | ||
| private final AbstractConfiguration configuration; | ||
|
|
||
| FileConfiguration(String name, AbstractConfiguration configuration) { | ||
| this.name = name; | ||
| this.configuration = configuration; | ||
| } | ||
|
|
||
| public AbstractConfiguration getConfiguration(String name) { | ||
| return configuration; | ||
| } | ||
|
|
||
| public String getName() { | ||
| return name; | ||
| } | ||
| } |
91 changes: 91 additions & 0 deletions
91
...src/main/java/org/apache/seatunnel/engine/imap/storage/file/config/HdfsConfiguration.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,91 @@ | ||
| /* | ||
| * 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.seatunnel.engine.imap.storage.file.config; | ||
|
|
||
| import org.apache.seatunnel.engine.imap.storage.api.exception.IMapStorageException; | ||
|
|
||
| import org.apache.commons.lang3.StringUtils; | ||
| import org.apache.hadoop.conf.Configuration; | ||
| import org.apache.hadoop.security.UserGroupInformation; | ||
|
|
||
| import java.io.IOException; | ||
| import java.util.Map; | ||
|
|
||
| import static org.apache.hadoop.fs.FileSystem.FS_DEFAULT_NAME_KEY; | ||
|
|
||
| public class HdfsConfiguration extends AbstractConfiguration { | ||
| private static final String HDFS_DEF_FS_NAME = "fs.defaultFS"; | ||
| private static final String KERBEROS_PRINCIPAL = "kerberosPrincipal"; | ||
| private static final String KERBEROS_KEYTAB_FILE_PATH = "kerberosKeytabFilePath"; | ||
| private static final String HADOOP_SECURITY_AUTHENTICATION_KEY = | ||
| "hadoop.security.authentication"; | ||
| private static final String KERBEROS_KEY = "kerberos"; | ||
| private static final String HDFS_IMPL = "org.apache.hadoop.hdfs.DistributedFileSystem"; | ||
| private static final String HDFS_IMPL_KEY = "fs.hdfs.impl"; | ||
|
|
||
| @Override | ||
| public Configuration buildConfiguration(Map<String, Object> config) | ||
| throws IMapStorageException { | ||
| checkConfiguration(config, HDFS_DEF_FS_NAME); | ||
| Configuration hadoopConf = new Configuration(); | ||
|
|
||
| hadoopConf.set(HDFS_DEF_FS_NAME, String.valueOf(config.get(HDFS_DEF_FS_NAME))); | ||
|
|
||
| hadoopConf.set(HDFS_IMPL_KEY, HDFS_IMPL); | ||
| hadoopConf.set(FS_DEFAULT_NAME_KEY, String.valueOf(config.get(FS_DEFAULT_NAME_KEY))); | ||
| if (config.containsKey(KERBEROS_PRINCIPAL) | ||
| && config.containsKey(KERBEROS_KEYTAB_FILE_PATH)) { | ||
| String kerberosPrincipal = String.valueOf(config.get(KERBEROS_PRINCIPAL)); | ||
| String kerberosKeytabFilePath = String.valueOf(config.get(KERBEROS_KEYTAB_FILE_PATH)); | ||
| if (StringUtils.isNotBlank(kerberosPrincipal) | ||
| && StringUtils.isNotBlank(kerberosKeytabFilePath)) { | ||
| hadoopConf.set(HADOOP_SECURITY_AUTHENTICATION_KEY, KERBEROS_KEY); | ||
| authenticateKerberos(kerberosPrincipal, kerberosKeytabFilePath, hadoopConf); | ||
| } | ||
| } | ||
| // todo support other hdfs optional config keys | ||
| return hadoopConf; | ||
| } | ||
|
|
||
| /** | ||
| * Authenticate kerberos | ||
| * | ||
| * @param kerberosPrincipal | ||
| * @param kerberosKeytabFilePath | ||
| * @param hdfsConf | ||
| * @throws IMapStorageException | ||
| */ | ||
| private void authenticateKerberos( | ||
| String kerberosPrincipal, String kerberosKeytabFilePath, Configuration hdfsConf) | ||
| throws IMapStorageException { | ||
| UserGroupInformation.setConfiguration(hdfsConf); | ||
| try { | ||
| UserGroupInformation.loginUserFromKeytab(kerberosPrincipal, kerberosKeytabFilePath); | ||
| } catch (IOException e) { | ||
| throw new IMapStorageException( | ||
| "Failed to login user from keytab : " | ||
| + kerberosKeytabFilePath | ||
| + " and kerberos principal : " | ||
| + kerberosPrincipal, | ||
| e); | ||
| } | ||
| } | ||
| } |
48 changes: 48 additions & 0 deletions
48
.../src/main/java/org/apache/seatunnel/engine/imap/storage/file/config/OssConfiguration.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,48 @@ | ||
| /* | ||
| * 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.seatunnel.engine.imap.storage.file.config; | ||
|
|
||
| import org.apache.seatunnel.engine.imap.storage.api.exception.IMapStorageException; | ||
|
|
||
| import org.apache.hadoop.conf.Configuration; | ||
|
|
||
| import java.util.Map; | ||
|
|
||
| import static org.apache.hadoop.fs.FileSystem.FS_DEFAULT_NAME_KEY; | ||
|
|
||
| public class OssConfiguration extends AbstractConfiguration { | ||
| public static final String OSS_BUCKET_KEY = "oss.bucket"; | ||
| private static final String OSS_IMPL_KEY = "fs.oss.impl"; | ||
| private static final String HDFS_OSS_IMPL = | ||
| "org.apache.hadoop.fs.aliyun.oss.AliyunOSSFileSystem"; | ||
| private static final String OSS_KEY = "fs.oss."; | ||
|
|
||
| @Override | ||
| public Configuration buildConfiguration(Map<String, Object> config) | ||
| throws IMapStorageException { | ||
| checkConfiguration(config, OSS_BUCKET_KEY); | ||
| Configuration hadoopConf = new Configuration(); | ||
| hadoopConf.set(FS_DEFAULT_NAME_KEY, String.valueOf(config.get(OSS_BUCKET_KEY))); | ||
| hadoopConf.set(OSS_IMPL_KEY, HDFS_OSS_IMPL); | ||
| setExtraConfiguration(hadoopConf, config, OSS_KEY); | ||
| return hadoopConf; | ||
| } | ||
| } |
Oops, something went wrong.