org/objectweb/proactive/core/body/BodyMap.java

/* 
 * ################################################################
 * 
 * ProActive: The Java(TM) library for Parallel, Distributed, 
 *            Concurrent computing with Security and Mobility
 * 
 * Copyright (C) 1997-2007 INRIA/University of Nice-Sophia Antipolis
 * Contact: proactive@objectweb.org
 * 
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or any later version.
 *  
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307
 * USA
 *  
 *  Initial developer(s):               The ProActive Team
 *                        http://www.inria.fr/oasis/ProActive/contacts.html
 *  Contributor(s): 
 * 
 * ################################################################
 */ 
package org.objectweb.proactive.core.body;

import java.util.Hashtable;

import org.objectweb.proactive.Body;
import org.objectweb.proactive.core.UniqueID;
import org.objectweb.proactive.core.event.AbstractEventProducer;
import org.objectweb.proactive.core.event.BodyEvent;
import org.objectweb.proactive.core.event.BodyEventListener;
import org.objectweb.proactive.core.event.ProActiveEvent;
import org.objectweb.proactive.core.event.ProActiveListener;


public class BodyMap extends AbstractEventProducer implements Cloneable,
    java.io.Externalizable {
    //
    // -- PRIVATE MEMBER -----------------------------------------------
    //
    private Hashtable<UniqueID, UniversalBody> idToBodyMap;

    //
    // -- CONSTRUCTORS -----------------------------------------------
    //
    public BodyMap() {
        idToBodyMap = new Hashtable<UniqueID, UniversalBody>();
    }

    //
    // -- PUBLIC METHODS -----------------------------------------------
    // 

    public synchronized void putBody(UniqueID id, UniversalBody b) {
        while (idToBodyMap.get(id) != null) {
            try {
                wait();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
        idToBodyMap.put(id, b);

        if (hasListeners()) {
            notifyAllListeners(new BodyEvent(b, BodyEvent.BODY_CREATED));
        }
    }

    public synchronized void updateBody(UniqueID id, UniversalBody b) {
        //remove old reference
        if (idToBodyMap.get(id) != null) {
            idToBodyMap.remove(id);
        }

        //add new reference
        idToBodyMap.put(id, b);

        if (hasListeners()) {
            notifyAllListeners(new BodyEvent(b, BodyEvent.BODY_CREATED));
        }
    }

    public synchronized void removeBody(UniqueID id) {
        UniversalBody b = idToBodyMap.remove(id);
        notifyAll();

        if ((b != null) && hasListeners()) {
            notifyAllListeners(new BodyEvent(b, BodyEvent.BODY_DESTROYED));
        }
    }

    public synchronized int size() {
        int val = idToBodyMap.size();

        return val;
    }

    public synchronized UniversalBody getBody(UniqueID id) {
        Object o = null;
        if (id != null) {
            o = idToBodyMap.get(id);
        }

        return (UniversalBody) o;
    }

    public synchronized boolean containsBody(UniqueID id) {
        return idToBodyMap.containsKey(id);
    }

    public java.util.Iterator<UniversalBody> bodiesIterator() {
        return idToBodyMap.values().iterator();
    }

    public synchronized String toString() {
        StringBuffer sb = new StringBuffer();

        sb.append(" -- BodyMap ------- \n");

        java.util.Set entrySet = idToBodyMap.entrySet();
        java.util.Iterator iterator = entrySet.iterator();

        while (iterator.hasNext()) {
            java.util.Map.Entry entry = (java.util.Map.Entry) iterator.next();
            sb.append(entry.getKey()).append("  body = ")
              .append(entry.getValue()).append("\n");
        }

        return sb.toString();
    }

    //
    // -- implements Cloneable -----------------------------------------------
    //
    public Object clone() {
        BodyMap newLocationTable = new BodyMap();
        newLocationTable.idToBodyMap = (Hashtable<UniqueID, UniversalBody>)idToBodyMap.clone();

        return newLocationTable;
    }

    //
    // -- methods for BodyEventProducer -----------------------------------------------
    //
    public void addBodyEventListener(BodyEventListener listener) {
        addListener(listener);
    }

    public void removeBodyEventListener(BodyEventListener listener) {
        removeListener(listener);
    }

    //
    // -- implements Externalizable -----------------------------------------------
    //

    public synchronized void readExternal(java.io.ObjectInput in)
        throws java.io.IOException, ClassNotFoundException {
        int size = in.readInt();

        for (int i = 0; i < size; i++) {
            UniqueID id = (UniqueID) in.readObject();
            UniversalBody remoteBody = (UniversalBody) in.readObject();
            idToBodyMap.put(id, remoteBody);
        }
    }

    public synchronized void writeExternal(java.io.ObjectOutput out)
        throws java.io.IOException {
        int size = idToBodyMap.size();
        out.writeInt(size);

        java.util.Set entrySet = idToBodyMap.entrySet();
        java.util.Iterator iterator = entrySet.iterator();

        while (iterator.hasNext()) {
            java.util.Map.Entry entry = (java.util.Map.Entry) iterator.next();
            out.writeObject(entry.getKey());

            Object value = entry.getValue();

            if (value instanceof Body) {
                out.writeObject(((Body) value).getRemoteAdapter());
            } else {
                out.writeObject(value);
            }
        }
    }

    //
    // -- PROTECTED METHODS -----------------------------------------------
    //
    protected void notifyOneListener(ProActiveListener listener,
        ProActiveEvent event) {
        BodyEvent bodyEvent = (BodyEvent) event;

        switch (bodyEvent.getType()) {
        case BodyEvent.BODY_CREATED:
            ((BodyEventListener) listener).bodyCreated(bodyEvent);
            break;
        case BodyEvent.BODY_DESTROYED:
            ((BodyEventListener) listener).bodyDestroyed(bodyEvent);
            break;
        }
    }
}

---