/* * 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 <http://www.gnu.org/licenses/> * * This file incorporates work covered by the following copyright and * permission notice: * * JBoss, Home of Professional Open Source * Copyright 2007-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.diameter.impl.ha.common.rx; import org.jdiameter.api.app.AppSession; import org.jdiameter.api.rx.ClientRxSession; import org.jdiameter.api.rx.ServerRxSession; import org.jdiameter.common.api.app.IAppSessionDataFactory; import org.jdiameter.common.api.app.rx.IRxSessionData; import org.jdiameter.common.api.data.ISessionDatasource; import org.restcomm.cluster.MobicentsCluster; import org.mobicents.diameter.impl.ha.client.rx.ClientRxSessionDataReplicatedImpl; import org.mobicents.diameter.impl.ha.data.ReplicatedSessionDatasource; import org.mobicents.diameter.impl.ha.server.rx.ServerRxSessionDataReplicatedImpl; /** * * @author <a href="mailto:baranowb@gmail.com"> Bartosz Baranowski </a> * @author <a href="mailto:brainslog@gmail.com"> Alexandre Mendonca </a> */ public class RxReplicatedSessionDataFactory implements IAppSessionDataFactory<IRxSessionData> { private ReplicatedSessionDatasource replicatedSessionDataSource; private MobicentsCluster mobicentsCluster; /** * @param replicatedSessionDataSource */ public RxReplicatedSessionDataFactory(ISessionDatasource replicatedSessionDataSource) { // Is this ok? super(); this.replicatedSessionDataSource = (ReplicatedSessionDatasource) replicatedSessionDataSource; this.mobicentsCluster = this.replicatedSessionDataSource.getMobicentsCluster(); } /* * (non-Javadoc) * * @see org.jdiameter.common.api.app.IAppSessionDataFactory#getAppSessionData(java.lang.Class, java.lang.String) */ @Override public IRxSessionData getAppSessionData(Class<? extends AppSession> clazz, String sessionId) { if (clazz.equals(ClientRxSession.class)) { ClientRxSessionDataReplicatedImpl data = new ClientRxSessionDataReplicatedImpl(sessionId, this.mobicentsCluster, this.replicatedSessionDataSource.getContainer()); return data; } else if (clazz.equals(ServerRxSession.class)) { ServerRxSessionDataReplicatedImpl data = new ServerRxSessionDataReplicatedImpl(sessionId, this.mobicentsCluster); return data; } throw new IllegalArgumentException(); } }