/*
 *  Copyright 2010 salaboy.
 * 
 *  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 at
 * 
 *       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.
 *  under the License.
 */
package org.drools.distributed.directory.impl;

import java.io.Serializable;
import java.rmi.RemoteException;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import java.util.logging.Level;
import java.util.logging.Logger;
import org.drools.grid.ConnectorException;
import org.drools.grid.ConnectorType;
import org.drools.grid.DirectoryNodeService;
import org.drools.grid.GenericConnection;
import org.drools.grid.GenericNodeConnector;
import org.drools.grid.NodeConnectionType;

/**
 *
 * @author salaboy 
 */
public class DistributedConnectionDirectory implements NodeConnectionType, Serializable {

    private final Map<Class<?>, Object> services = new ConcurrentHashMap<Class<?>, Object>();
    private GenericNodeConnector connector;
    private GenericConnection connection;

    public DistributedConnectionDirectory() {
    }

    


    public DistributedConnectionDirectory(GenericNodeConnector connector, GenericConnection connection) {
        this.connector = connector;
        this.connection = connection;
    }

    

    public void init() {
       
            services.put(DirectoryNodeService.class, new DirectoryNodeServiceGridClient(connector, connection));
       
    }

    public Set<Class<?>> getServicesKeys() {
        return services.keySet();
    }

    public <T> T getServiceImpl(Class<T> clazz) {
        return (T) services.get(clazz);
    }

    public void setConnector(GenericNodeConnector connector) {
        this.connector = connector;
    }

    public void setConnection(GenericConnection connection) {
        this.connection = connection;
    }

    public ConnectorType getConnectorType() {
        return ConnectorType.DISTRIBUTED;
    }
}