/*
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
*
* Copyright (c) 2013-2014 Oracle and/or its affiliates. All rights reserved.
*
* The contents of this file are subject to the terms of either the GNU
* General Public License Version 2 only ("GPL") or the Common Development
* and Distribution License("CDDL") (collectively, the "License"). You
* may not use this file except in compliance with the License. You can
* obtain a copy of the License at
* http://glassfish.java.net/public/CDDL+GPL_1_1.html
* or packager/legal/LICENSE.txt. See the License for the specific
* language governing permissions and limitations under the License.
*
* When distributing the software, include this License Header Notice in each
* file and include the License file at packager/legal/LICENSE.txt.
*
* GPL Classpath Exception:
* Oracle designates this particular file as subject to the "Classpath"
* exception as provided by Oracle in the GPL Version 2 section of the License
* file that accompanied this code.
*
* Modifications:
* If applicable, add the following below the License Header, with the fields
* enclosed by brackets [] replaced by your own identifying information:
* "Portions Copyright [year] [name of copyright owner]"
*
* Contributor(s):
* If you wish your version of this file to be governed by only the CDDL or
* only the GPL Version 2, indicate your decision by adding "[Contributor]
* elects to include this software in this distribution under the [CDDL or GPL
* Version 2] license." If you don't indicate a single choice of license, a
* recipient has the option to distribute your version of this file under
* either the CDDL, the GPL Version 2 or to extend the choice of license to
* its licensees as provided above. However, if you add GPL Version 2 code
* and therefore, elected the GPL Version 2 license, then the option applies
* only if the new code is made subject to such option by the copyright
* holder.
*/
package org.glassfish.tyrus.container.inmemory;
import java.io.IOException;
import java.net.URI;
import java.nio.ByteBuffer;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import javax.websocket.ClientEndpointConfig;
import javax.websocket.DeploymentException;
import javax.websocket.server.ServerApplicationConfig;
import javax.websocket.server.ServerEndpointConfig;
import org.glassfish.tyrus.core.RequestContext;
import org.glassfish.tyrus.core.TyrusUpgradeResponse;
import org.glassfish.tyrus.core.TyrusWebSocketEngine;
import org.glassfish.tyrus.server.TyrusServerContainer;
import org.glassfish.tyrus.spi.ClientContainer;
import org.glassfish.tyrus.spi.ClientEngine;
import org.glassfish.tyrus.spi.CompletionHandler;
import org.glassfish.tyrus.spi.Connection;
import org.glassfish.tyrus.spi.ReadHandler;
import org.glassfish.tyrus.spi.UpgradeRequest;
import org.glassfish.tyrus.spi.WebSocketEngine;
import org.glassfish.tyrus.spi.Writer;
/**
* In-Memory {@link org.glassfish.tyrus.spi.ClientContainer} implementation.
*
* @author Pavel Bucek (pavel.bucek at oracle.com)
*/
public class InMemoryClientContainer implements ClientContainer {
/**
* Property used to define server config used for in-memory container.
* <p/>
* Value has to be instance of {@link javax.websocket.server.ServerApplicationConfig} and is provided as user
* parameter in {@link javax.websocket.ClientEndpointConfig}.
*/
public static final String SERVER_CONFIG = "org.glassfish.tyrus.container.inmemory.ServerConfig";
@Override
public void openClientSocket(String url, ClientEndpointConfig cec, Map<String, Object> properties, ClientEngine clientEngine) throws DeploymentException, IOException {
final UpgradeRequest upgradeRequest = clientEngine.createUpgradeRequest(URI.create(url), null);
final ServerApplicationConfig serverApplicationConfig = getServerApplicationConfig(cec);
final TyrusServerContainer tyrusServerContainer = new TyrusServerContainer(serverApplicationConfig) {
private final WebSocketEngine webSocketEngine = TyrusWebSocketEngine.builder(getClientManager()).build();
@Override
public void register(Class<?> endpointClass) throws DeploymentException {
webSocketEngine.register(endpointClass, "/");
}
@Override
public void register(ServerEndpointConfig serverEndpointConfig) throws DeploymentException {
webSocketEngine.register(serverEndpointConfig, "/");
}
@Override
public WebSocketEngine getWebSocketEngine() {
return webSocketEngine;
}
};
tyrusServerContainer.doneDeployment();
// placeholder values, not used anywhere in this case.
tyrusServerContainer.start("/inmemory", 0);
final TyrusUpgradeResponse upgradeResponse = new TyrusUpgradeResponse();
StringBuilder sb = new StringBuilder();
sb.append(upgradeRequest.getRequestURI().getPath());
if (upgradeRequest.getRequestURI().getQuery() != null) {
sb.append('?').append(upgradeRequest.getRequestURI().getQuery());
}
if (sb.length() == 0) {
sb.append('/');
}
final RequestContext requestContext = new RequestContext.Builder().requestURI(URI.create(sb.toString())).build();
requestContext.getHeaders().putAll(upgradeRequest.getHeaders());
final WebSocketEngine.UpgradeInfo upgradeInfo = tyrusServerContainer.getWebSocketEngine().upgrade(requestContext, upgradeResponse);
switch (upgradeInfo.getStatus()) {
case HANDSHAKE_FAILED:
throw new DeploymentException("");
case NOT_APPLICABLE:
throw new DeploymentException("");
case SUCCESS:
final InMemoryWriter clientWriter = new InMemoryWriter();
final InMemoryWriter serverWriter = new InMemoryWriter();
final Connection serverConnection = upgradeInfo.createConnection(serverWriter, null);
final ClientEngine.ClientUpgradeInfo clientClientUpgradeInfo = clientEngine.processResponse(upgradeResponse, clientWriter, null);
final Connection clientConnection = clientClientUpgradeInfo.createConnection();
if (clientConnection == null) {
throw new DeploymentException("");
}
serverWriter.setReadHandler(clientConnection.getReadHandler());
clientWriter.setReadHandler(serverConnection.getReadHandler());
}
}
private ServerApplicationConfig getServerApplicationConfig(ClientEndpointConfig clientEndpointConfig) throws DeploymentException {
final Object o = clientEndpointConfig.getUserProperties().get(SERVER_CONFIG);
if (o != null && o instanceof ServerApplicationConfig) {
return (ServerApplicationConfig) o;
}
throw new DeploymentException("ServerApplicationConfig not present.");
}
private static class InMemoryWriter extends Writer {
private final List<ByteBuffer> cache = new ArrayList<ByteBuffer>();
private volatile ReadHandler readHandler = null;
@Override
public void write(ByteBuffer buffer, CompletionHandler<ByteBuffer> completionHandler) {
synchronized (cache) {
if (readHandler == null) {
cache.add(buffer);
} else {
readHandler.handle(buffer);
}
completionHandler.completed(buffer);
}
}
@Override
public void close() throws IOException {
// do nothing.
}
private void setReadHandler(ReadHandler readHandler) {
synchronized (cache) {
for (ByteBuffer buffer : cache) {
readHandler.handle(buffer);
}
this.readHandler = readHandler;
}
}
}
}