package com.csii.rmi;

import java.rmi.server.RMIClientSocketFactory;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import com.csii.service.comm.rmi.RMINaming;

public class RMILocator {
	  
	  private RMIClientSocketFactory rmiClientSocketFactory;
	  private static Log log = LogFactory.getLog(RMILocator.class);
	  private transient boolean inited;
	  private transient SecurityManager securityManager;
	  private String registryHost = "";

	  private int registryPort = 7700;
	  private int recoverDelay;
	  private int recoverRetry;

	  public void setRecoverDelay(int recoverDelay)
	  {
	    this.recoverDelay = recoverDelay;
	  }

	  public void setRecoverRetry(int recoverRetry) {
	    this.recoverRetry = recoverRetry;
	  }

	  public void setRmiClientSocketFactory(RMIClientSocketFactory rmiClientSocketFactory)
	  {
	    this.rmiClientSocketFactory = rmiClientSocketFactory;
	  }

	  public void setRegistryHost(String registryHost) {
	    this.registryHost = registryHost;
	  }

	  public void setRegistryPort(int registryPort) {
	    this.registryPort = registryPort;
	  }

	  private synchronized void init() {
	    if (!this.inited) {
	      if (this.securityManager != null)
	        System.setSecurityManager(this.securityManager);
	      this.inited = true;
	    }
	  }

	  public Object lookup(String name) {
	    init();

	    if (name == null)
	      throw new IllegalArgumentException("lookup name is null.");
	    if (!name.startsWith("//"))
	      name = "//" + this.registryHost + ":" + this.registryPort + "/" + name;
	    try
	    {
	      Object o = RMINaming.lookup(name, this.rmiClientSocketFactory);

	      log.info("lookup and found " + name);
	      return o;
	    } catch (Exception e) {
	      throw new RuntimeException("Lookup Object from registry error.", e);
	    }
	  }

	  public String[] list() {
	    init();
	    try
	    {
	      return RMINaming.list("//" + this.registryHost + ":" + this.registryPort + "/", this.rmiClientSocketFactory);
	    } catch (Exception e) {
	      throw new RuntimeException("List registry Objects error.", e);
	    }
	  }
}