com.github.autoconf.helper.WebServer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of config-core Show documentation
The newest version!
package com.github.autoconf.helper;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.management.AttributeNotFoundException;
import javax.management.ObjectName;
import java.util.Collection;

import static java.lang.Integer.MAX_VALUE;
import static java.lang.Integer.parseInt;
import static java.lang.Math.max;
import static java.lang.Math.min;
import static java.lang.Thread.sleep;

/**
 * 判断提供服务的容器
 *
 * Created by lirui on 2015/04/17 上午11:08.
 */
public class WebServer {
  private static final Logger LOG = LoggerFactory.getLogger(WebServer.class);

  public static boolean isTomcat() {
    return JmxHelper.mbeanExists("Catalina:type=Server");
  }

  public static boolean isJBoss() {
    return JmxHelper.mbeanExists("jboss.system:type=Server");
  }

  public static boolean isJetty() {
    return JmxHelper.mbeanExists("org.mortbay.jetty:type=server,id=0");
  }

  public static boolean isJetty5() {
    return JmxHelper.mbeanExists("org.mortbay:jetty=default");
  }

  public static boolean isResin() {
    return JmxHelper.mbeanExists("resin:type=Resin");
  }

  public static boolean isGlassfish() {
    return JmxHelper.mbeanExists("amx:pp=,type=domain-root") || JmxHelper.mbeanExists("com.sun.appserv:j2eeType=J2EEServer,category=runtime,*");
  }

  public static Integer getTomcatHttpPort() throws Exception {
    Collection connectors = getSelectors("*:type=Connector,*");

    int lowest = MAX_VALUE;
    for (final ObjectName connector : connectors) {
      try {
        String protocol = JmxHelper.queryString(connector, "protocol");
        Integer port = JmxHelper.queryInt(connector, "port");
        LOG.info("tomcat, protocol={}, port={}", protocol, port);
        if (protocol != null && protocol.toLowerCase().contains("http")) {
          if (port == null)
            continue;
          lowest = Math.min(lowest, port);
        }
      } catch (AttributeNotFoundException e) {
        // quietly skip this connector, it's probably the wrong kind
      }
    }

    // maybe there are no HTTP connectors?
    if (lowest == MAX_VALUE) {
      for (final ObjectName connector : connectors) {
        try {
          lowest = Math.min(lowest, JmxHelper.queryInt(connector, "port"));
        } catch (AttributeNotFoundException e) {
          // quietly skip this connector, it's probably the wrong kind
        }
      }
    }

    if (lowest == MAX_VALUE) {
      return null;
    }

    return lowest;
  }

  public static Integer getResinHttpPort() throws Exception {
    Collection selectors = getSelectors("resin:type=Port,*");
    int highestHttp = -1, highestOther = -1;
    for (final ObjectName selector : selectors) {
      final Integer port = JmxHelper.queryInt(selector, "Port");
      if (port == null)
        continue;
      String protocol = JmxHelper.queryString(selector, "ProtocolName");
      String address = JmxHelper.queryString(selector, "Address");
      LOG.info("resin, protocol={}\taddress={}\tport={}", protocol, address, port);
      if (protocol == null || !"http".equalsIgnoreCase(protocol.trim())) {
        highestOther = max(highestOther, port.intValue());
      } else {
        highestHttp = max(highestHttp, port.intValue());
      }
    }

    // maybe there are no HTTP connectors?
    if (highestHttp != -1) {
      return highestHttp;
    } else if (highestOther != -1) {
      return highestOther;
    } else {
      return null;
    }
  }

  public static Integer getJettyHttpPort() throws Exception {
    Collection selectors = getSelectors("org.mortbay.jetty.nio:type=selectchannelconnector,*");
    int lowest = MAX_VALUE;
    for (final ObjectName selector : selectors) {
      lowest = Math.min(lowest, JmxHelper.queryInt(selector, "port"));
    }

    if (lowest == MAX_VALUE) {
      return null;
    }

    return lowest;
  }

  public static Integer getJetty5HttpPort() throws Exception {
    Collection selectors = getSelectors("org.mortbay:jetty=default,*");
    int lowest = MAX_VALUE;
    for (final ObjectName selector : selectors) {
      if (selector.toString().matches("org.mortbay:jetty=default,SocketListener=[0=9]"))
        lowest = Math.min(lowest, JmxHelper.queryInt(selector, "port"));
    }

    if (lowest == MAX_VALUE) {
      return null;
    }

    return lowest;
  }

  public static Integer getGlassfishHttpPort() throws Exception {
    Collection selectors = getSelectors("com.sun.appserv:type=Selector,*");
    int lowest = MAX_VALUE;
    for (final ObjectName selector : selectors) {
      final String name = selector.toString();
      if (name.contains("http")) {
        lowest = min(lowest, parseInt(name.replaceAll("[^0-9]", "")));
      }
    }

    // maybe there are no HTTP connectors?
    if (lowest == MAX_VALUE) {
      for (final ObjectName selector : selectors) {
        final String name = selector.toString();
        lowest = min(lowest, parseInt(name.replaceAll("[^0-9]", "")));
      }
    }

    if (lowest == MAX_VALUE) {
      return null;
    }
    return lowest;
  }

  public static Collection getSelectors(String name) throws Exception {
    Collection selectors;
    for (int tries = 1; tries <= 30; tries++) {
      selectors = JmxHelper.queryNames(name);
      if (selectors.isEmpty()) {
        sleep(1000L);
        LOG.info("query({}), try {} times, waiting {} seconds", name, tries, tries);
      } else {
        return selectors;
      }
    }
    throw new IllegalStateException(name + ", selector MBeans were not loaded after 30 seconds, aborting");
  }

  /**
   * 根据协议和scheme获取服务端口号
   *
   * @throws Exception 识别端口过程中可能出现的异常
   * @return 端口号
   */
  public static Integer getHttpPort() throws Exception {
    if (WebServer.isTomcat() || WebServer.isJBoss()) {
      return WebServer.getTomcatHttpPort();
    }
    if (WebServer.isResin()) {
      return WebServer.getResinHttpPort();
    }
    if (WebServer.isJetty()) {
      return WebServer.getResinHttpPort();
    }
    if (WebServer.isJetty5()) {
      return WebServer.getJetty5HttpPort();
    }
    if (WebServer.isGlassfish()) {
      return WebServer.getGlassfishHttpPort();
    }
    LOG.error("cannot detect ContainerType");
    return null;
  }
}