handle argument

Author: bill23-kim

pulse.sample.webtime/samples.txt

@@ -0,0 +1,7 @@
+http://www.naver.com
+http://www.daum.net
+http://www.facebook.com
+http://www.nate.com
+http://www.yahoo.com
+http://www.youtube.com
+http://www.google.com

pulse.sample.webtime/src/main/java/scouter/pulse/webtime/App.java

@@ -1,7 +1,12 @@
 package scouter.pulse.webtime;
 
+import java.io.BufferedReader;
+import java.io.File;
+import java.io.FileReader;
 import java.net.InetAddress;
+import java.util.ArrayList;
 import java.util.HashMap;
+import java.util.List;
 import java.util.concurrent.Executors;
 import java.util.concurrent.ScheduledExecutorService;
 import java.util.concurrent.TimeUnit;
@@ -23,37 +28,68 @@
 public class App 
 {
 
-	public final static String[] TARGET_SITES = {
+	public final static String[] SAMPLE_SITES = {
 			"http://www.google.co.kr",
 			"http://www.google.com",
 			"http://www.google.co.uk",
 			"http://www.google.co.jp",
-			"http://www.google.co.be", // Belgium
-			"http://www.google.com.br", // Brazil
+			"http://www.google.co.be", // Belgium Google
+			"http://www.google.com.br", // Brazil Google
 			"http://www.google.de",
-			"http://www.google.co.za", // South Africa
+			"http://www.google.co.za", // SouthAfrica Google
 	};
 
 	static ScheduledExecutorService executor = Executors.newSingleThreadScheduledExecutor();
-
+	
+	static String ip = "127.0.0.1";
+	static String port = "6180";
+	
+	static List<String> siteList = null; 
 
     public static void main( String[] args ) throws Exception
     {
 
+    	if (args != null && args.length > 0) {
+    		ip = args[0];
+    		if (args.length > 1) {
+    			port = args[1];
+    			if (args.length > 2) {
+    				String filename = args[2];
+    				File f = new File(filename);
+    				if (f.canRead()) {
+    					BufferedReader br = new BufferedReader(new FileReader(f));
+    					String line = null;
+    					siteList = new ArrayList<String>();
+    					while ((line = br.readLine()) != null) {
+    						siteList.add(line);
+    					}
+    					br.close();
+    				}
+    			}
+    		}
+    	}
+    	
+    	if (siteList == null) {
+    		siteList = new ArrayList<String>();
+    		for (String site : SAMPLE_SITES) {
+    			siteList.add(site);
+    		}
+    	}
+    	
     	int status = registerMetadata();
 
     	if (status == HttpStatus.SC_CREATED) {
 
     		Runnable r = new Runnable() {
-    			HashMap<String, InternetObject> statusMap = new HashMap<String, InternetObject>(); 
+    			HashMap<String, WebObject> statusMap = new HashMap<String, WebObject>(); 
 
 				public void run() {
 					sendCounterData();
-					for (int i = 0; i < TARGET_SITES.length; i++) {
-		    			final String site = TARGET_SITES[i];
-		    			InternetObject obj = statusMap.get(site);
+					for (int i = 0; i < siteList.size(); i++) {
+		    			final String site = siteList.get(i);
+		    			WebObject obj = statusMap.get(site);
 		    			if (obj == null) {
-		    				obj = new InternetObject();
+		    				obj = new WebObject();
 		    				obj.site = site;
 		    				statusMap.put(site, obj);
 		    			}
@@ -63,13 +99,13 @@ public void run() {
 			    			Unirest.get(site).asStringAsync(new Callback<String>() {
 
 								public void failed(UnirestException e) {
-									InternetObject obj = statusMap.get(site);
+									WebObject obj = statusMap.get(site);
 									obj.endTime = System.currentTimeMillis();
 									obj.status = WebStatusEnum.FAILED;
 								}
 
 								public void completed(HttpResponse<String> response) {
-									InternetObject obj = statusMap.get(site);
+									WebObject obj = statusMap.get(site);
 									obj.endTime = System.currentTimeMillis();
 									obj.status = WebStatusEnum.DONE;
 								}
@@ -86,7 +122,7 @@ private void sendCounterData() {
 					try {
 						JSONArray jsonArray = new JSONArray();
 						for (String key : statusMap.keySet()) {
-							InternetObject obj = statusMap.get(key);
+							WebObject obj = statusMap.get(key);
 							JSONObject element = null;
 							JSONObject objectJson = null;
 							JSONArray countersArray = null;
@@ -103,7 +139,7 @@ private void sendCounterData() {
 									countersArray = new JSONArray();
 									element.put("counters", countersArray);
 									counterJson = new JSONObject();
-									counterJson.put("name",  "WebTime");
+									counterJson.put("name",  "Time");
 									counterJson.put("value", obj.endTime - obj.startTime);
 									countersArray.put(counterJson);
 									obj.status = WebStatusEnum.READY;
@@ -134,7 +170,7 @@ private void sendCounterData() {
 							}
 						}
 
-						HttpResponse<JsonNode> response = Unirest.post("http://127.0.0.1:6180/counter")
+						HttpResponse<JsonNode> response = Unirest.post("http://" + ip + ":" + port + "/counter")
 						    	.header("accept", "application/json")
 						    	.header("content-type", "application/json")
 						    	.body(jsonArray).asJson();
@@ -152,17 +188,16 @@ private static int registerMetadata() throws UnirestException {
     	JSONObject rootJson = new JSONObject();
     	JSONObject objectJson = new JSONObject();
     	objectJson.put("type", "website");
-    	objectJson.put("display", "WebSite");
+    	objectJson.put("display", "Web");
     	rootJson.put("object", objectJson);
     	JSONArray counterArray = new JSONArray();
     	rootJson.put("counters", counterArray);
 		JSONObject counterMap = new JSONObject();
 		counterArray.put(counterMap);
-		counterMap.put("name", "WebTime");
+		counterMap.put("name", "Time");
 		counterMap.put("unit", "ms");
 
-    	
-    	HttpResponse<JsonNode> response = Unirest.post("http://127.0.0.1:6180/register")
+    	HttpResponse<JsonNode> response = Unirest.post("http://" + ip + ":" + port + "/register")
     	.header("accept", "application/json")
     	.header("content-type", "application/json")
     	.body(rootJson).asJson();

pulse.sample.webtime/src/main/java/scouter/pulse/webtime/InternetObject.java renamed to pulse.sample.webtime/src/main/java/scouter/pulse/webtime/WebObject.java

@@ -1,6 +1,6 @@
 package scouter.pulse.webtime;
 
-public class InternetObject {
+public class WebObject {
 	public String site;
 	WebStatusEnum status = WebStatusEnum.READY;
 	long startTime;