AbstractEndpoint.java
/**
* Copyright (C) 2015 eXo Platform SAS.
*
* 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.exoplatform.commons.api.websocket;
import javax.websocket.CloseReason;
import javax.websocket.Endpoint;
import javax.websocket.EndpointConfig;
import javax.websocket.MessageHandler;
import javax.websocket.Session;
import org.exoplatform.container.PortalContainer;
import org.exoplatform.container.RootContainer;
public abstract class AbstractEndpoint extends Endpoint implements MessageHandler.Whole<String>,
MessageHandler.Partial<String> {
protected volatile Session _session;
private ExtensibleWSFilter filter;
@Override
public void onOpen(Session wsSession, EndpointConfig config) {
_session = wsSession;
wsSession.addMessageHandler(this);
getFilter().onOpen(wsSession, config, this);
}
@Override
public void onClose(Session wsSession, CloseReason closeReason) {
_session = wsSession;
getFilter().onClose(wsSession, closeReason, this);
}
@Override
public void onError(Session wsSession, Throwable failure) {
_session = wsSession;
getFilter().onError(wsSession, failure, this);
}
@Override
public void onMessage(String message) {
getFilter().onMessage(message, this);
}
@Override
public void onMessage(String message, boolean arg1) {
getFilter().onMessage(message, arg1, this);
}
private ExtensibleWSFilter getFilter() {
if (filter == null) {
PortalContainer container = RootContainer.getInstance().getPortalContainer(PortalContainer.getCurrentPortalContainerName());
filter = (ExtensibleWSFilter) container.getComponentInstanceOfType(ExtensibleWSFilter.class);
}
return filter;
}
protected abstract void doOpen(Session wsSession, EndpointConfig config);
protected abstract void doClose(Session wsSession, CloseReason closeReason);
protected abstract void doError(Session wsSession, Throwable failure);
protected abstract void doMessage(Session wsSession, String message);
protected abstract void doMessage(Session wsSession, String message, boolean arg1);
}