【源码品读】IPing机制解析

/**
 * Interface that defines how we "ping" a server to check if its alive
 * @author stonse
 *
 */
public interface IPing {
    
    /**
     * Checks whether the given <code>Server</code> is "alive" i.e. should be
     * considered a candidate while loadbalancing
     * 
     */
    public boolean isAlive(Server server);
}

public class DummyPing extends AbstractLoadBalancerPing {

    public DummyPing() {
    }

    public boolean isAlive(Server server) {
        return true;
    }

    @Override
    public void initWithNiwsConfig(IClientConfig clientConfig) {
    }
}

public class NoOpPing implements IPing {

    @Override
    public boolean isAlive(Server server) {
        return true;
    }

}

public class PingUrl implements IPing {

public String getPingAppendString() {
		return pingAppendString;
}

String pingAppendString = "";

public void setPingAppendString(String pingAppendString) {
		this.pingAppendString = (pingAppendString != null) ? pingAppendString : "";
}

public String getExpectedContent() {
	return expectedContent;
}

String expectedContent = null;

public void setExpectedContent(String expectedContent) {
	this.expectedContent = expectedContent;
}

public boolean isAlive(Server server) {
		String urlStr   = "";
		if (isSecure){
			urlStr = "https://";
		}else{
			urlStr = "http://";
		}
		urlStr += server.getId();
		urlStr += getPingAppendString();

		boolean isAlive = false;

		HttpClient httpClient = new DefaultHttpClient();
		HttpUriRequest getRequest = new HttpGet(urlStr);
		String content=null;
		try {
			HttpResponse response = httpClient.execute(getRequest);
			content = EntityUtils.toString(response.getEntity());
			isAlive = (response.getStatusLine().getStatusCode() == 200);
			if (getExpectedContent()!=null){
				LOGGER.debug("content:" + content);
				if (content == null){
					isAlive = false;
				}else{
					if (content.equals(getExpectedContent())){
						isAlive = true;
					}else{
						isAlive = false;
					}
				}
			}
		} catch (IOException e) {
			e.printStackTrace();
		}finally{
			// Release the connection.
			getRequest.abort();
		}

		return isAlive;
}

public static void main(String[] args){
	PingUrl p = new PingUrl(false,"/cs/hostRunning");
	p.setExpectedContent("true");
	Server s = new Server("ec2-75-101-231-85.compute-1.amazonaws.com", 7101);
	
	boolean isAlive = p.isAlive(s);
	System.out.println("isAlive:" + isAlive);
}

public class NIWSDiscoveryPing extends AbstractLoadBalancerPing {

public boolean isAlive(Server server) {
	boolean isAlive = true;
	if (server!=null && server instanceof DiscoveryEnabledServer){
		DiscoveryEnabledServer dServer = (DiscoveryEnabledServer)server;	            
		InstanceInfo instanceInfo = dServer.getInstanceInfo();
		if (instanceInfo!=null){	                
			InstanceStatus status = instanceInfo.getStatus();
			if (status!=null){
				isAlive = status.equals(InstanceStatus.UP);
			}
		}
	}
	return isAlive;
}