Activator.java

/*
 * Copyright (C) 2006-2013 Bitronix Software (http://www.bitronix.be)
 *
 * 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.
 */
package bitronix.tm.osgi;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileReader;
import java.io.IOException;
import java.net.URI;
import java.util.Dictionary;
import java.util.HashMap;
import java.util.Hashtable;
import java.util.Map;
import java.util.Properties;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import javax.sql.DataSource;
import javax.transaction.TransactionManager;
import javax.transaction.UserTransaction;

import org.osgi.framework.BundleActivator;
import org.osgi.framework.BundleContext;
import org.osgi.framework.ServiceRegistration;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import bitronix.tm.BitronixTransactionManager;
import bitronix.tm.Configuration;
import bitronix.tm.TransactionManagerServices;
import bitronix.tm.resource.common.XAResourceProducer;
import bitronix.tm.utils.InitializationException;

public class Activator implements BundleActivator {

	private final static Logger log = LoggerFactory.getLogger(Activator.class);

    private static final Pattern UNIQUE_NAME = Pattern.compile("^\\s*resource\\.[^\\.]*\\.uniqueName\\s*=\\s*([^\\s]+)\\s*$");

    private ServiceRegistration tmRegistration;
	private ServiceRegistration utRegistration;
	private Map<String, ServiceRegistration> dsRegistrations;

    @Override
	public void start(BundleContext context) throws Exception {
		dsRegistrations = new HashMap<String, ServiceRegistration>();

		String configArea = context.getProperty("osgi.configuration.area").replace(" ", "%20");

        String btmPropFile = System.getProperty("bitronix.tm.configuration", "bitronix-default-config.properties");

        URI btmConfig = URI.create(configArea + btmPropFile);
        File cfgFile = new File(btmConfig);
        if (!cfgFile.exists()) {
        	cfgFile = new File(btmPropFile);
        	if (!cfgFile.exists()) {
        		throw new InitializationException("Configuration file not found: " + btmPropFile);
        	}
        }

        FileReader fileReader = new FileReader(cfgFile);
        Properties btmProperties = new Properties();
        try {
        	btmProperties.load(fileReader);
        }
        finally {
        	fileReader.close();
        }

        System.setProperty("bitronix.tm.configuration", cfgFile.getAbsolutePath());

        File resourceFile = cfgFile;
        String resourcePropFile = btmProperties.getProperty("bitronix.tm.resource.configuration");
        if (resourcePropFile != null) {
        	URI resourceConfig = URI.create(configArea + resourcePropFile);
        	resourceFile = new File(resourceConfig);
            if (!resourceFile.exists()) {
            	resourceFile = new File(resourcePropFile);
            	if (!resourceFile.exists()) {
            		throw new InitializationException("Configuration file not found: " + resourcePropFile);
            	}
            }
            System.setProperty("bitronix.tm.resource.configuration", resourceFile.getAbsolutePath());
        }

        TransactionManager tm = TransactionManagerServices.getTransactionManager();
        tmRegistration = context.registerService(TransactionManager.class.getName(), tm, null);
        utRegistration = context.registerService(UserTransaction.class.getName(), tm, null);

        Map<String, Integer> uniqueNameLineNumbers = rankingOfUniqueNameProperties(resourceFile);
        Map<String, XAResourceProducer> resources = TransactionManagerServices.getResourceLoader().getResources();

        for (Map.Entry<String, XAResourceProducer> me : resources.entrySet()) {
            Integer ranking = uniqueNameLineNumbers.get(me.getKey());
            if (ranking == null) {
                ranking = 1;
            }

            Dictionary<String, Object> props = new Hashtable<String, Object>();
            props.put("service.pid", me.getKey());
            props.put("service.ranking", ranking);
            props.put("osgi.jndi.serviceName", me.getKey());
            ServiceRegistration sr = context.registerService(DataSource.class.getName(), me.getValue(), props);
            dsRegistrations.put(me.getKey(), sr);
        }

        Configuration conf = TransactionManagerServices.getConfiguration();
        log.info(String.format("Started JTA for server ID '%s'.", conf.getServerId()));
	}

    @Override
	public void stop(BundleContext context) throws Exception {
		BitronixTransactionManager tm = TransactionManagerServices.getTransactionManager();
        tm.shutdown();

        tmRegistration.unregister();
        utRegistration.unregister();

        for (ServiceRegistration reg : dsRegistrations.values()) {
            reg.unregister();
        }
        dsRegistrations.clear();

        Configuration conf = TransactionManagerServices.getConfiguration();
        log.info(String.format("Stopped JTA for server ID '%s'.", conf.getServerId()));
	}

	private Map<String, Integer> rankingOfUniqueNameProperties(File file) throws FileNotFoundException, IOException
    {
        Map<String, Integer> lineNumbers = new HashMap<String, Integer>();
        BufferedReader reader = new BufferedReader(new FileReader(file));
        int ranking = 1;
        for (String line = reader.readLine(); line != null; line = reader.readLine()) {
            Matcher matcher = UNIQUE_NAME.matcher(line);
            if (matcher.matches()) {
                lineNumbers.put(matcher.group(1), ranking);
                ranking++;
            }
        }
        reader.close();

        return lineNumbers;
    }
}