org.jdiameter.common.impl.data.LocalDataSource Maven / Gradle / Ivy
The newest version!
/*
* 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>();
private ConcurrentHashMap sessionIdToEntry = new ConcurrentHashMap();
private static final Logger logger = LoggerFactory.getLogger(LocalDataSource.class);
public LocalDataSource() {
appSessionDataFactories.put(ICCASessionData.class, new CCALocalSessionDataFactory());
appSessionDataFactories.put(IRoSessionData.class, new RoLocalSessionDataFactory());
appSessionDataFactories.put(IRfSessionData.class, new RfLocalSessionDataFactory());
appSessionDataFactories.put(IGxSessionData.class, new GxLocalSessionDataFactory());
appSessionDataFactories.put(IAccSessionData.class, new AccLocalSessionDataFactory());
appSessionDataFactories.put(IAuthSessionData.class, new AuthLocalSessionDataFactory());
appSessionDataFactories.put(IShSessionData.class, new ShLocalSessionDataFactory());
appSessionDataFactories.put(ICxDxSessionData.class, new CxDxLocalSessionDataFactory());
appSessionDataFactories.put(IRxSessionData.class, new RxLocalSessionDataFactory());
appSessionDataFactories.put(IS6aSessionData.class, new S6aLocalSessionDataFactory());
appSessionDataFactories.put(IS13SessionData.class, new S13LocalSessionDataFactory());
appSessionDataFactories.put(ISLhSessionData.class, new SLhLocalSessionDataFactory());
appSessionDataFactories.put(ISLgSessionData.class, new SLgLocalSessionDataFactory());
}
public LocalDataSource(IContainer container) {
this();
}
@Override
public boolean exists(String sessionId) {
return this.sessionIdToEntry.containsKey(sessionId);
}
@Override
public void setSessionListener(String sessionId, NetworkReqListener data) {
logger.debug("setSessionListener({}, {})", sessionId, data);
SessionEntry se = sessionIdToEntry.get(sessionId);
if (se != null) {
se.listener = data;
}
else {
throw new IllegalArgumentException("No Session entry for id: " + sessionId);
}
}
@Override
public NetworkReqListener getSessionListener(String sessionId) {
SessionEntry se = sessionIdToEntry.get(sessionId);
logger.debug("getSessionListener({}) => {}", sessionId, se);
return se != null ? se.listener : null;
}
@Override
public NetworkReqListener removeSessionListener(String sessionId) {
SessionEntry se = sessionIdToEntry.get(sessionId);
logger.debug("removeSessionListener({}) => {}", sessionId, se);
if (se != null) {
NetworkReqListener lst = se.listener;
se.listener = null;
return lst;
}
else {
return null;
}
}
@Override
public void addSession(BaseSession session) {
logger.debug("addSession({})", session);
SessionEntry se = null;
String sessionId = session.getSessionId();
//FIXME: check here replicable vs not replicable?
if (this.sessionIdToEntry.containsKey(sessionId)) {
se = this.sessionIdToEntry.get(sessionId);
if ( !(se.session instanceof ISession) || se.session.isReplicable()) { //must be not replicable so we can "overwrite"
throw new IllegalArgumentException("Sessin with id: " + sessionId + ", already exists!");
}
else {
this.sessionIdToEntry.put(sessionId, se);
}
}
else {
se = new SessionEntry();
}
se.session = session;
this.sessionIdToEntry.put(session.getSessionId(), se);
}
@Override
public BaseSession getSession(String sessionId) {
SessionEntry se = sessionIdToEntry.get(sessionId);
logger.debug("getSession({}) => {}", sessionId, se);
return se != null ? se.session : null;
}
@Override
public void removeSession(String sessionId) {
SessionEntry se = this.sessionIdToEntry.remove(sessionId);
logger.debug("removeSession({}) => {}", sessionId, se);
}
/* (non-Javadoc)
* @see org.jdiameter.common.api.data.ISessionDatasource#start()
*/
@Override
public void start() {
// NOP
}
/* (non-Javadoc)
* @see org.jdiameter.common.api.data.ISessionDatasource#stop()
*/
@Override
public void stop() {
// NOP
}
/* (non-Javadoc)
* @see org.jdiameter.common.api.data.ISessionDatasource#isClustered()
*/
@Override
public boolean isClustered() {
return false;
}
@Override
public String toString() {
return "LocalDataSource [sessionIdToEntry=" + sessionIdToEntry + "]";
}
//simple class to reduce collections overhead.
private class SessionEntry {
BaseSession session;
NetworkReqListener listener;
@Override
public String toString() {
return "SessionEntry [session=" + session + ", listener=" + listener + "]";
}
}
/*
* (non-Javadoc)
*
* @see
* org.jdiameter.common.api.data.ISessionDatasource#getDataFactory(java.
* lang.Class)
*/
@Override
public IAppSessionDataFactory getDataFactory(Class x) {
return this.appSessionDataFactories.get(x);
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy