/*
*
*
* Copyright 1997-2006 BBNT Solutions, LLC
* under sponsorship of the Defense Advanced Research Projects
* Agency (DARPA).
*
* You can redistribute this software and/or modify it under the
* terms of the Cougaar Open Source License as published on the
* Cougaar Open Source Website (www.cougaar.org).
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
*
*/
package org.cougaar.demo.ping;
import java.io.IOException;
import java.io.PrintWriter;
import java.text.DecimalFormat;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.cougaar.core.mts.MessageAddress;
import org.cougaar.core.relay.SimpleRelay;
import org.cougaar.core.service.BlackboardQueryService;
import org.cougaar.core.servlet.ComponentServlet;
import org.cougaar.util.IsInstanceOf;
/**
* This servlet shows our ping relays as an HTML page.
*
* Supports an optional Servlet path parameter, which defaults to "/ping".
*
* For simplicity, it's easiest to load a copy of this servlet into every agent.
*/
public class PingServlet extends ComponentServlet {
private static final Comparator RELAY_COMPARATOR =
new SimpleRelayComparator();
private long loadTime;
private BlackboardQueryService blackboard;
/** @return a default path if a plugin parameter is not specified */
protected String getPath() {
String ret = super.getPath();
return ret == null ? "/ping" : ret;
}
/** This method is called when the agent is created */
public void load() {
super.load();
// Record our load time
loadTime = System.currentTimeMillis();
// Get our required Cougaar services
this.blackboard =
getServiceBroker().getService(this, BlackboardQueryService.class, null);
}
/** This method is called whenever the browser loads our URL. */
public void doGet(
HttpServletRequest request,
HttpServletResponse response) throws IOException {
// Begin our HTML page response
response.setContentType("text/html");
PrintWriter out = response.getWriter();
String title = "Agent " + getEncodedAgentName();
out.println(
"" + title +
"" + title + "
");
// Write how long we've been running, to make it easy for the user
// to calculate the ping throughput
long runTime = System.currentTimeMillis() - loadTime;
out.println("Milliseconds since agent load: " + runTime + "");
// Find all relays and sort them by source then target
@SuppressWarnings("unchecked")
Collection col =
blackboard.query(new IsInstanceOf(SimpleRelay.class));
List l1 = new ArrayList(col);
Collections.sort(l1, RELAY_COMPARATOR);
List l = l1;
// Write the relays as an HTML table
out.println(
" | UID | " +
"Source | Target | Content | " +
"Response | Pings Per Second |
|---|
");
DecimalFormat formatter = new DecimalFormat("0.000");
for (int i = 0; i < l.size(); i++) {
SimpleRelay relay = l.get(i);
double throughput = Double.NaN;
if (relay.getQuery() instanceof Number && runTime > 0) {
double count = ((Number) relay.getQuery()).doubleValue();
double seconds = runTime / 1000;
throughput = count / seconds;
}
out.println(
"| " + i + " | " + relay.getUID() +
" | " + relay.getSource() + " | " +
relay.getTarget() + " | " + relay.getQuery() +
" | " + relay.getReply() + " | " +
formatter.format(throughput) + " |
");
}
out.println("
");
// Create a "reload" button for the user to invoke our servlet again
out.println(
"");
// End our HTML page
out.println("");
}
private static class SimpleRelayComparator implements Comparator {
public int compare(SimpleRelay r1, SimpleRelay r2) {
int ret;
ret = compareAddresses(r1.getSource(), r2.getSource());
if (ret != 0) {
return ret;
}
ret = compareAddresses(r1.getTarget(), r2.getTarget());
if (ret != 0) {
return ret;
}
return compare(r1.getUID(), r2.getUID());
}
private int compareAddresses(MessageAddress m1, MessageAddress m2) {
String a = m1 == null ? null : m1.getAddress();
String b = m2 == null ? null : m2.getAddress();
return compare(a, b);
}
@SuppressWarnings("unchecked")
private int compare(Comparable a, Comparable b) {
return a == null ? (b == null ? 0 : 1) : b == null ? -1 : a.compareTo(b);
}
}
}