asterix-metadata/src/main/java/edu/uci/ics/asterix/metadata/bootstrap/AsterixStateProxy.java

/*
 * Copyright 2009-2010 by The Regents of the University of California
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * you may obtain a copy of the License from
 * 
 *     http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package edu.uci.ics.asterix.metadata.bootstrap;

import java.rmi.RemoteException;
import java.rmi.server.UnicastRemoteObject;
import java.util.HashMap;
import java.util.logging.Logger;

import edu.uci.ics.asterix.metadata.api.IAsterixStateProxy;
import edu.uci.ics.asterix.metadata.api.IMetadataNode;

/**
 * Contains Asterix distributed state such as the AsterixProperties.
 */
public class AsterixStateProxy implements IAsterixStateProxy {
    private static final long serialVersionUID = 1L;
    private static final Logger LOGGER = Logger.getLogger(AsterixStateProxy.class.getName());

    private IMetadataNode metadataNode;
    private AsterixProperties asterixProperties;
    private static final IAsterixStateProxy cc = new AsterixStateProxy();
    private final HashMap<String, Object> nodeStateMap = new HashMap<String, Object>();

    public static IAsterixStateProxy registerRemoteObject() throws RemoteException {
        IAsterixStateProxy stub = (IAsterixStateProxy) UnicastRemoteObject.exportObject(cc, 0);
        LOGGER.info("Asterix Distributed State Proxy Bound");
        return stub;
    }

    public static void unregisterRemoteObject() throws RemoteException {
        UnicastRemoteObject.unexportObject(cc, true);
        LOGGER.info("Asterix Distributed State Proxy Unbound");
    }

    @Override
    public void setMetadataNode(IMetadataNode metadataNode) throws RemoteException {
        this.metadataNode = metadataNode;
    }

    @Override
    public IMetadataNode getMetadataNode() throws RemoteException {
        return this.metadataNode;
    }

    @Override
    public void setAsterixProperties(AsterixProperties asterixProperity) throws RemoteException {
        this.asterixProperties = asterixProperity;
    }

    @Override
    public AsterixProperties getAsterixProperties() throws RemoteException {
        return this.asterixProperties;
    }

    @Override
    synchronized public Object getAsterixNodeState(String nodeName) throws RemoteException {
        return nodeStateMap.get(nodeName);
    }

    @Override
    synchronized public void setAsterixNodeState(String nodeName, Object ns) throws RemoteException {
        nodeStateMap.put(nodeName, ns);
    }
}