package org.jboss.test.clusterbench.common.jvmroute;

import java.lang.management.ManagementFactory;
import java.util.logging.Level;
import java.util.logging.Logger;

import javax.management.AttributeNotFoundException;
import javax.management.InstanceNotFoundException;
import javax.management.MBeanException;
import javax.management.MBeanServer;
import javax.management.MalformedObjectNameException;
import javax.management.ObjectName;
import javax.management.ReflectionException;

public class CommonJvmRoute {
  private static final long serialVersionUID = 3691025554701624250L;
  private static final Logger log = Logger.getLogger(CommonJvmRoute.class.getName());
  private static final String[] properties = new String[] { "jvmRoute", "jboss.mod_cluster.jvmRoute", "instance-id", "jboss.domain.web.instance-id", "jboss.jvmRoute" };
  private static final String UNKNOWN = "unknown";

  public String jvmRoute() {
    // This should work for both AS5 and AS7 with reasonable set jvmRoute property.
    String jvmRoute1 = UNKNOWN;
    String jvmRoute2 = UNKNOWN;
    String jvmRoute3 = UNKNOWN;
    String jvmRoute4 = UNKNOWN;
    for (String property : properties) {
      String value = System.getProperty(property, UNKNOWN);
      if (!value.equals(UNKNOWN) && value.length() > 0) {
        jvmRoute1 = value;
        break;
      }
    }

    // Hmm, maybe, we are deployed into Tomcat?
    if (jvmRoute1.equals(UNKNOWN)) {
      try {
        MBeanServer mbsc = ManagementFactory.getPlatformMBeanServer();
        ObjectName on = null;
        try {
          on = new ObjectName("Catalina:type=Engine");
          jvmRoute2 = (String) mbsc.getAttribute(on, "jvmRoute");
        } catch (InstanceNotFoundException ex1) {
          try {

            // Crap. It was not Catalina..., let's try jboss.web:
            on = new ObjectName("jboss.web:type=Engine");
            jvmRoute3 = (String) mbsc.getAttribute(on, "jvmRoute");

            // Well, it looks like AS7 with no jvmRoute set, let's try to retrieve instance-id...
          } catch (InstanceNotFoundException ex2) {
            try {
              on = new ObjectName("jboss.as:subsystem=web");
              jvmRoute4 = (String) mbsc.getAttribute(on, "instance-id");
            } catch (InstanceNotFoundException ex3) {
              jvmRoute4 = UNKNOWN;
              log.log(Level.WARNING, "We have failed to determine the jvmRoute.");
            } catch (NullPointerException e) {
              jvmRoute4 = UNKNOWN;
              log.log(Level.WARNING, "We have failed to determine the jvmRoute.");
            }
          }
        }

      } catch (AttributeNotFoundException e) {
        log.log(Level.SEVERE, ":-(");
        e.printStackTrace();
      } catch (MalformedObjectNameException e) {
        log.log(Level.SEVERE, ":-(");
        e.printStackTrace();
      } catch (MBeanException e) {
        log.log(Level.SEVERE, ":-(");
        e.printStackTrace();
      } catch (ReflectionException e) {
        log.log(Level.SEVERE, ":-(");
        e.printStackTrace();
      } catch (NullPointerException e) {
        log.log(Level.SEVERE, ":-(");
        e.printStackTrace();
      }
    }
    String jvmRoute = UNKNOWN;
    if (jvmRoute4 != null && !jvmRoute1.equals(UNKNOWN)) {
      jvmRoute = jvmRoute1;
    } else if (jvmRoute4 != null && !jvmRoute2.equals(UNKNOWN)) {
      jvmRoute = jvmRoute2;
    } else if (jvmRoute4 != null && !jvmRoute3.equals(UNKNOWN)) {
      jvmRoute = jvmRoute3;
    } else if (jvmRoute4 != null && !jvmRoute4.equals(UNKNOWN)) {
      jvmRoute = jvmRoute4;
    } else {
      log.log(Level.WARNING, "We have failed to determine the jvmRoute.");
    }
    return jvmRoute;
  }
}

@@ -3208,33 +3208,33 @@
     if (jvmRoute
-4
+1
  != null && !jvm
@@ -3302,33 +3302,33 @@
 lse if (jvmRoute
-4
+2
  != null && !jvm
@@ -3396,33 +3396,33 @@
 lse if (jvmRoute
-4
+3
  != null && !jvm

Fix for NullPointerException