bitronix.tm.resource.ehcache.EhCacheXAResourceProducer Maven / Gradle / Ivy
/*
* Bitronix Transaction Manager
*
* Copyright (c) 2010, Bitronix Software.
*
* This copyrighted material is made available to anyone wishing to use, modify,
* copy, or redistribute it subject to the terms and conditions of the GNU
* Lesser General Public License, as published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License
* for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this distribution; if not, write to:
* Free Software Foundation, Inc.
* 51 Franklin Street, Fifth Floor
* Boston, MA 02110-1301 USA
*/
package bitronix.tm.resource.ehcache;
import bitronix.tm.internal.BitronixRuntimeException;
import bitronix.tm.internal.XAResourceHolderState;
import bitronix.tm.recovery.RecoveryException;
import bitronix.tm.resource.ResourceObjectFactory;
import bitronix.tm.resource.ResourceRegistrar;
import bitronix.tm.resource.common.RecoveryXAResourceHolder;
import bitronix.tm.resource.common.ResourceBean;
import bitronix.tm.resource.common.XAResourceHolder;
import bitronix.tm.resource.common.XAResourceProducer;
import bitronix.tm.resource.common.XAStatefulHolder;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import javax.naming.NamingException;
import javax.naming.Reference;
import javax.naming.StringRefAddr;
import javax.transaction.xa.XAResource;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
import java.util.concurrent.atomic.AtomicInteger;
/**
* EHCache implementation of BTM's XAResourceProducer.
*
* Copyright 2003-2010 Terracotta, Inc.
*
* @author lorban
*/
public final class EhCacheXAResourceProducer extends ResourceBean implements XAResourceProducer {
private static final Logger log = LoggerFactory.getLogger(EhCacheXAResourceProducer.class.getName());
private static final ConcurrentMap producers = new ConcurrentHashMap();
private final ConcurrentMap xaResourceHolders = new ConcurrentHashMap();
private final AtomicInteger xaResourceHolderCounter = new AtomicInteger();
private volatile RecoveryXAResourceHolder recoveryXAResourceHolder;
private EhCacheXAResourceProducer() {
setApplyTransactionTimeout(true);
}
/**
* Register an XAResource of a cache with BTM. The first time a XAResource is registered a new
* EhCacheXAResourceProducer is created to hold it.
* @param uniqueName the uniqueName of this XAResourceProducer, usually the cache's name
* @param xaResource the XAResource to be registered
*/
public static void registerXAResource(String uniqueName, XAResource xaResource) {
EhCacheXAResourceProducer xaResourceProducer = producers.get(uniqueName);
if (xaResourceProducer == null) {
xaResourceProducer = new EhCacheXAResourceProducer();
xaResourceProducer.setUniqueName(uniqueName);
// the initial xaResource must be added before init() can be called
xaResourceProducer.addXAResource(xaResource);
EhCacheXAResourceProducer previous = producers.putIfAbsent(uniqueName, xaResourceProducer);
if (previous == null) {
xaResourceProducer.init();
} else {
previous.addXAResource(xaResource);
}
} else {
xaResourceProducer.addXAResource(xaResource);
}
}
/**
* Unregister an XAResource of a cache from BTM.
* @param uniqueName the uniqueName of this XAResourceProducer, usually the cache's name
* @param xaResource the XAResource to be registered
*/
public static void unregisterXAResource(String uniqueName, XAResource xaResource) {
EhCacheXAResourceProducer xaResourceProducer = producers.get(uniqueName);
if (xaResourceProducer != null) {
boolean found = xaResourceProducer.removeXAResource(xaResource);
if (!found) {
log.error("no XAResource " + xaResource + " found in XAResourceProducer with name " + uniqueName);
}
if (xaResourceProducer.xaResourceHolders.isEmpty()) {
xaResourceProducer.close();
producers.remove(uniqueName);
}
} else {
log.error("no XAResourceProducer registered with name " + uniqueName);
}
}
private void addXAResource(XAResource xaResource) {
EhCacheXAResourceHolder xaResourceHolder = new EhCacheXAResourceHolder(xaResource, this);
int key = xaResourceHolderCounter.incrementAndGet();
xaResourceHolders.put(key, xaResourceHolder);
}
private boolean removeXAResource(XAResource xaResource) {
for (Map.Entry entry : xaResourceHolders.entrySet()) {
Integer key = entry.getKey();
EhCacheXAResourceHolder xaResourceHolder = entry.getValue();
if (xaResourceHolder.getXAResource() == xaResource) {
xaResourceHolders.remove(key);
return true;
}
}
return false;
}
/**
* {@inheritDoc}
*/
public XAResourceHolderState startRecovery() throws RecoveryException {
if (recoveryXAResourceHolder != null) {
throw new RecoveryException("recovery already in progress on " + this);
}
if (xaResourceHolders.isEmpty()) {
throw new RecoveryException("no XAResource registered, recovery cannot be done on " + this);
}
recoveryXAResourceHolder = new RecoveryXAResourceHolder(xaResourceHolders.values().iterator().next());
return new XAResourceHolderState(recoveryXAResourceHolder, this);
}
/**
* {@inheritDoc}
*/
public void endRecovery() throws RecoveryException {
recoveryXAResourceHolder = null;
}
/**
* {@inheritDoc}
*/
public void setFailed(boolean failed) {
// cache cannot fail as it's not connection oriented
}
/**
* {@inheritDoc}
*/
public XAResourceHolder findXAResourceHolder(XAResource xaResource) {
for (EhCacheXAResourceHolder xaResourceHolder : xaResourceHolders.values()) {
if (xaResource == xaResourceHolder.getXAResource()) {
return xaResourceHolder;
}
}
return null;
}
/**
* {@inheritDoc}
*/
public void init() {
try {
ResourceRegistrar.register(this);
} catch (RecoveryException ex) {
throw new BitronixRuntimeException("error recovering " + this, ex);
}
}
/**
* {@inheritDoc}
*/
public void close() {
xaResourceHolders.clear();
xaResourceHolderCounter.set(0);
ResourceRegistrar.unregister(this);
}
/**
* {@inheritDoc}
*/
public XAStatefulHolder createPooledConnection(Object xaFactory, ResourceBean bean) throws Exception {
throw new UnsupportedOperationException("Ehcache is not connection-oriented");
}
/**
* {@inheritDoc}
*/
public Reference getReference() throws NamingException {
return new Reference(EhCacheXAResourceProducer.class.getName(),
new StringRefAddr("uniqueName", getUniqueName()),
ResourceObjectFactory.class.getName(), null);
}
public String toString() {
return "a EhCacheXAResourceProducer with uniqueName " + getUniqueName();
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy