package org.glassfish.jersey.server.internal.monitoring.jmx;
import java.lang.management.ManagementFactory;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.logging.Level;
import java.util.logging.Logger;
import jakarta.ws.rs.ProcessingException;
import jakarta.inject.Inject;
import jakarta.inject.Provider;
import javax.management.JMException;
import javax.management.MBeanServer;
import javax.management.ObjectName;
import org.glassfish.jersey.server.internal.LocalizationMessages;
import org.glassfish.jersey.server.monitoring.ApplicationInfo;
import org.glassfish.jersey.server.monitoring.MonitoringStatistics;
import org.glassfish.jersey.server.monitoring.MonitoringStatisticsListener;
import org.glassfish.jersey.server.monitoring.ResourceStatistics;
import org.glassfish.jersey.server.spi.AbstractContainerLifecycleListener;
import org.glassfish.jersey.server.spi.Container;
public class MBeanExposer extends AbstractContainerLifecycleListener implements MonitoringStatisticsListener {
private static final Logger LOGGER = Logger.getLogger(MBeanExposer.class.getName());
private static final String PROPERTY_SUBTYPE_GLOBAL = "Global";
static final String PROPERTY_EXECUTION_TIMES_REQUESTS = "RequestTimes";
static final String PROPERTY_EXECUTION_TIMES_METHODS = "MethodTimes";
private volatile ExecutionStatisticsDynamicBean requestMBean;
private volatile ResponseMXBeanImpl responseMXBean;
private volatile ResourcesMBeanGroup uriStatsGroup;
private volatile ResourcesMBeanGroup resourceClassStatsGroup;
private volatile ExceptionMapperMXBeanImpl exceptionMapperMXBean;
private final AtomicBoolean destroyed = new AtomicBoolean(false);
private final Object LOCK = new Object();
private volatile String domain;
@Inject
private Provider<ApplicationInfo> applicationInfoProvider;
private Map<String, ResourceStatistics> transformToStringKeys(Map<Class<?>, ResourceStatistics> stats) {
Map<String, ResourceStatistics> newMap = new HashMap<>();
for (Map.Entry<Class<?>, ResourceStatistics> entry : stats.entrySet()) {
newMap.put(entry.getKey().getName(), entry.getValue());
}
return newMap;
}
static String convertToObjectName(String name, boolean isUri) {
if (!isUri) {
return name;
}
String str = name.replace("\\", "\\\\");
str = str.replace("?", "\\?");
str = str.replace("*", "\\*");
return "\"" + str + "\"";
}
void registerMBean(Object mbean, String namePostfix) {
final MBeanServer mBeanServer = ManagementFactory.getPlatformMBeanServer();
final String name = domain + namePostfix;
try {
synchronized (LOCK) {
if (destroyed.get()) {
return;
}
final ObjectName objectName = new ObjectName(name);
if (mBeanServer.isRegistered(objectName)) {
LOGGER.log(Level.WARNING,
LocalizationMessages.WARNING_MONITORING_MBEANS_BEAN_ALREADY_REGISTERED(objectName));
mBeanServer.unregisterMBean(objectName);
}
mBeanServer.registerMBean(mbean, objectName);
}
} catch (JMException e) {
throw new ProcessingException(LocalizationMessages.ERROR_MONITORING_MBEANS_REGISTRATION(name), e);
}
}
private void unregisterJerseyMBeans(boolean destroy) {
final MBeanServer mBeanServer = ManagementFactory.getPlatformMBeanServer();
try {
synchronized (LOCK) {
if (destroy) {
destroyed.set(true);
}
if (domain == null) {
return;
}
final Set<ObjectName> names = mBeanServer.queryNames(new ObjectName(domain + ",*"), null);
for (ObjectName name : names) {
mBeanServer.unregisterMBean(name);
}
}
} catch (Exception e) {
throw new ProcessingException(LocalizationMessages.ERROR_MONITORING_MBEANS_UNREGISTRATION_DESTROY(), e);
}
}
@Override
public void onStatistics(MonitoringStatistics statistics) {
if (domain == null) {
final String globalSubType = ",subType=" + PROPERTY_SUBTYPE_GLOBAL;
final ApplicationInfo appStats = applicationInfoProvider.get();
String appName = appStats.getResourceConfig().getApplicationName();
if (appName == null) {
appName = "App_" + Integer.toHexString(appStats.getResourceConfig().hashCode());
}
domain = "org.glassfish.jersey:type=" + appName;
unregisterJerseyMBeans(false);
uriStatsGroup = new ResourcesMBeanGroup(statistics.getUriStatistics(), true, this, ",subType=Uris");
Map<String, ResourceStatistics> newMap = transformToStringKeys(statistics.getResourceClassStatistics());
resourceClassStatsGroup = new ResourcesMBeanGroup(newMap, false, this, ",subType=Resources");
responseMXBean = new ResponseMXBeanImpl();
registerMBean(responseMXBean, globalSubType + ",global=Responses");
requestMBean = new ExecutionStatisticsDynamicBean(statistics.getRequestStatistics(),
this, globalSubType, "AllRequestTimes");
exceptionMapperMXBean = new ExceptionMapperMXBeanImpl(statistics.getExceptionMapperStatistics(), this,
globalSubType);
new ApplicationMXBeanImpl(appStats, this, globalSubType);
}
requestMBean.updateExecutionStatistics(statistics.getRequestStatistics());
uriStatsGroup.updateResourcesStatistics(statistics.getUriStatistics());
responseMXBean.updateResponseStatistics(statistics.getResponseStatistics());
exceptionMapperMXBean.updateExceptionMapperStatistics(statistics.getExceptionMapperStatistics());
this.resourceClassStatsGroup.updateResourcesStatistics(transformToStringKeys(statistics.getResourceClassStatistics()));
}
@Override
public void onShutdown(Container container) {
unregisterJerseyMBeans(true);
}
}