[jvm-packages] handle null to avoid potential NullPointerException

dmlc · Jun 28, 2024 · a9cb728 · a9cb728
1 parent 824fba7
commit a9cb728
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/jvm-packages/xgboost4j/src/main/java/ml/dmlc/xgboost4j/java/TrackerProperties.java b/jvm-packages/xgboost4j/src/main/java/ml/dmlc/xgboost4j/java/TrackerProperties.java
@@ -22,10 +22,12 @@ private TrackerProperties() {
     InputStream inputStream = null;
 
     try {
-      URL propertiesFileURL =
-          Thread.currentThread().getContextClassLoader().getResource(PROPERTIES_FILENAME);
-      if (propertiesFileURL != null){
-        inputStream = propertiesFileURL.openStream();
+      ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
+      if (classLoader != null) {
+        URL propertiesFileURL = classLoader.getResource(PROPERTIES_FILENAME);
+        if (propertiesFileURL != null) {
+          inputStream = propertiesFileURL.openStream();
+        }
       }
     } catch (IOException e) {
       logger.warn("Could not load " + PROPERTIES_FILENAME + " file. ", e);