From a9cb7283f367e4a568afe553ec10480ab0513163 Mon Sep 17 00:00:00 2001 From: Shangjin Tang Date: Fri, 28 Jun 2024 10:59:44 +0800 Subject: [PATCH] [jvm-packages] handle null to avoid potential NullPointerException --- .../java/ml/dmlc/xgboost4j/java/TrackerProperties.java | 10 ++++++---- 1 file changed, 6 insertions(+), 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 index 45a6b1e062a7..4bbab9b12300 100644 --- 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);