package fr.in2p3.jsaga.adaptor.job;
   
   import fr.in2p3.jsaga.adaptor.base.defaults.Default;
   import fr.in2p3.jsaga.adaptor.base.usage.*;
   import fr.in2p3.jsaga.adaptor.job.monitor.JobStatusNotifier;
   import fr.in2p3.jsaga.adaptor.job.monitor.ListenIndividualJob;
   import org.apache.log4j.Logger;
   import org.globus.gram.*;
   import org.globus.gram.internal.GRAMConstants;
  import org.ietf.jgss.GSSException;
  import org.ogf.saga.error.*;
  
  import java.net.InetAddress;
  import java.net.UnknownHostException;
  import java.util.Map;
  
  /* ***************************************************
  * *** Centre de Calcul de l'IN2P3 - Lyon (France) ***
  * ***             http://cc.in2p3.fr/             ***
  * ***************************************************
  * File:   GatekeeperJobMonitorAdaptor
  * Author: Sylvain Reynaud (sreynaud@in2p3.fr)
  * Date:   16 nov. 2007
  * ***************************************************
  * Description:                                      */
  /**
   *
   */
  public class GatekeeperJobMonitorAdaptor extends GkCommonJobMonitorAdaptor implements ListenIndividualJob  {
  	private Logger logger = Logger.getLogger(GatekeeperJobMonitorAdaptor.class.getName());	
  
      /** override super.getType() */
      public String getType() {
          return "gatekeeper";
      }
  
      /** override super.getUsage() */
      public Usage getUsage() {
          return new UAnd.Builder()
                          .and(new UOptional(IP_ADDRESS))
                          .and(new UOptional(TCP_PORT_RANGE))
                          .build();
      }
  
      /** override super.getDefaults() */
      public Default[] getDefaults(Map attributes) throws IncorrectStateException {
      	try {
  			String defaultIp = InetAddress.getLocalHost().getHostAddress();
  			String defaultTcpPortRange="40000,45000";
  			return new Default[]{new Default(IP_ADDRESS, defaultIp),new Default(TCP_PORT_RANGE, defaultTcpPortRange)};
  		} catch (UnknownHostException e) {
  			return null;
  		}
      }
  
      public void subscribeJob(String nativeJobId, JobStatusNotifier notifier) throws TimeoutException, NoSuccessException {
          GramJob job = getGramJobById(nativeJobId);
          GramJobListener listener = new GatekeeperJobStatusListener(notifier);
          job.addListener(listener);
          try {
          	job.bind();
          } catch (GramException e) {
          	job.removeListener(listener);
              if (e.getErrorCode() == GramException.CONNECTION_FAILED ||
              		e.getErrorCode() == GramException.JOB_QUERY_DENIAL ||
              		e.getErrorCode() == GramException.HTTP_UNFRAME_FAILED) {
                  //WARNING: Globus does not distinguish job DONE and job manager stopped
              	logger.warn("Globus job manager may be stopped: status DONE returned in subscribeJob() for job "+nativeJobId);
                  GatekeeperJobStatus status = new GatekeeperJobStatus(nativeJobId, new Integer(GRAMConstants.STATUS_DONE), "DONE");
                  notifier.notifyChange(status);
              } else {
                  super.rethrowException(e);
              }
          } catch (GSSException e) {
              job.removeListener(listener);
              throw new NoSuccessException(e);
          }
      }
  
      public void unsubscribeJob(String nativeJobId) throws TimeoutException, NoSuccessException {
          GramJob job = getGramJobById(nativeJobId);
          try {
              job.unbind();
          } catch (GramException e) {
          	if (e.getErrorCode() == GramException.CONNECTION_FAILED ||
              		e.getErrorCode() == GramException.JOB_QUERY_DENIAL ||
              		e.getErrorCode() == GramException.HTTP_UNFRAME_FAILED) {
                  // ignore (Globus does not distinguish job DONE and job manager stopped)
              } else {
                  super.rethrowException(e);
              }
          } catch (GSSException e) {
              throw new NoSuccessException(e);
          }
          //NOTICE: no need to remove GramJobListener since job is unregistered from CallbackHandler
      }
  }