HTTP Pool data added to JMX management

java/src/main/java/com/genexus/internet/HttpClientJavaLib.java

@@ -11,14 +11,17 @@
 import java.security.UnrecoverableKeyException;
 import java.security.cert.CertificateException;
 import java.util.*;
-import com.genexus.ModelContext;
-import com.genexus.util.IniFile;
+import java.net.URI;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Executors;
+import javax.net.ssl.SSLContext;
+
 import org.apache.http.*;
-import com.genexus.CommonUtil;
-import com.genexus.specific.java.*;
 import org.apache.http.HttpResponse;
 import org.apache.http.client.config.CookieSpecs;
+import org.apache.http.conn.routing.HttpRoute;
 import org.apache.http.conn.ssl.NoopHostnameVerifier;
+import org.apache.http.impl.conn.PoolingHttpClientConnectionManager;
 import org.apache.http.protocol.HttpContext;
 import org.apache.http.auth.AuthSchemeProvider;
 import org.apache.http.auth.AuthScope;
@@ -43,17 +46,21 @@
 import org.apache.http.impl.auth.SPNegoSchemeFactory;
 import org.apache.http.impl.client.*;
 import org.apache.http.impl.conn.DefaultProxyRoutePlanner;
-import org.apache.http.impl.conn.PoolingHttpClientConnectionManager;
 import org.apache.http.message.BasicHeaderElementIterator;
 import org.apache.http.protocol.HTTP;
 import org.apache.http.ssl.SSLContextBuilder;
 import org.apache.http.ssl.SSLContexts;
 import org.apache.http.util.EntityUtils;
 import org.apache.logging.log4j.Logger;
-import com.genexus.webpanels.HttpContextWeb;
-import java.net.URI;
 
-import javax.net.ssl.SSLContext;
+import com.genexus.webpanels.HttpContextWeb;
+import com.genexus.ModelContext;
+import com.genexus.management.HTTPConnectionJMX;
+import com.genexus.management.HTTPPoolJMX;
+import com.genexus.util.IniFile;
+import com.genexus.Application;
+import com.genexus.CommonUtil;
+import com.genexus.specific.java.*;
 
 public class HttpClientJavaLib extends GXHttpClient {
 
@@ -75,12 +82,21 @@ private static void getPoolInstance() {
 			connManager = new PoolingHttpClientConnectionManager(socketFactoryRegistry);
 			connManager.setMaxTotal((int) CommonUtil.val(clientCfg.getProperty("Client", "HTTPCLIENT_MAX_SIZE", "1000")));
 			connManager.setDefaultMaxPerRoute((int) CommonUtil.val(clientCfg.getProperty("Client", "HTTPCLIENT_MAX_PER_ROUTE", "1000")));
+
+			if (Application.isJMXEnabled())
+				HTTPPoolJMX.CreateHTTPPoolJMX(connManager);
 		}
 		else {
 			connManager.closeExpiredConnections();
 		}
 	}
 
+	@Override
+	protected void finalize() {
+		this.closeOpenedStreams();
+		executor.shutdown();
+	}
+
 	private ConnectionKeepAliveStrategy generateKeepAliveStrategy() {
 		return new ConnectionKeepAliveStrategy() {
 			@Override
@@ -110,7 +126,6 @@ public void setTimeout(int timeout)
 	}
 
 	private static Logger logger = org.apache.logging.log4j.LogManager.getLogger(HttpClientJavaLib.class);
-
 	private static PoolingHttpClientConnectionManager connManager = null;
 	private Integer statusCode = 0;
 	private String reasonLine = "";
@@ -127,8 +142,9 @@ public void setTimeout(int timeout)
 	private static IniFile clientCfg = new ModelContext(ModelContext.getModelContextPackageClass()).getPreferences().getIniFile();
 	private static final String SET_COOKIE = "Set-Cookie";
 	private static final String COOKIE = "Cookie";
-
 	private java.util.Vector<InputStream> streamsToClose;
+	private static HashSet<HttpRoute> storedRoutes = new HashSet<>();
+
 
 	private void closeOpenedStreams()
 	{
@@ -599,8 +615,12 @@ public void execute(String method, String url) {
 			this.reasonLine = "";
 		}
 		finally {
-			if (getIsURL())
-			{
+			if (Application.isJMXEnabled()){
+				if (executor.isShutdown())
+					executor = Executors.newSingleThreadExecutor();
+				executor.submit(this::displayHTTPConnections);
+			}
+			if (getIsURL()) {
 				this.setHost(getPrevURLhost());
 				this.setBaseURL(getPrevURLbaseURL());
 				this.setPort(getPrevURLport());
@@ -611,6 +631,21 @@ public void execute(String method, String url) {
 		}
 	}
 
+	private static ExecutorService executor = Executors.newSingleThreadExecutor();
+	private synchronized void displayHTTPConnections(){
+		Iterator<HttpRoute> iterator = storedRoutes.iterator();
+		while (iterator.hasNext()) {
+			HttpRoute route = iterator.next();
+			HTTPConnectionJMX.DestroyHTTPConnectionJMX(route);
+			iterator.remove();
+		}
+
+		for (HttpRoute route : connManager.getRoutes()){
+			HTTPConnectionJMX.CreateHTTPConnectionJMX(route);
+			storedRoutes.add(route);
+		}
+	}
+
 	public int getStatusCode() {
 		return statusCode;
 	}
@@ -738,10 +773,4 @@ public void cleanup() {
 		resetErrorsAndConnParams();
 	}
 
-	@Override
-	protected void finalize()
-	{
-		this.closeOpenedStreams();
-	}
-
 }

java/src/main/java/com/genexus/management/HTTPConnectionJMX.java

@@ -0,0 +1,41 @@
+package com.genexus.management;
+
+import org.apache.http.conn.routing.HttpRoute;
+import org.apache.logging.log4j.Logger;
+
+public class HTTPConnectionJMX implements HTTPConnectionJMXBean{
+
+	private static Logger log = org.apache.logging.log4j.LogManager.getLogger(HTTPConnectionJMX.class);
+
+	HttpRoute httpRoute;
+
+	public HTTPConnectionJMX(HttpRoute httpRoute) {
+		this.httpRoute = httpRoute;
+	}
+
+	static public void CreateHTTPConnectionJMX(HttpRoute connection) {
+		try {
+			MBeanUtils.createMBean(connection);
+		}
+		catch (Exception e) {
+			log.error("Failed to register HTTP connection MBean.", e);
+		}
+	}
+
+	static public void DestroyHTTPConnectionJMX(HttpRoute connection) {
+		try {
+			MBeanUtils.destroyMBean(connection);
+		}
+		catch (Exception e) {
+			log.error("Failed to destroy HTTP connection MBean.", e);
+		}
+	}
+
+	public int getPort() {
+		return httpRoute.getTargetHost().getPort();
+	}
+
+	public String getHost() {
+		return httpRoute.getTargetHost().getHostName();
+	}
+}

java/src/main/java/com/genexus/management/HTTPConnectionJMXBean.java

@@ -0,0 +1,6 @@
+package com.genexus.management;
+
+public interface HTTPConnectionJMXBean {
+	int getPort();
+	String getHost();
+}

java/src/main/java/com/genexus/management/HTTPPoolJMX.java

@@ -0,0 +1,58 @@
+package com.genexus.management;
+
+import javax.management.MBeanNotificationInfo;
+import javax.management.Notification;
+import javax.management.NotificationBroadcasterSupport;
+
+import org.apache.http.impl.conn.PoolingHttpClientConnectionManager;
+import org.apache.logging.log4j.Logger;
+public class HTTPPoolJMX extends NotificationBroadcasterSupport implements HTTPPoolJMXMBean{
+
+	private long sequenceNumber=0;
+	PoolingHttpClientConnectionManager connectionPool;
+	private long lastUserWaitingForLongTimeNotif = 0L;
+	private long lastPoollsFullNotif = 0L;
+	private static Logger log = org.apache.logging.log4j.LogManager.getLogger(HTTPPoolJMX.class);
+
+	public HTTPPoolJMX(PoolingHttpClientConnectionManager connectionPool) {
+		this.connectionPool = connectionPool;
+	}
+
+	static public void CreateHTTPPoolJMX(PoolingHttpClientConnectionManager httpConnectionPool) {
+		try {
+			MBeanUtils.createMBean(httpConnectionPool);
+		}
+		catch(Exception e) {
+			log.error("Failed to register HTTP connection pool MBean.", e);
+		}
+	}
+
+	public int getNumberOfConnectionsInUse(){
+		return connectionPool.getTotalStats().getLeased();
+	}
+
+	public int getNumberOfRequestsWaiting(){
+		return connectionPool.getTotalStats().getPending();
+	}
+
+	public int getNumberOfAvailableConnections(){
+		return connectionPool.getTotalStats().getAvailable();
+	}
+
+	public int getMaxNumberOfConnections(){
+		return connectionPool.getTotalStats().getMax();
+	}
+
+	public MBeanNotificationInfo[] getNotificationInfo() {
+		String[] types = new String[] {"com.genexus.managment.fullpool"};
+		String name = Notification.class.getName();
+		String description = "The Connection Pool does not have available connections ";
+		MBeanNotificationInfo info = new MBeanNotificationInfo(types, name, description);
+
+		types = new String[] {"com.genexus.managment.longtimeuserwaiting"};
+		description = "User waiting a connection for a long time";
+		MBeanNotificationInfo info1 = new MBeanNotificationInfo(types, name, description);
+
+		return new MBeanNotificationInfo[] {info, info1};
+	}
+}

java/src/main/java/com/genexus/management/HTTPPoolJMXMBean.java

@@ -0,0 +1,8 @@
+package com.genexus.management;
+
+public interface HTTPPoolJMXMBean {
+	int getNumberOfConnectionsInUse();
+	int getNumberOfRequestsWaiting();
+	int getNumberOfAvailableConnections();
+	int getMaxNumberOfConnections();
+}