org.mobicents.diameter.impl.ha.data.ReplicatedSessionDatasource Maven / Gradle / Ivy
/*
* TeleStax, Open Source Cloud Communications
* Copyright 2011-2016, TeleStax Inc. and individual contributors
* by the @authors tag.
*
* This program is free software: you can redistribute it and/or modify
* under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation; either version 3 of
* the License, or (at your option) any later version.
*
* 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 Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see , IAppSessionDataFactory> appSessionDataFactories =
new HashMap, IAppSessionDataFactory>();
// Constants
// ----------------------------------------------------------------
public static final String SESSIONS = "/diameter/appsessions";
public static final Fqn SESSIONS_FQN = Fqn.fromString(SESSIONS);
public ReplicatedSessionDatasource(IContainer container) {
this(container, new LocalDataSource(), ReplicatedSessionDatasource.class.getClassLoader().getResource(CLUSTER_DS_DEFAULT_FILE) == null ?
"config/" + CLUSTER_DS_DEFAULT_FILE : CLUSTER_DS_DEFAULT_FILE);
}
public ReplicatedSessionDatasource(IContainer container, ISessionDatasource localDataSource, String cacheConfigFilename) {
super();
this.localDataSource = localDataSource;
MobicentsCache mcCache = new MobicentsCache(cacheConfigFilename);
TransactionManager txMgr = null;
try {
Class txMgrClass = Class.forName(mcCache.getJBossCache().getConfiguration().getTransactionManagerLookupClass());
Object txMgrLookup = txMgrClass.getConstructor(new Class[]{}).newInstance(new Object[]{});
txMgr = (TransactionManager) txMgrClass.getMethod("getTransactionManager", new Class[]{}).invoke(txMgrLookup, new Object[]{});
}
catch (Exception e) {
logger.debug("Could not fetch TxMgr. Not using one.", e);
// let's not have Tx Manager than...
}
this.mobicentsCluster = new DefaultMobicentsCluster(mcCache, txMgr, new DefaultClusterElector());
this.mobicentsCluster.addDataRemovalListener(this); // register, so we know WHEN some other node removes session.
this.mobicentsCluster.startCluster();
this.container = container;
// this is coded, its tied to specific impl of SessionDatasource
appSessionDataFactories.put(IAuthSessionData.class, new AuthReplicatedSessionDataFactory(this));
appSessionDataFactories.put(IAccSessionData.class, new AccReplicatedSessionDataFactory(this));
appSessionDataFactories.put(ICCASessionData.class, new CCAReplicatedSessionDataFactory(this));
appSessionDataFactories.put(IRoSessionData.class, new RoReplicatedSessionDataFactory(this));
appSessionDataFactories.put(IRfSessionData.class, new RfReplicatedSessionDataFactory(this));
appSessionDataFactories.put(IShSessionData.class, new ShReplicatedSessionDataFactory(this));
appSessionDataFactories.put(ICxDxSessionData.class, new CxDxReplicatedSessionDataFactory(this));
appSessionDataFactories.put(IGxSessionData.class, new GxReplicatedSessionDataFactory(this));
appSessionDataFactories.put(IRxSessionData.class, new RxReplicatedSessionDataFactory(this));
appSessionDataFactories.put(IS13SessionData.class, new S13ReplicatedSessionDataFactory(this));
}
@Override
public boolean exists(String sessionId) {
return this.localDataSource.exists(sessionId) ? true : this.existReplicated(sessionId);
}
/*
* (non-Javadoc)
*
* @see org.jdiameter.common.api.ha.ISessionDatasource#addSession(org.jdiameter .api.BaseSession)
*/
@Override
public void addSession(BaseSession session) {
// Simple as is, if its replicated, it will be already there :)
this.localDataSource.addSession(session);
}
/*
* (non-Javadoc)
*
* @see org.jdiameter.common.api.ha.ISessionDatasource#getSession(java.lang.String )
*/
@Override
public BaseSession getSession(String sessionId) {
if (this.localDataSource.exists(sessionId)) {
return this.localDataSource.getSession(sessionId);
}
else if (!this.localMode && this.existReplicated(sessionId)) {
this.makeLocal(sessionId);
return this.localDataSource.getSession(sessionId);
}
return null;
}
/*
* (non-Javadoc)
*
* @see org.jdiameter.common.api.ha.ISessionDatasource#getSessionListener(java .lang.String)
*/
@Override
public NetworkReqListener getSessionListener(String sessionId) {
if (this.localDataSource.exists(sessionId)) {
return this.localDataSource.getSessionListener(sessionId);
}
else if (!this.localMode && this.existReplicated(sessionId)) {
this.makeLocal(sessionId);
return this.localDataSource.getSessionListener(sessionId);
}
return null;
}
/*
* (non-Javadoc)
*
* @see org.jdiameter.common.api.ha.ISessionDatasource#removeSession(java.lang .String)
*/
@Override
public void removeSession(String sessionId) {
logger.debug("removeSession({}) in Local DataSource", sessionId);
if (this.localDataSource.exists(sessionId)) {
this.localDataSource.removeSession(sessionId);
}
else if (!this.localMode && this.existReplicated(sessionId)) {
// FIXME: remove node.
}
}
/*
* (non-Javadoc)
*
* @see org.jdiameter.common.api.ha.ISessionDatasource#removeSessionListener( java.lang.String)
*/
@Override
public NetworkReqListener removeSessionListener(String sessionId) {
if (this.localDataSource.exists(sessionId)) {
return this.localDataSource.removeSessionListener(sessionId);
}
else if (!this.localMode && this.existReplicated(sessionId)) {
// does not make much sense ;[
this.makeLocal(sessionId);
return this.localDataSource.removeSessionListener(sessionId);
}
return null;
}
/*
* (non-Javadoc)
*
* @see org.jdiameter.common.api.ha.ISessionDatasource#setSessionListener(java .lang.String, org.jdiameter.api.NetworkReqListener)
*/
@Override
public void setSessionListener(String sessionId, NetworkReqListener data) {
if (this.localDataSource.exists(sessionId)) {
this.localDataSource.setSessionListener(sessionId, data);
}
else if (!this.localMode && this.existReplicated(sessionId)) {
// does not make much sense ;[
this.makeLocal(sessionId);
this.localDataSource.setSessionListener(sessionId, data);
}
}
@Override
public void start() {
mobicentsCluster.getMobicentsCache().startCache();
localMode = mobicentsCluster.getMobicentsCache().isLocalMode();
}
@Override
public void stop() {
mobicentsCluster.getMobicentsCache().stopCache();
}
/*
* (non-Javadoc)
*
* @see org.jdiameter.common.api.data.ISessionDatasource#isClustered()
*/
@Override
public boolean isClustered() {
return !localMode;
}
/*
* (non-Javadoc)
*
* @see org.jdiameter.common.api.data.ISessionDatasource#getDataFactory(java. lang.Class)
*/
@Override
public IAppSessionDataFactory getDataFactory(Class x) {
return this.appSessionDataFactories.get(x);
}
// remove lst;
public MobicentsCluster getMobicentsCluster() {
return this.mobicentsCluster;
}
@Override
public void dataRemoved(Fqn sessionFqn) {
String sessionId = (String) sessionFqn.getLastElement();
this.localDataSource.removeSession(sessionId);
}
@Override
public Fqn getBaseFqn() {
return SESSIONS_FQN;
}
/**
* @param sessionId
* @return
*/
private boolean existReplicated(String sessionId) {
if (!this.localMode && this.mobicentsCluster.getMobicentsCache().getJBossCache().getNode(Fqn.fromRelativeElements(SESSIONS_FQN, sessionId)) != null) {
return true;
}
return false;
}
/**
* @param sessionId
*/
private void makeLocal(String sessionId) {
try {
// this is APP session, always
Class appSessionInterfaceClass =
AppSessionDataReplicatedImpl.getAppSessionIface(this.mobicentsCluster.getMobicentsCache(), sessionId);
// get factory;
// FIXME: make it a field?
IAppSessionFactory fct = ((ISessionFactory) this.container.getSessionFactory()).getAppSessionFactory(appSessionInterfaceClass);
if (fct == null) {
logger.warn("Session with id:{}, is in replicated data source, but no Application Session Factory for:{}.", sessionId, appSessionInterfaceClass);
return;
}
else {
BaseSession session = fct.getSession(sessionId, appSessionInterfaceClass);
this.localDataSource.addSession(session);
// hmmm
this.localDataSource.setSessionListener(sessionId, (NetworkReqListener) session);
return;
}
}
catch (IllegalDiameterStateException e) {
if (logger.isErrorEnabled()) {
logger.error("Failed to obtain factory from stack...");
}
}
}
// ------- local getter
public IContainer getContainer() {
return this.container;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy