/servers/diameter/core/jdiameter/impl/src/main/java/org/jdiameter/common/impl/app/s6a/S6aSessionFactoryImpl.java

/*

 * 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.jdiameter.common.impl.app.s6a;



import org.jdiameter.api.Answer;

import org.jdiameter.api.ApplicationId;

import org.jdiameter.api.IllegalDiameterStateException;

import org.jdiameter.api.InternalException;

import org.jdiameter.api.OverloadException;

import org.jdiameter.api.Request;

import org.jdiameter.api.RouteException;

import org.jdiameter.api.SessionFactory;

import org.jdiameter.api.app.AppAnswerEvent;

import org.jdiameter.api.app.AppRequestEvent;

import org.jdiameter.api.app.AppSession;

import org.jdiameter.api.app.StateChangeListener;

import org.jdiameter.api.s6a.ClientS6aSession;

import org.jdiameter.api.s6a.ClientS6aSessionListener;

import org.jdiameter.api.s6a.ServerS6aSession;

import org.jdiameter.api.s6a.ServerS6aSessionListener;

import org.jdiameter.api.s6a.events.JAuthenticationInformationAnswer;

import org.jdiameter.api.s6a.events.JAuthenticationInformationRequest;

import org.jdiameter.api.s6a.events.JCancelLocationAnswer;

import org.jdiameter.api.s6a.events.JCancelLocationRequest;

import org.jdiameter.api.s6a.events.JDeleteSubscriberDataAnswer;

import org.jdiameter.api.s6a.events.JDeleteSubscriberDataRequest;

import org.jdiameter.api.s6a.events.JInsertSubscriberDataAnswer;

import org.jdiameter.api.s6a.events.JInsertSubscriberDataRequest;

import org.jdiameter.api.s6a.events.JNotifyAnswer;

import org.jdiameter.api.s6a.events.JNotifyRequest;

import org.jdiameter.api.s6a.events.JPurgeUEAnswer;

import org.jdiameter.api.s6a.events.JPurgeUERequest;

import org.jdiameter.api.s6a.events.JResetAnswer;

import org.jdiameter.api.s6a.events.JResetRequest;

import org.jdiameter.api.s6a.events.JUpdateLocationAnswer;

import org.jdiameter.api.s6a.events.JUpdateLocationRequest;

import org.jdiameter.client.api.ISessionFactory;

import org.jdiameter.client.impl.app.s6a.IClientS6aSessionData;

import org.jdiameter.client.impl.app.s6a.S6aClientSessionImpl;

import org.jdiameter.common.api.app.IAppSessionDataFactory;

import org.jdiameter.common.api.app.s6a.IS6aMessageFactory;

import org.jdiameter.common.api.app.s6a.IS6aSessionData;

import org.jdiameter.common.api.app.s6a.IS6aSessionFactory;

import org.jdiameter.common.api.data.ISessionDatasource;

import org.jdiameter.server.impl.app.s6a.IServerS6aSessionData;

import org.jdiameter.server.impl.app.s6a.S6aServerSessionImpl;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;



/**

 * 

 * @author <a href="mailto:brainslog@gmail.com"> Alexandre Mendonca </a>

 * @author <a href="mailto:richard.good@smilecoms.com"> Richard Good </a>

 * @author <a href="mailto:paul.carter-brown@smilecoms.com"> Paul Carter-Brown </a>

 */

public class S6aSessionFactoryImpl implements IS6aSessionFactory, ServerS6aSessionListener, ClientS6aSessionListener, IS6aMessageFactory, StateChangeListener<AppSession> {



  private static final Logger logger = LoggerFactory.getLogger(S6aSessionFactoryImpl.class);



  protected ISessionFactory sessionFactory;



  protected ServerS6aSessionListener serverSessionListener;

  protected ClientS6aSessionListener clientSessionListener;



  protected IS6aMessageFactory messageFactory;

  protected StateChangeListener<AppSession> stateListener;

  protected ISessionDatasource iss;

  protected IAppSessionDataFactory<IS6aSessionData> sessionDataFactory;



  public S6aSessionFactoryImpl(SessionFactory sessionFactory) {

    super();



    this.sessionFactory = (ISessionFactory) sessionFactory;

    this.iss = this.sessionFactory.getContainer().getAssemblerFacility().getComponentInstance(ISessionDatasource.class);

    this.sessionDataFactory = (IAppSessionDataFactory<IS6aSessionData>) this.iss.getDataFactory(IS6aSessionData.class);

  }



  /**

   * @return the serverSessionListener

   */

  public ServerS6aSessionListener getServerSessionListener() {

    return serverSessionListener != null ? serverSessionListener : this;

  }



  /**

   * @param serverSessionListener

   *            the serverSessionListener to set

   */

  public void setServerSessionListener(ServerS6aSessionListener serverSessionListener) {

    this.serverSessionListener = serverSessionListener;

  }



  /**

   * @return the serverSessionListener

   */

  public ClientS6aSessionListener getClientSessionListener() {

    return clientSessionListener != null ? clientSessionListener : this;

  }



  /**

   * @param serverSessionListener

   *            the serverSessionListener to set

   */

  public void setClientSessionListener(ClientS6aSessionListener clientSessionListener) {

    this.clientSessionListener = clientSessionListener;

  }



  /**

   * @return the messageFactory

   */

  public IS6aMessageFactory getMessageFactory() {

    return messageFactory != null ? messageFactory: this;

  }



  /**

   * @param messageFactory

   *            the messageFactory to set

   */

  public void setMessageFactory(IS6aMessageFactory messageFactory) {

    this.messageFactory = messageFactory;

  }



  /**

   * @return the stateListener

   */

  public StateChangeListener<AppSession> getStateListener() {

    return stateListener != null ? stateListener : this;

  }



  /**

   * @param stateListener

   *            the stateListener to set

   */

  public void setStateListener(StateChangeListener<AppSession> stateListener) {

    this.stateListener = stateListener;

  }



  public AppSession getSession(String sessionId, Class<? extends AppSession> aClass) {

    if (sessionId == null) {

      throw new IllegalArgumentException("SessionId must not be null");

    }

    if (!this.iss.exists(sessionId)) {

      return null;

    }

    AppSession appSession = null;

    try {

      if (aClass == ServerS6aSession.class) {

        IServerS6aSessionData sessionData = (IServerS6aSessionData) this.sessionDataFactory.getAppSessionData(ServerS6aSession.class, sessionId);

        S6aServerSessionImpl serverSession = new S6aServerSessionImpl(sessionData, getMessageFactory(), sessionFactory, this.getServerSessionListener());

        serverSession.getSessions().get(0).setRequestListener(serverSession);

        appSession = serverSession;

      }

      else if (aClass == ClientS6aSession.class) {

        IClientS6aSessionData sessionData = (IClientS6aSessionData) this.sessionDataFactory.getAppSessionData(ClientS6aSession.class, sessionId);

        S6aClientSessionImpl clientSession = new S6aClientSessionImpl(sessionData, getMessageFactory(), sessionFactory, this.getClientSessionListener());

        clientSession.getSessions().get(0).setRequestListener(clientSession);

        appSession = clientSession;

      }

      else {

        throw new IllegalArgumentException("Wrong session class: " + aClass + ". Supported[" + ServerS6aSession.class + "]");

      }

    }

    catch (Exception e) {

      logger.error("Failure to obtain new S6a Session.", e);

    }



    return appSession;

  }



  /*

   * (non-Javadoc)

   * @see org.jdiameter.common.api.app.IAppSessionFactory#getNewSession(java.lang.String, java.lang.Class, org.jdiameter.api.ApplicationId, java.lang.Object[])

   */

  public AppSession getNewSession(String sessionId, Class<? extends AppSession> aClass, ApplicationId applicationId, Object[] args) {

    AppSession appSession = null;



    if (aClass == ServerS6aSession.class) {

      if (sessionId == null) {

        if (args != null && args.length > 0 && args[0] instanceof Request) {

          Request request = (Request) args[0];

          sessionId = request.getSessionId();

        }

        else {

          sessionId = this.sessionFactory.getSessionId();

        }

      }

      IServerS6aSessionData sessionData = (IServerS6aSessionData) this.sessionDataFactory.getAppSessionData(ServerS6aSession.class, sessionId);

      S6aServerSessionImpl serverSession = new S6aServerSessionImpl(sessionData, getMessageFactory(), sessionFactory, this.getServerSessionListener());



      iss.addSession(serverSession);

      serverSession.getSessions().get(0).setRequestListener(serverSession);

      appSession = serverSession;

    }

    else if (aClass == ClientS6aSession.class) {

      if (sessionId == null) {

        if (args != null && args.length > 0 && args[0] instanceof Request) {

          Request request = (Request) args[0];

          sessionId = request.getSessionId();

        }

        else {

          sessionId = this.sessionFactory.getSessionId();

        }

      }

      IClientS6aSessionData sessionData = (IClientS6aSessionData) this.sessionDataFactory.getAppSessionData(ClientS6aSession.class, sessionId);

      S6aClientSessionImpl clientSession = new S6aClientSessionImpl(sessionData, getMessageFactory(), sessionFactory, this.getClientSessionListener());



      iss.addSession(clientSession);

      clientSession.getSessions().get(0).setRequestListener(clientSession);

      appSession = clientSession;

    }

    else {

      throw new IllegalArgumentException("Wrong session class: " + aClass + ". Supported[" + ServerS6aSession.class + "]");

    }



    return appSession;

  }



  /*

   * (non-Javadoc)

   * @see org.jdiameter.api.app.StateChangeListener#stateChanged(java.lang.Enum, java.lang.Enum)

   */

  public void stateChanged(Enum oldState, Enum newState) {

    logger.info("Diameter S6a Session Factory :: stateChanged :: oldState[{}], newState[{}]", oldState, newState);

  }



  public long getApplicationId() {

    return 16777251;

  }



  /*

   * (non-Javadoc)

   * @see org.jdiameter.api.app.StateChangeListener#stateChanged(java.lang.Object, java.lang.Enum, java.lang.Enum)

   */

  public void stateChanged(AppSession source, Enum oldState, Enum newState) {

    logger.info("Diameter S6a Session Factory :: stateChanged :: Session, [{}], oldState[{}], newState[{}]", new Object[]{source, oldState, newState});

  }



  public JAuthenticationInformationAnswer createAuthenticationInformationAnswer(Answer answer) {

    return new JAuthenticationInformationAnswerImpl(answer);

  }



  public JAuthenticationInformationRequest createAuthenticationInformationRequest(Request request) {

    return new JAuthenticationInformationRequestImpl(request);

  }



  public JUpdateLocationRequest createUpdateLocationRequest(Request request) {

    return new JUpdateLocationRequestImpl(request);

  }



  public JUpdateLocationAnswer createUpdateLocationAnswer(Answer answer) {

    return new JUpdateLocationAnswerImpl(answer);

  }



  public JPurgeUERequest createPurgeUERequest(Request request) {

    return new JPurgeUERequestImpl(request);

  }



  public JPurgeUEAnswer createPurgeUEAnswer(Answer answer) {

    return new JPurgeUEAnswerImpl(answer);

  }



  public JCancelLocationRequest createCancelLocationRequest(Request request) {

    return new JCancelLocationRequestImpl(request);

  }



  public JCancelLocationAnswer createCancelLocationAnswer(Answer answer) {

    return new JCancelLocationAnswerImpl(answer);

  }



  public JInsertSubscriberDataRequest createInsertSubscriberDataRequest(Request request) {

    return new JInsertSubscriberDataRequestImpl(request);

  }



  public JInsertSubscriberDataAnswer createInsertSubscriberDataAnswer(Answer answer) {

    return new JInsertSubscriberDataAnswerImpl(answer);

  }



  public JDeleteSubscriberDataRequest createDeleteSubscriberDataRequest(Request request) {

    return new JDeleteSubscriberDataRequestImpl(request);

  }



  public JDeleteSubscriberDataAnswer createDeleteSubscriberDataAnswer(Answer answer) {

    return new JDeleteSubscriberDataAnswerImpl(answer);

  }



  public JResetRequest createResetRequest(Request request) {

    return new JResetRequestImpl(request);

  }



  public JResetAnswer createResetAnswer(Answer answer) {

    return new JResetAnswerImpl(answer);

  }



  public JNotifyRequest createNotifyRequest(Request request) {

    return new JNotifyRequestImpl(request);

  }



  public JNotifyAnswer createNotifyAnswer(Answer answer) {

    return new JNotifyAnswerImpl(answer);

  }



  public void doAuthenticationInformationRequestEvent(ServerS6aSession appSession, JAuthenticationInformationRequest request) throws InternalException, IllegalDiameterStateException, RouteException, OverloadException {

    logger.info("Diameter S6a Session Factory :: doAuthenticationInformationRequest :: appSession[{}], Request[{}]", appSession, request);

  }



  public void doPurgeUERequestEvent(ServerS6aSession appSession, JPurgeUERequest request) throws InternalException, IllegalDiameterStateException, RouteException, OverloadException {

    logger.info("Diameter S6a Session Factory :: doPurgeUERequest :: appSession[{}], Request[{}]", appSession, request);

  }



  public void doUpdateLocationRequestEvent(ServerS6aSession appSession, JUpdateLocationRequest request) throws InternalException, IllegalDiameterStateException, RouteException, OverloadException {

    logger.info("Diameter S6a Session Factory :: doUpdateLocationRequest :: appSession[{}], Request[{}]", appSession, request);

  }



  public void doNotifyRequestEvent(ServerS6aSession appSession, JNotifyRequest request) throws InternalException, IllegalDiameterStateException, RouteException, OverloadException {

    logger.info("Diameter S6a Session Factory :: doNotifyRequest :: appSession[{}], Request[{}]", appSession, request);

  }



  public void doCancelLocationAnswerEvent(ServerS6aSession appSession, JCancelLocationRequest request, JCancelLocationAnswer answer) throws InternalException, IllegalDiameterStateException, RouteException, OverloadException {

    logger.info("Diameter S6a Session Factory :: doCancelLocationAnswer :: appSession[{}], Request[{}], Answer[{}]", new Object[]{appSession, request, answer});

  }



  public void doInsertSubscriberDataAnswerEvent(ServerS6aSession appSession, JInsertSubscriberDataRequest request, JInsertSubscriberDataAnswer answer) throws InternalException, IllegalDiameterStateException, RouteException, OverloadException {

    logger.info("Diameter S6a Session Factory :: doInsertSubscriberDataAnswer :: appSession[{}], Request[{}], Answer[{}]", new Object[]{appSession, request, answer});

  }



  public void doDeleteSubscriberDataAnswerEvent(ServerS6aSession appSession, JDeleteSubscriberDataRequest request, JDeleteSubscriberDataAnswer answer) throws InternalException, IllegalDiameterStateException, RouteException, OverloadException {

    logger.info("Diameter S6a Session Factory :: doDeleteSubscriberDataAnswer :: appSession[{}], Request[{}], Answer[{}]", new Object[]{appSession, request, answer});

  }



  public void doResetAnswerEvent(ServerS6aSession appSession, JResetRequest request, JResetAnswer answer) throws InternalException, IllegalDiameterStateException, RouteException, OverloadException {

    logger.info("Diameter S6a Session Factory :: doResetAnswer :: appSession[{}], Request[{}], Answer[{}]", new Object[]{appSession, request, answer});

  }



  public void doOtherEvent(AppSession appSession, AppRequestEvent request, AppAnswerEvent answer) throws InternalException, IllegalDiameterStateException, RouteException, OverloadException {

    logger.info("Diameter S6a Session Factory :: doOtherEvent :: appSession[{}], Request[{}], Answer[{}]", new Object[]{appSession, request, answer});

  }



  public void doCancelLocationRequestEvent(ClientS6aSession appSession, JCancelLocationRequest request) throws InternalException, IllegalDiameterStateException, RouteException, OverloadException {

    logger.info("Diameter S6a Session Factory :: doCancelLocationRequest :: appSession[{}], Request[{}]", new Object[]{appSession, request});

  }



  public void doInsertSubscriberDataRequestEvent(ClientS6aSession appSession, JInsertSubscriberDataRequest request) throws InternalException, IllegalDiameterStateException, RouteException, OverloadException {

    logger.info("Diameter S6a Session Factory :: doInsertSubscriberDataRequest :: appSession[{}], Request[{}]", new Object[]{appSession, request});

  }



  public void doDeleteSubscriberDataRequestEvent(ClientS6aSession appSession, JDeleteSubscriberDataRequest request) throws InternalException, IllegalDiameterStateException, RouteException, OverloadException {

    logger.info("Diameter S6a Session Factory :: doDeleteSubscriberDataRequest :: appSession[{}], Request[{}]", new Object[]{appSession, request});

  }



  public void doResetRequestEvent(ClientS6aSession appSession, JResetRequest request) throws InternalException, IllegalDiameterStateException, RouteException, OverloadException {

    logger.info("Diameter S6a Session Factory :: doResetRequest :: appSession[{}], Request[{}]", new Object[]{appSession, request});

  }



  public void doAuthenticationInformationAnswerEvent(ClientS6aSession appSession, JAuthenticationInformationRequest request, JAuthenticationInformationAnswer answer) throws InternalException, IllegalDiameterStateException, RouteException, OverloadException {

    logger.info("Diameter S6a Session Factory :: doAuthenticationInformationAnswer :: appSession[{}], Request[{}], Answer[{}]", new Object[]{appSession, request, answer});

  }



  public void doPurgeUEAnswerEvent(ClientS6aSession appSession, JPurgeUERequest request, JPurgeUEAnswer answer) throws InternalException, IllegalDiameterStateException, RouteException, OverloadException {

    logger.info("Diameter S6a Session Factory :: doPurgeUEAnswer :: appSession[{}], Request[{}], Answer[{}]", new Object[]{appSession, request, answer});

  }



  public void doUpdateLocationAnswerEvent(ClientS6aSession appSession, JUpdateLocationRequest request, JUpdateLocationAnswer answer) throws InternalException, IllegalDiameterStateException, RouteException, OverloadException {

    logger.info("Diameter S6a Session Factory :: doUpdateLocationAnswer :: appSession[{}], Request[{}], Answer[{}]", new Object[]{appSession, request, answer});

  }



  public void doNotifyAnswerEvent(ClientS6aSession appSession, JNotifyRequest request, JNotifyAnswer answer) throws InternalException, IllegalDiameterStateException, RouteException, OverloadException {

    logger.info("Diameter S6a Session Factory :: doNotifyAnswer :: appSession[{}], Request[{}], Answer[{}]", new Object[]{appSession, request, answer});

  }



}