hub/core/src/main/java/org/apache/directory/server/hub/core/HubClientManager.java

/*
 *   Licensed to the Apache Software Foundation (ASF) under one
 *   or more contributor license agreements.  See the NOTICE file
 *   distributed with this work for additional information
 *   regarding copyright ownership.  The ASF licenses this file
 *   to you 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.
 *
 */

package org.apache.directory.server.hub.core;


import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Set;

import org.apache.commons.collections.MultiMap;
import org.apache.commons.collections.map.MultiValueMap;
import org.apache.directory.server.hub.api.AbstractComponentListener;
import org.apache.directory.server.hub.api.ComponentHub;
import org.apache.directory.server.hub.api.ComponentListener;
import org.apache.directory.server.hub.api.component.DcConfiguration;
import org.apache.directory.server.hub.api.component.DirectoryComponent;
import org.apache.directory.server.hub.api.exception.HubAbortException;
import org.apache.directory.server.hub.api.meta.DcMetadataDescriptor;


public class HubClientManager
{
    private ComponentHub hub;

    private MultiMap typeRegistrations = new MultiValueMap();
    private MultiMap clientRegistrations = new MultiValueMap();


    public synchronized void registerHubClient( ComponentListener hubClient, String type )
    {
        typeRegistrations.put( type, hubClient );
        clientRegistrations.put( hubClient, type );

        List<DirectoryComponent> existing = hub.getDCRegistry().getComponents();
        if ( existing == null )
        {
            return;
        }

        for ( DirectoryComponent component : existing )
        {
            DcMetadataDescriptor metadata = hub.getMetaRegistry().getMetadataDescriptor(
                component.getComponentManagerPID() );
            if ( metadata.is( type ) )
            {
                if ( component.getRuntimeInfo() != null )
                {
                    hubClient.componentActivated( component );
                }
            }
        }
    }


    public void unregisterHubClient( ComponentListener hubClient, String type )
    {
        if ( type == null )
        {
            Collection<String> registeredTypes = ( Collection ) clientRegistrations.remove( hubClient );
            if ( registeredTypes != null )
            {
                for ( String reg : registeredTypes )
                {
                    typeRegistrations.remove( reg, hubClient );
                }
            }
        }
        else
        {
            typeRegistrations.remove( type, hubClient );
            clientRegistrations.remove( hubClient, type );
        }
    }


    public HubClientManager( ComponentHub hub )
    {
        this.hub = hub;
    }


    public synchronized void fireDCActivated( DirectoryComponent component )
    {
        List<AbstractComponentListener> clients = getRegisteredClients( component );
        for ( ComponentListener client : clients )
        {
            client.componentActivated( component );
        }
    }


    public void fireDCDeactivated( DirectoryComponent component )
    {
        List<AbstractComponentListener> clients = getRegisteredClients( component );
        for ( ComponentListener client : clients )
        {
            client.componentDeactivated( component );
        }
    }


    public void fireDCDeactivating( DirectoryComponent component ) throws HubAbortException
    {
        List<AbstractComponentListener> clients = getRegisteredClients( component );
        for ( ComponentListener client : clients )
        {
            client.componentDeactivating( component );
        }
    }


    public void fireDCReconfigured( DirectoryComponent component )
    {
        List<AbstractComponentListener> clients = getRegisteredClients( component );
        for ( ComponentListener client : clients )
        {
            client.componentReconfigured( component );
        }
    }


    public List<AbstractComponentListener> getRegisteredClients( DirectoryComponent component )
    {
        List<AbstractComponentListener> registeredClients = new ArrayList<AbstractComponentListener>();

        DcMetadataDescriptor metadata = hub.getMetaRegistry()
            .getMetadataDescriptor( component.getComponentManagerPID() );

        Set<String> registeredTypes = typeRegistrations.keySet();
        for ( String type : registeredTypes )
        {
            if ( metadata.is( type ) )
            {
                registeredClients.addAll( ( Collection ) typeRegistrations.get( type ) );
            }
        }

        return registeredClients;
    }
}