Java实现监控tomcat线程池

import jodd.util.StringUtil;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Service;

import javax.management.MBeanServer;
import javax.management.MBeanServerFactory;
import javax.management.MalformedObjectNameException;
import javax.management.ObjectName;
import java.lang.management.ManagementFactory;
import java.net.InetAddress;
import java.util.HashSet;
import java.util.Set;

/**
 * tomcat线程池监控
 *
 * @author huchenfei
 * @version 1.0
 * @class TomcatThreadPoolMonitorJob
 * @date 2021/1/2 20:15
 * @description
 */
@EnableScheduling
@Service
public class TomcatThreadPoolMonitorJob {

    private final static Logger LOCAL_LOGGER = LoggerFactory.getLogger(TomcatThreadPoolMonitorJob.class);

    private static final String MONITOR_PREFIX = "tomcat_threadPool.";

    private static MBeanServer mbeanServer;

    private static ObjectName objectName = null;

    static {
        try {
            objectName = new ObjectName("Catalina:type=ThreadPool,*");
            mbeanServer = getMBeanServer();
        } catch (MalformedObjectNameException e) {
            // ii
        }
    }

    @Scheduled(cron = "0 0/1 * * * ?")
    public void execute() {
        try {

            // collect thread metrics
            Set<ObjectName> connectorNames = getObjectNames(objectName);
            InetAddress localHost = InetAddress.getLocalHost();
            String hostName = localHost.getHostName().replace('.', '_');
            String host = hostName + ":";
            for (ObjectName connectorName : connectorNames) {
                // only check one
                String subType = connectorName.getKeyProperty("subType");
                if (StringUtil.isNotEmpty(subType) || !isTomcatServer(connectorName.getDomain())) {
                    continue;
                }

                String name = ObjectName.unquote(connectorName.getKeyProperty("name"));
                // 基本不使用的协议
                if (name.startsWith("ajp")) {
                    continue;
                }

                String monitorPort = String.valueOf(mbeanServer.getAttribute(connectorName, "port"));
                String prefix = MONITOR_PREFIX + host + monitorPort + ".";

                int currentThreadsBusy = (Integer) mbeanServer.getAttribute(connectorName, "currentThreadsBusy");
                LOCAL_LOGGER.info("tomcat.thread.pool.busyCount {}", currentThreadsBusy);

                int currentThreadCount = (Integer) mbeanServer.getAttribute(connectorName, "currentThreadCount");
                LOCAL_LOGGER.info("tomcat.thread.pool.currentCount {}", currentThreadCount);

                int minSpareThreads = (Integer) mbeanServer.getAttribute(connectorName, "minSpareThreads");
                LOCAL_LOGGER.info("tomcat.thread.pool.spareCount {}", minSpareThreads);

                int maxThreads = (Integer) mbeanServer.getAttribute(connectorName, "maxThreads");
                LOCAL_LOGGER.info("tomcat.thread.pool.maxCount {}", maxThreads);
            }
        } catch (Exception e) {
            LOCAL_LOGGER.error("Exception occur when getting connector global stats: ", e);
        }
    }

    public Set<ObjectName> getObjectNames(ObjectName objectName) {
        if (objectName == null) {
            return new HashSet<>();
        }
        Set<ObjectName> objectNames = mbeanServer.queryNames(objectName, null);
        if (objectNames != null && !objectNames.isEmpty()) {
            return objectNames;
        }
        return new HashSet<>();
    }

    public static MBeanServer getMBeanServer() {
        MBeanServer mBeanServer = ManagementFactory.getPlatformMBeanServer();
        if (MBeanServerFactory.findMBeanServer(null).size() > 0) {
            mBeanServer = MBeanServerFactory.findMBeanServer(null).get(0);
        }
        return mBeanServer;
    }

    private boolean isTomcatServer(String domain) {
        if (StringUtil.isEmpty(domain)) {
            return false;
        }
        return StringUtils.equalsIgnoreCase("tomcat", domain) ||
                StringUtils.equalsIgnoreCase("catalina", domain);
    }
}