package org.jboss.ha.framework.server.util;
import java.util.ArrayList;
import java.util.Hashtable;
import javax.naming.InitialContext;
import javax.naming.NamingException;
import javax.naming.Context;
import org.jboss.system.ServiceMBeanSupport;
import org.jboss.logging.Logger;
import org.jboss.ha.framework.server.util.TopologyMonitorService.AddressPort;
public class PingJndi extends ServiceMBeanSupport
implements PingJndiMBean
{
private String urlPrefix;
private String urlSuffix;
private String urlPattern;
private String[] lookupNames;
public String[] getLookupNames()
{
return lookupNames;
}
public void setLookupNames(String[] names)
{
this.lookupNames = names;
}
public String getProviderURLPattern()
{
return urlPattern;
}
public void setProviderURLPattern(String regex)
{
this.urlPattern = regex;
this.urlPrefix = regex;
this.urlSuffix = "";
String hostExp = "{host}";
int hostIndex = regex.indexOf(hostExp);
if( hostIndex >= 0 )
{
urlPrefix = regex.substring(0, hostIndex);
int endIndex = hostIndex + hostExp.length();
urlSuffix = regex.substring(endIndex);
}
}
public void membershipChanged(ArrayList removed, ArrayList added,
ArrayList members, String logCategoryName)
{
log.debug("membershipChanged");
Logger tmsLog = Logger.getLogger(logCategoryName);
Hashtable localEnv = null;
try
{
InitialContext localCtx = new InitialContext();
localEnv = localCtx.getEnvironment();
}
catch(NamingException e)
{
tmsLog.error("Failed to obtain InitialContext env", e);
return;
}
tmsLog.info("Checking removed hosts JNDI binding");
doLookups(localEnv, tmsLog, removed);
tmsLog.info("Checking added hosts JNDI binding");
doLookups(localEnv, tmsLog, added);
tmsLog.info("Checking members hosts JNDI binding");
doLookups(localEnv, tmsLog, members);
}
private void doLookups(Hashtable localEnv, Logger tmsLog, ArrayList nodes)
{
for(int n = 0; n < nodes.size(); n ++)
{
AddressPort addrInfo = (AddressPort) nodes.get(n);
String providerURL = urlPrefix + addrInfo.getHostName() + urlSuffix;
Hashtable env = new Hashtable(localEnv);
env.put(Context.PROVIDER_URL, providerURL);
tmsLog.info("Checking names on: "+addrInfo);
try
{
InitialContext ctx = new InitialContext(env);
for(int s = 0; s < lookupNames.length; s ++)
{
String name = lookupNames[s];
Object value = ctx.lookup(name);
tmsLog.info("lookup("+name+"): "+value);
}
}
catch(Exception e)
{
tmsLog.error("Failed lookups on: "+addrInfo, e);
}
}
}
}