/servers/jain-sip-ha/jboss-5/src/main/java/org/mobicents/ha/javax/sip/cache/SIPDialogCacheData.java
Java | 442 lines | 372 code | 19 blank | 51 comment | 89 complexity | 78244d322a7726df0ffa95c27f9b00a4 MD5 | raw file
Possible License(s): LGPL-3.0, GPL-3.0, LGPL-2.1, GPL-2.0, CC-BY-SA-3.0, CC0-1.0, Apache-2.0, BSD-3-Clause
- /*
- * JBoss, Home of Professional Open Source
- * Copyright 2011, Red Hat, Inc. and individual contributors
- * by the @authors tag. See the copyright.txt in the distribution for a
- * full listing of individual contributors.
- *
- * This is free software; you can redistribute it and/or modify it
- * under the terms of the GNU Lesser General Public License as
- * published by the Free Software Foundation; either version 2.1 of
- * the License, or (at your option) any later version.
- *
- * This software 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 software; if not, write to the Free
- * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
- * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
- */
- package org.mobicents.ha.javax.sip.cache;
- import gov.nist.core.CommonLogger;
- import gov.nist.core.StackLogger;
- import gov.nist.javax.sip.SipProviderImpl;
- import gov.nist.javax.sip.message.SIPResponse;
- import gov.nist.javax.sip.stack.AbstractHASipDialog;
- import gov.nist.javax.sip.stack.SIPDialog;
- import java.text.ParseException;
- import java.util.Map;
- import java.util.Map.Entry;
- import javax.sip.PeerUnavailableException;
- import javax.sip.SipFactory;
- import javax.sip.address.Address;
- import javax.sip.header.ContactHeader;
- import javax.transaction.RollbackException;
- import javax.transaction.Status;
- import javax.transaction.TransactionManager;
- import org.jboss.cache.Cache;
- import org.jboss.cache.CacheException;
- import org.jboss.cache.Fqn;
- import org.jboss.cache.Node;
- import org.jboss.cache.config.Configuration;
- import org.mobicents.cache.CacheData;
- import org.mobicents.cache.MobicentsCache;
- import org.mobicents.ha.javax.sip.ClusteredSipStack;
- import org.mobicents.ha.javax.sip.HASipDialog;
- import org.mobicents.ha.javax.sip.HASipDialogFactory;
- /**
- * @author jean.deruelle@gmail.com
- * @author martins
- *
- */
- public class SIPDialogCacheData extends CacheData {
- private static final String APPDATA = "APPDATA";
- private ClusteredSipStack clusteredSipStack;
- private static StackLogger logger = CommonLogger.getLogger(SIPDialogCacheData.class);
- public SIPDialogCacheData(Fqn nodeFqn, MobicentsCache mobicentsCache, ClusteredSipStack clusteredSipStack) {
- super(nodeFqn, mobicentsCache);
- this.clusteredSipStack = clusteredSipStack;
- }
-
- public SIPDialog getSIPDialog(String dialogId) throws SipCacheException {
- HASipDialog haSipDialog = null;
- final Cache jbossCache = getMobicentsCache().getJBossCache();
- Configuration config = jbossCache.getConfiguration();
- final boolean isBuddyReplicationEnabled = config.getBuddyReplicationConfig() != null && config.getBuddyReplicationConfig().isEnabled();
- TransactionManager transactionManager = config.getRuntimeConfig().getTransactionManager();
- boolean doTx = false;
- try {
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logDebug("transaction manager :" + transactionManager);
- }
- if(transactionManager != null && transactionManager.getTransaction() == null) {
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logDebug("transaction manager begin transaction");
- }
- transactionManager.begin();
- doTx = true;
- }
- // Issue 1517 : http://code.google.com/p/mobicents/issues/detail?id=1517
- // Adding code to handle Buddy replication to force data gravitation
- if(isBuddyReplicationEnabled) {
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logDebug("forcing data gravitation since buddy replication is enabled");
- }
- jbossCache.getInvocationContext().getOptionOverrides().setForceDataGravitation(true);
- }
- final Node<String,Object> childNode = getNode().getChild(dialogId);
- if(childNode != null) {
- try {
- final Map<String, Object> dialogMetaData = childNode.getData();
- final Object dialogAppData = childNode.get(APPDATA);
-
- haSipDialog = createDialog(dialogId, dialogMetaData, dialogAppData);
-
- } catch (CacheException e) {
- throw new SipCacheException("A problem occured while retrieving the following dialog " + dialogId + " from the Cache", e);
- }
- }
- } catch (Exception ex) {
- try {
- if(transactionManager != null) {
- // Let's set it no matter what.
- transactionManager.setRollbackOnly();
- }
- } catch (Exception exn) {
- logger.logError("Problem rolling back session mgmt transaction",
- exn);
- }
- } finally {
- if (doTx) {
- try {
- if (transactionManager.getTransaction().getStatus() != Status.STATUS_MARKED_ROLLBACK) {
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logDebug("transaction manager committing transaction");
- }
- transactionManager.commit();
- } else {
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logDebug("endBatch(): rolling back batch");
- }
- transactionManager.rollback();
- }
- } catch (RollbackException re) {
- // Do nothing here since cache may rollback automatically.
- logger.logWarning("endBatch(): rolling back transaction with exception: "
- + re);
- } catch (RuntimeException re) {
- throw re;
- } catch (Exception e) {
- throw new RuntimeException(
- "endTransaction(): Caught Exception ending batch: ",
- e);
- }
- }
- }
- return (SIPDialog) haSipDialog;
- }
-
- /*
- * (non-Javadoc)
- * @see org.mobicents.ha.javax.sip.cache.SipCache#updateDialog(gov.nist.javax.sip.stack.SIPDialog)
- */
- public void updateSIPDialog(SIPDialog sipDialog) throws SipCacheException {
- final String dialogId = sipDialog.getDialogId();
- final Cache jbossCache = getMobicentsCache().getJBossCache();
- Configuration config = jbossCache.getConfiguration();
- final boolean isBuddyReplicationEnabled = config.getBuddyReplicationConfig() != null && config.getBuddyReplicationConfig().isEnabled();
- TransactionManager transactionManager = config.getRuntimeConfig().getTransactionManager();
- boolean doTx = false;
- try {
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logDebug("transaction manager :" + transactionManager);
- }
- if(transactionManager != null && transactionManager.getTransaction() == null) {
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logDebug("transaction manager begin transaction");
- }
- transactionManager.begin();
- doTx = true;
- }
- // Issue 1517 : http://code.google.com/p/mobicents/issues/detail?id=1517
- // Adding code to handle Buddy replication to force data gravitation
- if(isBuddyReplicationEnabled) {
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logDebug("forcing data gravitation since buddy replication is enabled");
- }
- jbossCache.getInvocationContext().getOptionOverrides().setForceDataGravitation(true);
- }
- final Node<String,Object> childNode = getNode().getChild(dialogId);
- if(childNode != null) {
- try {
- final Map<String, Object> dialogMetaData = childNode.getData();
-
- final HASipDialog haSipDialog = (HASipDialog) sipDialog;
- final Object dialogAppData = childNode.get(APPDATA);
- updateDialog(haSipDialog, dialogMetaData, dialogAppData);
- } catch (CacheException e) {
- throw new SipCacheException("A problem occured while retrieving the following dialog " + dialogId + " from the Cache", e);
- }
- }
- } catch (Exception ex) {
- try {
- if(transactionManager != null) {
- // Let's set it no matter what.
- transactionManager.setRollbackOnly();
- }
- } catch (Exception exn) {
- logger.logError("Problem rolling back session mgmt transaction",
- exn);
- }
- } finally {
- if (doTx) {
- try {
- if (transactionManager.getTransaction().getStatus() != Status.STATUS_MARKED_ROLLBACK) {
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logDebug("transaction manager committing transaction");
- }
- transactionManager.commit();
- } else {
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logDebug("endBatch(): rolling back batch");
- }
- transactionManager.rollback();
- }
- } catch (RollbackException re) {
- // Do nothing here since cache may rollback automatically.
- logger.logWarning("endBatch(): rolling back transaction with exception: "
- + re);
- } catch (RuntimeException re) {
- throw re;
- } catch (Exception e) {
- throw new RuntimeException(
- "endTransaction(): Caught Exception ending batch: ",
- e);
- }
- }
- }
- }
-
- public HASipDialog createDialog(String dialogId, Map<String, Object> dialogMetaData, Object dialogAppData) throws SipCacheException {
- HASipDialog haSipDialog = null;
- if(dialogMetaData != null) {
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logDebug("sipStack " + this + " dialog " + dialogId + " is present in the distributed cache, recreating it locally");
- }
- final String lastResponseStringified = (String) dialogMetaData.get(AbstractHASipDialog.LAST_RESPONSE);
- try {
- final SIPResponse lastResponse = (SIPResponse) SipFactory.getInstance().createMessageFactory().createResponse(lastResponseStringified);
- haSipDialog = HASipDialogFactory.createHASipDialog(clusteredSipStack.getReplicationStrategy(), (SipProviderImpl)clusteredSipStack.getSipProviders().next(), lastResponse);
- haSipDialog.setDialogId(dialogId);
- updateDialogMetaData(dialogMetaData, dialogAppData, haSipDialog, true);
- // setLastResponse won't be called on recreation since version will be null on recreation
- haSipDialog.setLastResponse(lastResponse);
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logDebug("HA SIP Dialog " + dialogId + " localTag = " + haSipDialog.getLocalTag());
- logger.logDebug("HA SIP Dialog " + dialogId + " remoteTag = " + haSipDialog.getRemoteTag());
- logger.logDebug("HA SIP Dialog " + dialogId + " localParty = " + haSipDialog.getLocalParty());
- logger.logDebug("HA SIP Dialog " + dialogId + " remoteParty = " + haSipDialog.getRemoteParty());
- }
- } catch (PeerUnavailableException e) {
- throw new SipCacheException("A problem occured while retrieving the following dialog " + dialogId + " from the Cache", e);
- } catch (ParseException e) {
- throw new SipCacheException("A problem occured while retrieving the following dialog " + dialogId + " from the Cache", e);
- }
- }
-
- return haSipDialog;
- }
-
- public void updateDialog(HASipDialog haSipDialog, Map<String, Object> dialogMetaData,
- Object dialogAppData) throws SipCacheException {
- if(dialogMetaData != null) {
- final long currentVersion = haSipDialog.getVersion();
- final Long cacheVersion = ((Long)dialogMetaData.get(AbstractHASipDialog.VERSION));
- if(cacheVersion != null && currentVersion < cacheVersion.longValue()) {
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logDebug("HA SIP Dialog " + haSipDialog + " with dialogId " + haSipDialog.getDialogIdToReplicate() + " is older " + currentVersion + " than the one in the cache " + cacheVersion + " updating it");
- }
- try {
- final String lastResponseStringified = (String) dialogMetaData.get(AbstractHASipDialog.LAST_RESPONSE);
- final SIPResponse lastResponse = (SIPResponse) SipFactory.getInstance().createMessageFactory().createResponse(lastResponseStringified);
- haSipDialog.setLastResponse(lastResponse);
- updateDialogMetaData(dialogMetaData, dialogAppData, haSipDialog, false);
- } catch (PeerUnavailableException e) {
- throw new SipCacheException("A problem occured while retrieving the following dialog " + haSipDialog.getDialogIdToReplicate() + " from the TreeCache", e);
- } catch (ParseException e) {
- throw new SipCacheException("A problem occured while retrieving the following dialog " + haSipDialog.getDialogIdToReplicate() + " from the TreeCache", e);
- }
- } else {
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logDebug("HA SIP Dialog " + haSipDialog + " with dialogId " + haSipDialog.getDialogIdToReplicate() + " is not older " + currentVersion + " than the one in the cache " + cacheVersion + ", not updating it");
- }
- }
- }
- }
- /**
- * Update the haSipDialog passed in param with the dialogMetaData and app meta data
- * @param dialogMetaData
- * @param dialogAppData
- * @param haSipDialog
- * @throws ParseException
- * @throws PeerUnavailableException
- */
- private void updateDialogMetaData(Map<String, Object> dialogMetaData, Object dialogAppData, HASipDialog haSipDialog, boolean recreation) throws ParseException,
- PeerUnavailableException {
- haSipDialog.setMetaDataToReplicate(dialogMetaData, recreation);
- haSipDialog.setApplicationDataToReplicate(dialogAppData);
- final String contactStringified = (String) dialogMetaData.get(AbstractHASipDialog.CONTACT_HEADER);
- if(contactStringified != null) {
- Address contactAddress = SipFactory.getInstance().createAddressFactory().createAddress(contactStringified);
- ContactHeader contactHeader = SipFactory.getInstance().createHeaderFactory().createContactHeader(contactAddress);
- haSipDialog.setContactHeader(contactHeader);
- }
- }
-
- public void putSIPDialog(SIPDialog dialog) throws SipCacheException {
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logStackTrace();
- }
- final HASipDialog haSipDialog = (HASipDialog) dialog;
- final String dialogId = haSipDialog.getDialogIdToReplicate();
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logDebug("put HA SIP Dialog " + dialog + " with dialog " + dialogId);
- }
- final Cache jbossCache = getMobicentsCache().getJBossCache();
- TransactionManager transactionManager = jbossCache.getConfiguration().getRuntimeConfig().getTransactionManager();
- boolean doTx = false;
- try {
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logDebug("transaction manager :" + transactionManager);
- }
- if(transactionManager != null && transactionManager.getTransaction() == null) {
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logDebug("transaction manager begin transaction");
- }
- transactionManager.begin();
- doTx = true;
- }
- final Node childNode = getNode().addChild(Fqn.fromElements(dialogId));
- for (Entry<String, Object> metaData : haSipDialog.getMetaDataToReplicate().entrySet()) {
- childNode.put(metaData.getKey(), metaData.getValue());
- }
- final Object dialogAppData = haSipDialog.getApplicationDataToReplicate();
- if(dialogAppData != null) {
- childNode.put(APPDATA, dialogAppData);
- }
- } catch (Exception ex) {
- try {
- if(transactionManager != null) {
- // Let's set it no matter what.
- transactionManager.setRollbackOnly();
- }
- } catch (Exception exn) {
- logger.logError("Problem rolling back session mgmt transaction",
- exn);
- }
- } finally {
- if (doTx) {
- try {
- if (transactionManager.getTransaction().getStatus() != Status.STATUS_MARKED_ROLLBACK) {
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logDebug("transaction manager committing transaction");
- }
- transactionManager.commit();
- } else {
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logDebug("endBatch(): rolling back batch");
- }
- transactionManager.rollback();
- }
- } catch (RollbackException re) {
- // Do nothing here since cache may rollback automatically.
- logger.logWarning("endBatch(): rolling back transaction with exception: "
- + re);
- } catch (RuntimeException re) {
- throw re;
- } catch (Exception e) {
- throw new RuntimeException(
- "endTransaction(): Caught Exception ending batch: ",
- e);
- }
- }
- }
- }
- public boolean removeSIPDialog(String dialogId) {
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logDebug("remove HA SIP Dialog " + dialogId);
- }
- boolean succeeded = false;
- final Cache jbossCache = getMobicentsCache().getJBossCache();
- TransactionManager transactionManager = jbossCache.getConfiguration().getRuntimeConfig().getTransactionManager();
- boolean doTx = false;
- try {
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logDebug("transaction manager :" + transactionManager);
- }
- if(transactionManager != null && transactionManager.getTransaction() == null) {
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logDebug("transaction manager begin transaction");
- }
- transactionManager.begin();
- doTx = true;
- }
- succeeded = getNode().removeChild(dialogId);
- } catch (Exception ex) {
- try {
- if(transactionManager != null) {
- // Let's set it no matter what.
- transactionManager.setRollbackOnly();
- }
- } catch (Exception exn) {
- logger.logError("Problem rolling back session mgmt transaction",
- exn);
- }
- } finally {
- if (doTx) {
- try {
- if (transactionManager.getTransaction().getStatus() != Status.STATUS_MARKED_ROLLBACK) {
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logDebug("transaction manager committing transaction");
- }
- transactionManager.commit();
- } else {
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logDebug("endBatch(): rolling back batch");
- }
- transactionManager.rollback();
- }
- } catch (RollbackException re) {
- // Do nothing here since cache may rollback automatically.
- logger.logWarning("endBatch(): rolling back transaction with exception: "
- + re);
- } catch (RuntimeException re) {
- throw re;
- } catch (Exception e) {
- throw new RuntimeException(
- "endTransaction(): Caught Exception ending batch: ",
- e);
- }
- }
- }
- return succeeded;
- }
- public void evictSIPDialog(String dialogId) {
- getMobicentsCache().getJBossCache().evict(Fqn.fromElements(getNodeFqn(), Fqn.fromString(dialogId)));
- if(logger.isLoggingEnabled(StackLogger.TRACE_DEBUG)) {
- logger.logDebug("HA SIP Dialog " + dialogId + " evicted");
- }
- }
- }