Apache
/
druid
mirror of https://github.com/apache/druid.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix hdfs initialization issue (#14276) 

* fix hdfs initialization issue

* add PR

* remove conf settings

* Improve comments

* move hdfs storage validation to start handler

* restore exception
This commit is contained in:
panhongan 2023-05-31 03:41:54 +08:00 committed by GitHub
parent 70c06fc0e1
commit c244c3de53
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 70 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

68
extensions-core/hdfs-storage/src/main/java/org/apache/druid/storage/hdfs/HdfsStorageAvailabilityChecker.java Normal file
View File

@ -0,0 +1,68 @@
/*
* 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.druid.storage.hdfs;
import com.google.inject.Inject;
import org.apache.druid.guice.Hdfs;
import org.apache.druid.guice.ManageLifecycle;
import org.apache.druid.java.util.common.ISE;
import org.apache.druid.java.util.common.lifecycle.LifecycleStart;
import org.apache.druid.java.util.common.lifecycle.LifecycleStop;
import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.fs.FileSystem;
import org.apache.hadoop.fs.Path;
import java.io.IOException;
@ManageLifecycle
public class HdfsStorageAvailabilityChecker
{
private final Configuration hadoopConf;
@Inject
public HdfsStorageAvailabilityChecker(@Hdfs Configuration hadoopConf)
{
this.hadoopConf = hadoopConf;
}
@LifecycleStart
public void checkHdfsAvailability()
{
try {
// If cache is enabled, need to check the FileSystem object by access hdfs because FileSystem object will be used later, like in push stage.
// If FileSystem is invalid, the peon task should stop immediately.
boolean disableCache = hadoopConf.getBoolean("fs.hdfs.impl.disable.cache", false);
if (!disableCache) {
FileSystem fs = FileSystem.get(hadoopConf);
fs.exists(new Path("/"));
}
}
catch (IOException ex) {
throw new ISE(ex, "Failed to access hdfs.");
}
}
@LifecycleStop
public void stop()
{
//noop
}
}

2
extensions-core/hdfs-storage/src/main/java/org/apache/druid/storage/hdfs/HdfsStorageDruidModule.java
View File

@ -117,6 +117,8 @@ public class HdfsStorageDruidModule implements DruidModule
JsonConfigProvider.bind(binder, "druid.hadoop.security.kerberos", HdfsKerberosConfig.class);
binder.bind(HdfsStorageAuthentication.class).in(ManageLifecycle.class);
LifecycleModule.register(binder, HdfsStorageAuthentication.class);
binder.bind(HdfsStorageAvailabilityChecker.class).in(ManageLifecycle.class);
LifecycleModule.register(binder, HdfsStorageAvailabilityChecker.class);
JsonConfigProvider.bind(binder, "druid.ingestion.hdfs", HdfsInputSourceConfig.class);
}