org.eclipse.osee.ote.connection.jini/src/org/eclipse/osee/ote/connection/jini/JiniServiceSideConnector.java

/*******************************************************************************
 * Copyright (c) 2004, 2007 Boeing.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     Boeing - initial API and implementation
 *******************************************************************************/
package org.eclipse.osee.ote.connection.jini;

import java.io.File;
import java.io.Serializable;
import java.net.URI;
import java.net.UnknownHostException;
import java.rmi.Remote;
import java.rmi.server.ExportException;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Timer;
import java.util.logging.Level;
import net.jini.core.entry.Entry;
import net.jini.core.lookup.ServiceID;
import net.jini.core.lookup.ServiceItem;
import net.jini.core.lookup.ServiceRegistrar;
import net.jini.core.lookup.ServiceRegistration;
import net.jini.id.Uuid;
import net.jini.id.UuidFactory;
import net.jini.jeri.BasicILFactory;
import net.jini.jeri.BasicJeriExporter;
import net.jini.jeri.tcp.TcpServerEndpoint;
import org.eclipse.osee.framework.jdk.core.util.EnhancedProperties;
import org.eclipse.osee.framework.jdk.core.util.Network;
import org.eclipse.osee.ote.connection.jini.util.LeaseRenewTask;

/**
 * @author Ken J. Aguilar
 */
public class JiniServiceSideConnector extends JiniConnector implements IJiniConnectorLink {
   public static final String TYPE = "jini.service-end";
   private final HashMap<ServiceRegistration, LeaseRenewTask> registrations =
      new HashMap<ServiceRegistration, LeaseRenewTask>();
   private Remote service;
   private Remote serviceProxy;
   private BasicJeriExporter serviceExporter;
   private ServiceID serviceId;
   private final Timer timer = new Timer();
   private ServiceItem serviceItem;
   private BasicJeriExporter linkExporter;
   private IJiniConnectorLink exportedThis;
   private boolean stopped = false;

   public JiniServiceSideConnector() {
      super();
   }

   @Override
   public void init(Object service) throws UnknownHostException, ExportException {

      this.service = (Remote) service;
      serviceId = generateServiceId();
      serviceExporter =
         new BasicJeriExporter(TcpServerEndpoint.getInstance(Network.getValidIP().getHostAddress(), 0),
            new BasicILFactory(null, null, Activator.getDefault().getExportClassLoader()), false, false);
      linkExporter =
         new BasicJeriExporter(TcpServerEndpoint.getInstance(Network.getValidIP().getHostAddress(), 0),
            new BasicILFactory(null, null, Activator.getDefault().getExportClassLoader()), false, false);
      serviceProxy = serviceExporter.export(this.service);
      exportedThis = (IJiniConnectorLink) linkExporter.export(this);
      setProperty(LINK_PROPERTY, (Serializable) exportedThis);
      serviceItem = new ServiceItem(serviceId, serviceProxy, createEntries());
   }

   public JiniServiceSideConnector(Remote service, EnhancedProperties props) throws UnknownHostException, ExportException {
      super(props);
      this.service = service;
      serviceId = generateServiceId();
      serviceExporter =
         new BasicJeriExporter(TcpServerEndpoint.getInstance(Network.getValidIP().getHostAddress(), 0),
            new BasicILFactory(null, null, Activator.getDefault().getExportClassLoader()), false, false);
      linkExporter =
         new BasicJeriExporter(TcpServerEndpoint.getInstance(Network.getValidIP().getHostAddress(), 0),
            new BasicILFactory(null, null, Activator.getDefault().getExportClassLoader()), false, false);
      serviceProxy = serviceExporter.export(service);
      exportedThis = (IJiniConnectorLink) linkExporter.export(this);
      props.setProperty(LINK_PROPERTY, (Serializable) exportedThis);
      serviceItem = new ServiceItem(serviceId, serviceProxy, createEntries());
   }

   @Override
   public Remote getService() {
      return serviceProxy;
   }

   @Override
   public synchronized void stop() throws Exception {
      if (stopped) {
         return;
      }
      stopped = true;
      super.stop();
      removeAllRegistrations();
      serviceExporter.unexport(true);
      linkExporter.unexport(true);
   }

   /**
    * this method will cancel all current registrations of this connector
    */
   synchronized void removeAllRegistrations() {
      for (ServiceRegistration registration : registrations.keySet()) {
         try {
            LeaseRenewTask task = registrations.get(registration);
            if (task != null) {
               task.cancel();
            }
         } catch (Exception e) {
            Activator.log(Level.SEVERE, "exception removing registration", e);
         }
      }
      registrations.clear();
   }

   private ServiceID generateServiceId() {
      Uuid uuid = UuidFactory.generate();
      Long lsb = new Long(uuid.getLeastSignificantBits());
      Long msb = new Long(uuid.getMostSignificantBits());
      return new ServiceID(msb.longValue(), lsb.longValue());
   }

   synchronized void addRegistration(ServiceRegistration registration, ServiceRegistrar registrar) {
      registrations.put(registration, new LeaseRenewTask(timer, registration, registrar));
   }

   ServiceItem getServiceItem() {
      return serviceItem;
   }

   private synchronized void setAttributes(Entry[] entry) {
      Iterator<ServiceRegistration> iter = registrations.keySet().iterator();
      while (iter.hasNext()) {
         ServiceRegistration registration = iter.next();
         try {
            registration.setAttributes(entry);
         } catch (Exception ex) {
            Activator.log(Level.SEVERE, "exception setting attributes", ex);
            registrations.remove(registration);
         }
      }
   }

   @Override
   public String getConnectorType() {
      return TYPE;
   }

   @Override
   public void setProperty(String key, Serializable value) {
      super.setProperty(key, value);
      setAttributes(createEntries());
   }

   @Override
   public URI upload(File file) throws Exception {
      return null;
   }

   @Override
   public boolean ping() {
      return true;
   }

   @Override
   public String getUniqueServerId() {
      return serviceId.toString();
   }

}