MemoryMapProxy.java

/*******************************************************************************
 * Copyright (c) 2009, 2012 Wind River Systems, Inc. and others.
 * 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:
 *     Wind River Systems - initial API and implementation
 *******************************************************************************/
package org.eclipse.tcf.internal.services.remote;

import java.io.IOException;
import java.util.Collection;
import java.util.HashMap;
import java.util.Map;

import org.eclipse.tcf.core.Command;
import org.eclipse.tcf.protocol.IChannel;
import org.eclipse.tcf.protocol.IToken;
import org.eclipse.tcf.protocol.JSON;
import org.eclipse.tcf.services.IMemoryMap;
import org.eclipse.tcf.util.TCFMemoryRegion;

public class MemoryMapProxy implements IMemoryMap {

    private final IChannel channel;
    private final Map<MemoryMapListener,IChannel.IEventListener> listeners =
        new HashMap<MemoryMapListener,IChannel.IEventListener>();

    public MemoryMapProxy(IChannel channel) {
        this.channel = channel;
    }

    public String getName() {
        return NAME;
    }

    public IToken get(String id, final DoneGet done) {
        return new Command(channel, this, "get", new Object[]{ id }) {
            @Override
            public void done(Exception error, Object[] args) {
                MemoryRegion[] map = null;
                if (error == null) {
                    assert args.length == 2;
                    error = toError(args[0]);
                    map = toMemoryMap(args[1]);
                }
                done.doneGet(token, error, map);
            }
        }.token;
    }

    public IToken set(String id, MemoryRegion[] map, final DoneSet done) {
        return new Command(channel, this, "set", new Object[]{ id, map }) {
            @Override
            public void done(Exception error, Object[] args) {
                if (error == null) {
                    assert args.length == 1;
                    error = toError(args[0]);
                }
                done.doneSet(token, error);
            }
        }.token;
    }

    @SuppressWarnings("unchecked")
    private MemoryRegion[] toMemoryMap(Object o) {
        if (o == null) return null;
        int i = 0;
        Collection<Object> c = (Collection<Object>)o;
        MemoryRegion[] map = new MemoryRegion[c.size()];
        for (Object x : c) map[i++] = toMemoryRegion(x);
        return map;
    }

    @SuppressWarnings("unchecked")
    private MemoryRegion toMemoryRegion(Object o) {
        if (o == null) return null;
        return new TCFMemoryRegion((Map<String,Object>)o);
    }

    public void addListener(final MemoryMapListener listener) {
        IChannel.IEventListener l = new IChannel.IEventListener() {

            public void event(String name, byte[] data) {
                try {
                    Object[] args = JSON.parseSequence(data);
                    if (name.equals("changed")) {
                        assert args.length == 1;
                        listener.changed((String)args[0]);
                    }
                    else {
                        throw new IOException("Memory Map service: unknown event: " + name);
                    }
                }
                catch (Throwable x) {
                    channel.terminate(x);
                }
            }
        };
        channel.addEventListener(this, l);
        listeners.put(listener, l);
    }

    public void removeListener(MemoryMapListener listener) {
        IChannel.IEventListener l = listeners.remove(listener);
        if (l != null) channel.removeEventListener(this, l);
    }

    static {
        JSON.addObjectWriter(MemoryRegion.class, new JSON.ObjectWriter<MemoryRegion>() {
            public void write(MemoryRegion r) throws IOException {
                JSON.writeObject(r.getProperties());
            }
        });
    }
}