[nifi] branch support/nifi-1.x updated: NIFI-12239 Add OS and Java details at startup in bootstrap log

[exceptionfactory]

This is an automated email from the ASF dual-hosted git repository.

exceptionfactory pushed a commit to branch support/nifi-1.x
in repository https://gitbox.apache.org/repos/asf/nifi.git


The following commit(s) were added to refs/heads/support/nifi-1.x by this push:
     new 17738df3b0 NIFI-12239 Add OS and Java details at startup in bootstrap 
log
17738df3b0 is described below

commit 17738df3b06c4fb7933dfc59281857016fbccfe6
Author: Pierre Villard <pierre.villard...@gmail.com>
AuthorDate: Tue Oct 17 10:40:44 2023 -0500

    NIFI-12239 Add OS and Java details at startup in bootstrap log
    
    This closes #7891
    
    Signed-off-by: David Handermann <exceptionfact...@apache.org>
    
    (cherry picked from commit e4e2336ebe022ee8f42cd29b52d6e210c9ceb24c)
---
 .../java/org/apache/nifi/bootstrap/RunNiFi.java    | 47 ++++++++++++++++++++++
 1 file changed, 47 insertions(+)

diff --git 
a/nifi-bootstrap/src/main/java/org/apache/nifi/bootstrap/RunNiFi.java 
b/nifi-bootstrap/src/main/java/org/apache/nifi/bootstrap/RunNiFi.java
index d2b23b388e..586b6fc53a 100644
--- a/nifi-bootstrap/src/main/java/org/apache/nifi/bootstrap/RunNiFi.java
+++ b/nifi-bootstrap/src/main/java/org/apache/nifi/bootstrap/RunNiFi.java
@@ -40,6 +40,7 @@ import java.io.InputStream;
 import java.io.InputStreamReader;
 import java.io.OutputStream;
 import java.io.Reader;
+import java.lang.management.ManagementFactory;
 import java.lang.reflect.Method;
 import java.net.InetAddress;
 import java.net.InetSocketAddress;
@@ -59,6 +60,7 @@ import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.HashMap;
 import java.util.HashSet;
+import java.util.LinkedHashMap;
 import java.util.List;
 import java.util.Map;
 import java.util.Map.Entry;
@@ -74,6 +76,9 @@ import java.util.concurrent.locks.Condition;
 import java.util.concurrent.locks.Lock;
 import java.util.concurrent.locks.ReentrantLock;
 
+import javax.management.MBeanServer;
+import javax.management.ObjectName;
+
 /**
  * <p>
  * The class which bootstraps Apache NiFi. This class looks for the
@@ -1172,6 +1177,9 @@ public class RunNiFi {
 
         nifiPropsFilename = nifiPropsFilename.trim();
 
+        String maximumHeapSize = null;
+        String minimumHeapSize = null;
+
         final List<String> javaAdditionalArgs = new ArrayList<>();
         for (final Map.Entry<String, String> entry : props.entrySet()) {
             final String key = entry.getKey();
@@ -1179,6 +1187,12 @@ public class RunNiFi {
 
             if (key.startsWith("java.arg")) {
                 javaAdditionalArgs.add(value);
+                if (value.startsWith("-Xms")) {
+                    minimumHeapSize = StringUtils.substringAfter(value, 
"-Xms");
+                }
+                if (value.startsWith("-Xmx")) {
+                    maximumHeapSize = StringUtils.substringAfter(value, 
"-Xmx");
+                }
             }
         }
 
@@ -1225,6 +1239,7 @@ public class RunNiFi {
         if (RuntimeVersionProvider.isMajorVersionDeprecated(javaMajorVersion)) 
{
             deprecationLogger.warn("Support for Java {} is deprecated. Java {} 
is the minimum recommended version", javaMajorVersion, 
RuntimeVersionProvider.getMinimumMajorVersion());
         }
+        defaultLogger.info(getPlatformDetails(minimumHeapSize, 
maximumHeapSize));
 
         final StringBuilder classPathBuilder = new StringBuilder();
         for (int i = 0; i < cpFiles.size(); i++) {
@@ -1277,6 +1292,7 @@ public class RunNiFi {
         cmd.add("-classpath");
         cmd.add(classPath);
         cmd.addAll(javaAdditionalArgs);
+
         cmd.add("-Dnifi.properties.file.path=" + nifiPropsFilename);
         cmd.add("-Dnifi.bootstrap.listen.port=" + listenPort);
         cmd.add("-Dapp=NiFi");
@@ -1434,6 +1450,37 @@ public class RunNiFi {
         }
     }
 
+    private String getPlatformDetails(final String minimumHeapSize, final 
String maximumHeapSize) {
+        final Map<String, String> details = new LinkedHashMap<String, 
String>(6);
+
+        details.put("javaVersion", System.getProperty("java.version"));
+        details.put("availableProcessors", 
Integer.toString(Runtime.getRuntime().availableProcessors()));
+
+        try {
+            final ObjectName osObjectName = 
ManagementFactory.getOperatingSystemMXBean().getObjectName();
+            final MBeanServer mBeanServer = 
ManagementFactory.getPlatformMBeanServer();
+            final Object maxOpenFileCount = 
mBeanServer.getAttribute(osObjectName, "MaxFileDescriptorCount");
+            if (maxOpenFileCount!= null) {
+                details.put("maxOpenFileDescriptors", 
String.valueOf(maxOpenFileCount));
+            }
+            final Object totalPhysicalMemory = 
mBeanServer.getAttribute(osObjectName, "TotalPhysicalMemorySize");
+            if (totalPhysicalMemory != null) {
+                details.put("totalPhysicalMemoryMB", String.valueOf(((Long) 
totalPhysicalMemory) / (1024*1024)));
+            }
+        } catch (final Throwable t) {
+            // Ignore. This will throw either ClassNotFound or 
NoClassDefFoundError if unavailable in this JVM.
+        }
+
+        if (minimumHeapSize != null) {
+            details.put("minimumHeapSize", minimumHeapSize);
+        }
+        if (maximumHeapSize != null) {
+            details.put("maximumHeapSize", maximumHeapSize);
+        }
+
+        return details.toString();
+    }
+
     private void writeSensitiveKeyFile(Map<String, String> props, Path 
sensitiveKeyFile) throws IOException {
         BufferedWriter sensitiveKeyWriter = 
Files.newBufferedWriter(sensitiveKeyFile, StandardCharsets.UTF_8);
         sensitiveKeyWriter.write(props.get(NIFI_BOOTSTRAP_SENSITIVE_KEY));