/*
* 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 ServiceRegistration tmRegistration;
private ServiceRegistration utRegistration;
private Map<String, ServiceRegistration> dsRegistrations;
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.service.name", 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()));
}
public void stop(BundleContext context) throws Exception {
BitronixTransactionManager tm = (BitronixTransactionManager) 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
{
Pattern uniqueName = Pattern.compile("^\\s*resource\\.[^\\.]*\\.uniqueName\\s*=\\s*([^\\s]+)\\s*$");
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 = uniqueName.matcher(line);
if (matcher.matches()) {
lineNumbers.put(matcher.group(1), ranking);
ranking++;
}
}
reader.close();
return lineNumbers;
}
}