org.bimserver.servlets.websockets.jsr356.Jsr356Impl Maven / Gradle / Ivy
package org.bimserver.servlets.websockets.jsr356;
/******************************************************************************
* Copyright (C) 2009-2016 BIMserver.org
*
* This program is free software: you can redistribute it and/or modify
* it 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 {@literal }.
*****************************************************************************/
import java.io.IOException;
/******************************************************************************
* Copyright (C) 2009-2015 BIMserver.org
*
* This program is free software: you can redistribute it and/or modify
* it 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 .
*****************************************************************************/
import java.io.StringReader;
import java.nio.ByteBuffer;
import javax.servlet.ServletContext;
import javax.servlet.ServletContextEvent;
import javax.servlet.ServletContextListener;
import javax.servlet.annotation.WebListener;
import javax.websocket.CloseReason;
import javax.websocket.EndpointConfig;
import javax.websocket.OnClose;
import javax.websocket.OnError;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.ServerEndpoint;
import org.bimserver.BimServer;
import org.bimserver.servlets.Streamer;
import org.bimserver.shared.StreamingSocketInterface;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.google.gson.JsonObject;
@WebListener
@ServerEndpoint(value = "/stream")
public class Jsr356Impl implements StreamingSocketInterface, ServletContextListener {
private static final Logger LOGGER = LoggerFactory.getLogger(Jsr356Impl.class);
private Streamer streamer;
private Session websocketSession;
// This is a static reference unfortunately. This is because the only other
// way of getting the ServletContext (and thus BimServer instance)
// to the websocket methods would be via the httpsession, which we don't
// have/use/want. This will probably be fixed in a later
// version of JSR-356 and implementations
private static ServletContext servletContext;
@OnOpen
public void onOpen(Session websocketSession, EndpointConfig config) {
LOGGER.info("WebSocket open");
try {
this.websocketSession = websocketSession;
BimServer bimServer = (BimServer) servletContext.getAttribute("bimserver");
streamer = new Streamer(this, bimServer);
streamer.onOpen();
} catch (Throwable t) {
LOGGER.error("", t);
}
}
@Override
public void contextInitialized(ServletContextEvent servletContextEvent) {
LOGGER.info("WebSocket context initialized");
servletContext = servletContextEvent.getServletContext();
}
@Override
public void contextDestroyed(ServletContextEvent servletContextEvent) {
LOGGER.info("WebSocket context destroyed");
}
@OnMessage
public void onMessage(String message, Session session) {
streamer.onText(new StringReader(message));
}
@OnError
public void onError(Throwable exception, Session session) {
LOGGER.error("", exception);
}
@OnClose
public void onClose(Session session, CloseReason closeReason) {
streamer.onClose();
}
@Override
public void send(JsonObject request) {
synchronized (this) {
try {
websocketSession.getBasicRemote().sendText(request.toString());
} catch (IOException e) {
LOGGER.error("", e);
}
}
}
@Override
public void send(byte[] data, int start, int length) {
synchronized (this) {
try {
websocketSession.getBasicRemote().sendBinary(ByteBuffer.wrap(data, start, length));
} catch (IOException e) {
LOGGER.error("", e);
}
}
}
@Override
public void sendBlocking(byte[] data, int start, int length) {
synchronized (this) {
websocketSession.getAsyncRemote().sendBinary(ByteBuffer.wrap(data, start, length));
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy