dsw/cxf-dsw/src/main/java/org/apache/cxf/dosgi/dsw/Activator.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.cxf.dosgi.dsw;

import java.util.Dictionary;
import java.util.Hashtable;

import org.apache.cxf.Bus;
import org.apache.cxf.BusFactory;
import org.apache.cxf.dosgi.dsw.api.DistributionProvider;
import org.apache.cxf.dosgi.dsw.decorator.ServiceDecorator;
import org.apache.cxf.dosgi.dsw.decorator.ServiceDecoratorBundleListener;
import org.apache.cxf.dosgi.dsw.decorator.ServiceDecoratorImpl;
import org.apache.cxf.dosgi.dsw.handlers.CXFDistributionProvider;
import org.apache.cxf.dosgi.dsw.handlers.HttpServiceManager;
import org.apache.cxf.dosgi.dsw.qos.DefaultIntentMapFactory;
import org.apache.cxf.dosgi.dsw.qos.IntentManager;
import org.apache.cxf.dosgi.dsw.qos.IntentManagerImpl;
import org.apache.cxf.dosgi.dsw.qos.IntentMap;
import org.apache.cxf.dosgi.dsw.qos.IntentTracker;
import org.osgi.framework.BundleActivator;
import org.osgi.framework.BundleContext;
import org.osgi.framework.BundleListener;
import org.osgi.framework.Constants;
import org.osgi.framework.ServiceRegistration;
import org.osgi.service.cm.ConfigurationException;
import org.osgi.service.cm.ManagedService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

// registered as spring bean -> start / stop called accordingly
public class Activator implements ManagedService, BundleActivator {

    private static final Logger LOG = LoggerFactory.getLogger(Activator.class);
    private static final int DEFAULT_INTENT_TIMEOUT = 30000;
    private static final String CONFIG_SERVICE_PID = "cxf-dsw";
    private ServiceRegistration<?> rsaFactoryReg;
    private ServiceRegistration<?> decoratorReg;
    private IntentTracker intentTracker;
    private HttpServiceManager httpServiceManager;
    private BundleContext bc;
    private BundleListener bundleListener;
    private Dictionary<String, Object> curConfiguration;
    private Bus bus;

    public void start(BundleContext bundlecontext) throws Exception {
        LOG.debug("RemoteServiceAdmin Implementation is starting up");
        this.bc = bundlecontext;
        // Disable the fast infoset as it's not compatible (yet) with OSGi
        System.setProperty("org.apache.cxf.nofastinfoset", "true");
        curConfiguration = getDefaultConfig();
        init(curConfiguration);
        registerManagedService(bc);
    }

    private Dictionary<String, Object> getDefaultConfig() {
        return new Hashtable<String, Object>();
    }

    private synchronized void init(Dictionary<String, Object> config) {
        bus = BusFactory.newInstance().createBus();
        
        String httpBase = (String) config.get(org.apache.cxf.dosgi.dsw.Constants.HTTP_BASE);
        String cxfServletAlias = (String) config.get(org.apache.cxf.dosgi.dsw.Constants.CXF_SERVLET_ALIAS);

        IntentMap intentMap = new IntentMap(new DefaultIntentMapFactory().create());
        intentTracker = new IntentTracker(bc, intentMap);
        intentTracker.open();
        IntentManager intentManager = new IntentManagerImpl(intentMap, DEFAULT_INTENT_TIMEOUT);
        httpServiceManager = new HttpServiceManager(bc, httpBase, cxfServletAlias);
        DistributionProvider cxfProvider
            = new CXFDistributionProvider(bc, intentManager, httpServiceManager);
        Dictionary<String, Object> props = new Hashtable<String, Object>();
        String[] supportedIntents = intentMap.keySet().toArray(new String[] {});
        props.put("remote.intents.supported", supportedIntents);
        props.put("remote.configs.supported", cxfProvider.getSupportedTypes());
        rsaFactoryReg = bc.registerService(DistributionProvider.class.getName(), cxfProvider, props);
        ServiceDecoratorImpl serviceDecorator = new ServiceDecoratorImpl();
        bundleListener = new ServiceDecoratorBundleListener(serviceDecorator);
        bc.addBundleListener(bundleListener);
        decoratorReg = bc.registerService(ServiceDecorator.class.getName(), serviceDecorator, null);
    }

    private synchronized void uninit() {
        if (decoratorReg != null) {
            decoratorReg.unregister();
            decoratorReg = null;
        }
        if (bundleListener != null) {
            bc.removeBundleListener(bundleListener);
            bundleListener = null;
        }
        if (rsaFactoryReg != null) {
            // This also triggers the unimport and unexport of the remote services
            rsaFactoryReg.unregister();
            rsaFactoryReg = null;
        }
        if (httpServiceManager != null) {
            httpServiceManager.close();
            httpServiceManager = null;
        }
        if (intentTracker != null) {
            intentTracker.close();
            intentTracker = null;
        }
    }

    private void registerManagedService(BundleContext bundlecontext) {
        Dictionary<String, String> props = new Hashtable<String, String>();
        props.put(Constants.SERVICE_PID, CONFIG_SERVICE_PID);
        // No need to store the registration. Will be unregistered in stop by framework
        bundlecontext.registerService(ManagedService.class.getName(), this, props);
    }

    public void stop(BundleContext context) throws Exception {
        LOG.debug("RemoteServiceAdmin Implementation is shutting down now");
        uninit();
        shutdownCXFBus();
    }

    /**
     * Causes also the shutdown of the embedded HTTP server
     */
    private void shutdownCXFBus() {
        if (bus != null) {
            LOG.debug("Shutting down the CXF Bus");
            bus.shutdown(true);
        }
    }

    @SuppressWarnings({ "rawtypes", "unchecked" })
    public synchronized void updated(Dictionary config) throws ConfigurationException {
        LOG.debug("RemoteServiceAdmin Implementation configuration is updated with {}", config);
        // config is null if it doesn't exist, is being deleted or has not yet been loaded
        // in which case we run with defaults (just like we do manually when bundle is first started)
        Dictionary<String, Object> configMap = config == null ? getDefaultConfig() : config;
        if (!configMap.equals(curConfiguration)) { // only if something actually changed
            curConfiguration = configMap;
            uninit();
            init(configMap);
        }
    }

}