org.apache.activemq.plugin.UpdateVirtualDestinationsTask Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of activemq-runtime-config Show documentation
ActiveMQ Runtime Configuration
There is a newer version: 6.1.5
Show newest version
/**
 * 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.activemq.plugin;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import org.apache.activemq.broker.ConnectionContext;
import org.apache.activemq.broker.region.CompositeDestinationInterceptor;
import org.apache.activemq.broker.region.DestinationInterceptor;
import org.apache.activemq.broker.region.RegionBroker;
import org.apache.activemq.broker.region.virtual.VirtualDestination;
import org.apache.activemq.broker.region.virtual.VirtualDestinationInterceptor;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public abstract class UpdateVirtualDestinationsTask implements Runnable {

    public static final Logger LOG = LoggerFactory.getLogger(UpdateVirtualDestinationsTask.class);
    private final AbstractRuntimeConfigurationBroker plugin;

    public UpdateVirtualDestinationsTask(
            AbstractRuntimeConfigurationBroker plugin) {
        super();
        this.plugin = plugin;
    }

    @Override
    public void run() {

        boolean updatedExistingInterceptor = false;
        RegionBroker regionBroker = (RegionBroker) plugin.getBrokerService()
                .getRegionBroker();

        for (DestinationInterceptor destinationInterceptor : plugin
                .getBrokerService().getDestinationInterceptors()) {
            if (destinationInterceptor instanceof VirtualDestinationInterceptor) {
                // update existing interceptor
                final VirtualDestinationInterceptor virtualDestinationInterceptor = (VirtualDestinationInterceptor) destinationInterceptor;

                Set existingVirtualDests = new HashSet<>();
                Collections.addAll(existingVirtualDests, virtualDestinationInterceptor.getVirtualDestinations());

                Set newVirtualDests = new HashSet<>();
                Collections.addAll(newVirtualDests, getVirtualDestinations());

                Set addedVirtualDests = new HashSet<>();
                Set removedVirtualDests = new HashSet<>();
                //detect new virtual destinations
                for (VirtualDestination newVirtualDest : newVirtualDests) {
                    if (!existingVirtualDests.contains(newVirtualDest)) {
                        addedVirtualDests.add(newVirtualDest);
                    }
                }
                //detect removed virtual destinations
                for (VirtualDestination existingVirtualDest : existingVirtualDests) {
                    if (!newVirtualDests.contains(existingVirtualDest)) {
                        removedVirtualDests.add(existingVirtualDest);
                    }
                }

                virtualDestinationInterceptor
                        .setVirtualDestinations(getVirtualDestinations());
                plugin.info("applied updates to: "
                        + virtualDestinationInterceptor);
                updatedExistingInterceptor = true;

                ConnectionContext connectionContext;
                try {
                    connectionContext = plugin.getBrokerService().getAdminConnectionContext();
                    //signal updates
                    if (plugin.getBrokerService().isUseVirtualDestSubs()) {
                        for (VirtualDestination removedVirtualDest : removedVirtualDests) {
                            plugin.virtualDestinationRemoved(connectionContext, removedVirtualDest);
                            LOG.info("Removing virtual destination: {}", removedVirtualDest);
                        }

                        for (VirtualDestination addedVirtualDest : addedVirtualDests) {
                            plugin.virtualDestinationAdded(connectionContext, addedVirtualDest);
                            LOG.info("Adding virtual destination: {}", addedVirtualDest);
                        }
                    }

                } catch (Exception e) {
                    LOG.warn("Could not process virtual destination advisories", e);
                }
            }
        }

        if (!updatedExistingInterceptor) {
            // add
            VirtualDestinationInterceptor virtualDestinationInterceptor = new VirtualDestinationInterceptor();
            virtualDestinationInterceptor.setVirtualDestinations(getVirtualDestinations());

            List interceptorsList = new ArrayList();
            interceptorsList.addAll(Arrays.asList(plugin.getBrokerService()
                    .getDestinationInterceptors()));
            interceptorsList.add(virtualDestinationInterceptor);

            DestinationInterceptor[] destinationInterceptors = interceptorsList
                    .toArray(new DestinationInterceptor[] {});
            plugin.getBrokerService().setDestinationInterceptors(
                    destinationInterceptors);

            ((CompositeDestinationInterceptor) regionBroker
                    .getDestinationInterceptor())
                    .setInterceptors(destinationInterceptors);
            plugin.info("applied new: " + interceptorsList);
        }
        regionBroker.reapplyInterceptor();
    }

    protected abstract VirtualDestination[] getVirtualDestinations();
}