From 3e71418bf8ef4ebcb3a628e7c7f8b0dff626acc2 Mon Sep 17 00:00:00 2001
From: Obique PSWG <obique.pswg@gmail.com>
Date: Thu, 27 Aug 2015 17:03:49 -0500
Subject: [PATCH] Added in CPU and Memory details to online graph, including
 for overall system

---
 res/webserver/index.html                      |  2 +-
 .../admin/OnlineInterfaceService.java         |  4 +-
 src/services/admin/WebserverData.java         | 51 +++++++++++++--
 src/services/admin/WebserverHandler.java      | 64 ++++++++++++++-----
 4 files changed, 96 insertions(+), 25 deletions(-)

diff --git a/res/webserver/index.html b/res/webserver/index.html
index fafa95f75..f5ba90e5b 100644
--- a/res/webserver/index.html
+++ b/res/webserver/index.html
@@ -66,7 +66,7 @@
 		<div class="main_container">
 			<div width="100%">
 				<div class="memory_container">
-					Memory Usage:<br />
+					Resource Usage:<br />
 					<image id="memory_usage" src="memory_usage.png" width="100%"/>
 				</div>
 				<iframe id="server_info" class="server_info" src="server_info.html" frameBorder="0" seamless="yes">
diff --git a/src/services/admin/OnlineInterfaceService.java b/src/services/admin/OnlineInterfaceService.java
index 29eab83b5..5a94b1cca 100644
--- a/src/services/admin/OnlineInterfaceService.java
+++ b/src/services/admin/OnlineInterfaceService.java
@@ -131,9 +131,7 @@ public class OnlineInterfaceService extends Service implements HttpServerCallbac
 	}
 	
 	private void collectData() {
-		Runtime r = Runtime.getRuntime();
-		double memUsage = 1 - ((double) r.totalMemory() / r.maxMemory());
-		data.addMemoryUsageData(memUsage);
+		data.updateResourceUsage();
 	}
 	
 	private InetAddress getBindAddr(Config c, String firstTry, String secondTry) {
diff --git a/src/services/admin/WebserverData.java b/src/services/admin/WebserverData.java
index 727c9dac3..77473e1b2 100644
--- a/src/services/admin/WebserverData.java
+++ b/src/services/admin/WebserverData.java
@@ -5,6 +5,8 @@ import java.io.File;
 import java.io.FileInputStream;
 import java.io.IOException;
 import java.io.InputStreamReader;
+import java.lang.management.ManagementFactory;
+import java.lang.management.OperatingSystemMXBean;
 import java.nio.charset.Charset;
 import java.util.Arrays;
 import java.util.Collections;
@@ -18,17 +20,27 @@ class WebserverData {
 	private static final Charset ASCII = Charset.forName("ASCII");
 	
 	private final double [] memoryUsage;
+	private final double [] cpuUsage;
+	private final double [] systemMemoryUsage;
+	private final double [] systemCpuUsage;
 	private final Set<Player> onlinePlayers;
 	
 	public WebserverData() {
 		memoryUsage = new double[60 * 5];
+		cpuUsage = new double[memoryUsage.length];
+		systemMemoryUsage = new double[memoryUsage.length];
+		systemCpuUsage = new double[memoryUsage.length];
 		onlinePlayers = new HashSet<>();
 	}
 	
-	public void addMemoryUsageData(double percent) {
-		for (int i = 1; i < memoryUsage.length; i++)
-			memoryUsage[i-1] = memoryUsage[i];
-		memoryUsage[memoryUsage.length-1] = percent;
+	@SuppressWarnings("restriction")
+	public void updateResourceUsage() {
+		OperatingSystemMXBean os = ManagementFactory.getOperatingSystemMXBean();
+		if (os instanceof com.sun.management.OperatingSystemMXBean) {
+			com.sun.management.OperatingSystemMXBean castedOS = (com.sun.management.OperatingSystemMXBean) os;
+			updateMemory(castedOS);
+			updateCPU(castedOS);
+		}
 	}
 	
 	public void addOnlinePlayer(Player player) {
@@ -43,6 +55,18 @@ class WebserverData {
 		return Arrays.copyOf(memoryUsage, memoryUsage.length);
 	}
 	
+	public double [] getCpuUsage() {
+		return Arrays.copyOf(cpuUsage, cpuUsage.length);
+	}
+
+	public double [] getSystemMemoryUsage() {
+		return Arrays.copyOf(systemMemoryUsage, systemMemoryUsage.length);
+	}
+	
+	public double [] getSystemCpuUsage() {
+		return Arrays.copyOf(systemCpuUsage, systemCpuUsage.length);
+	}
+	
 	public Set<Player> getOnlinePlayers() {
 		return Collections.unmodifiableSet(onlinePlayers);
 	}
@@ -58,4 +82,23 @@ class WebserverData {
 		}
 	}
 	
+	@SuppressWarnings("restriction")
+	private void updateMemory(com.sun.management.OperatingSystemMXBean os) {
+		Runtime r = Runtime.getRuntime();
+		appendData(memoryUsage, (double) r.totalMemory() / os.getTotalPhysicalMemorySize());
+		appendData(systemMemoryUsage, 1 - ((double) os.getFreePhysicalMemorySize() / os.getTotalPhysicalMemorySize()));
+	}
+	
+	@SuppressWarnings("restriction")
+	private void updateCPU(com.sun.management.OperatingSystemMXBean os) {
+		appendData(cpuUsage, os.getProcessCpuLoad());
+		appendData(systemCpuUsage, os.getSystemCpuLoad());
+	}
+	
+	private void appendData(double [] data, double value) {
+		for (int i = 1; i < data.length; i++)
+			data[i-1] = data[i];
+		data[data.length-1] = value;
+	}
+	
 }
diff --git a/src/services/admin/WebserverHandler.java b/src/services/admin/WebserverHandler.java
index dbc4c295b..96ba02355 100644
--- a/src/services/admin/WebserverHandler.java
+++ b/src/services/admin/WebserverHandler.java
@@ -2,6 +2,7 @@ package services.admin;
 
 import java.awt.Color;
 import java.awt.Font;
+import java.awt.FontMetrics;
 import java.awt.Graphics2D;
 import java.awt.RenderingHints;
 import java.awt.image.BufferedImage;
@@ -13,7 +14,9 @@ import java.io.IOException;
 import java.io.InputStream;
 import java.io.InputStreamReader;
 import java.nio.charset.Charset;
+import java.util.HashMap;
 import java.util.Locale;
+import java.util.Map;
 import java.util.Set;
 import java.util.regex.Matcher;
 import java.util.regex.Pattern;
@@ -60,39 +63,66 @@ class WebserverHandler {
 		final int graphHeight = 300;
 		BufferedImage image = new BufferedImage(900, graphHeight + 25, BufferedImage.TYPE_3BYTE_BGR);
 		Graphics2D g = image.createGraphics();
-		int prevX = -1;
-		int prevY = -1;
 		Font font = g.getFont();
-		g.setFont(font.deriveFont(Math.min(image.getHeight()/10.0f, 25.0f)));
-		RenderingHints rh = new RenderingHints(RenderingHints.KEY_TEXT_ANTIALIASING, RenderingHints.VALUE_TEXT_ANTIALIAS_ON);
-		g.setRenderingHints(rh);
+		g.setFont(font.deriveFont(Font.TRUETYPE_FONT, Math.min(image.getHeight()/10.0f, 25.0f)));
+		Map <RenderingHints.Key, Object> hints = new HashMap<>();
+		hints.put(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
+		hints.put(RenderingHints.KEY_TEXT_ANTIALIASING, RenderingHints.VALUE_TEXT_ANTIALIAS_ON);
+		g.setRenderingHints(new RenderingHints(hints));
 		g.setColor(Color.WHITE);
-		g.drawRect(0, 0, image.getWidth()-1, graphHeight-1);
-		g.drawLine(0, image.getHeight()-1, image.getWidth(), image.getHeight()-1);
+		renderGraph(g, image.getWidth(), image.getHeight(), graphHeight);
+		FontMetrics fm = g.getFontMetrics();
+		String inUse = getMemoryInUse();
+		int width = fm.stringWidth(inUse);
+		int x = 5;
+		x += drawStr(g, image.getHeight(), x, "Proc:");
+		g.drawString(inUse, image.getWidth()-width - 5, image.getHeight() - 5);
+		x += drawDataSet(g, Color.RED, " M:", image.getWidth(), image.getHeight(), graphHeight, x, memoryUsage);
+		x += drawDataSet(g, Color.CYAN, " C:", image.getWidth(), image.getHeight(), graphHeight, x, data.getCpuUsage());
+		g.setColor(Color.WHITE);
+		x += drawStr(g, image.getHeight(), x, "  Sys:");
+		x += drawDataSet(g, Color.YELLOW, " M:", image.getWidth(), image.getHeight(), graphHeight, x, data.getSystemMemoryUsage());
+		x += drawDataSet(g, Color.GREEN, " C:", image.getWidth(), image.getHeight(), graphHeight, x, data.getSystemCpuUsage());
+		return image;
+	}
+	
+	private void renderGraph(Graphics2D g, int width, int height, int graphHeight) {
+		g.drawRect(0, 0, width-1, graphHeight-1);
+		g.drawLine(0, height-1, width, height-1);
 		for (int i = 1; i < 10; i++) {
 			g.drawLine(0, (int) ((i/10.0)*graphHeight), 10, (int) ((i/10.0)*graphHeight));
 		}
-		String inUse = getMemoryInUse();
-		int width = g.getFontMetrics().stringWidth(inUse);
-		g.drawString("Current: " + (int) (memoryUsage[memoryUsage.length-1] * 100) + "%", 5, image.getHeight() - 5);
-		g.drawString(inUse, image.getWidth()-width - 5, image.getHeight() - 5);
-		g.setColor(Color.RED);
+	}
+	
+	private int drawDataSet(Graphics2D g, Color c, String pre, int width, int height, int graphHeight, int x, double [] data) {
+		g.setColor(c);
+		drawLines(g, data, width, graphHeight-1);
+		return drawStr(g, height, x, String.format("%s%.2f%%", pre, (int) (data[data.length-1] * 10000) / 100.0));
+	}
+	
+	private int drawStr(Graphics2D g, int height, int x, String str) {
+		g.drawString(str, x, height - 5);
+		return g.getFontMetrics().stringWidth(str);
+	}
+	
+	private void drawLines(Graphics2D g, double [] data, int width, int height) {
 		int start = 0;
-		for (double d : memoryUsage) {
+		for (double d : data) {
 			if (d == 0)
 				start++;
 			else
 				break;
 		}
-		for (int i = start; i < memoryUsage.length; i++) {
-			int x = (int) ((double) i / memoryUsage.length * image.getWidth());
-			int y = (int) ((1-memoryUsage[i]) * graphHeight);
+		int prevX = -1;
+		int prevY = -1;
+		for (int i = start; i < data.length; i++) {
+			int x = (int) ((double) i / data.length * width);
+			int y = (int) ((1-data[i]) * height);
 			if (prevX != -1 && prevY != -1)
 				g.drawLine(prevX, prevY, x, y);
 			prevX = x;
 			prevY = y;
 		}
-		return image;
 	}
 	
 	private String getMemoryInUse() {