Java tutorial
/* * Copyright 2002-2013 the original author or authors. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package org.springframework.web.socket.client.jetty; import java.net.URI; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import org.springframework.context.SmartLifecycle; import org.springframework.http.HttpHeaders; import org.springframework.util.Assert; import org.springframework.web.socket.WebSocketHandler; import org.springframework.web.socket.WebSocketSession; import org.springframework.web.socket.adapter.JettyWebSocketListenerAdapter; import org.springframework.web.socket.adapter.JettyWebSocketSessionAdapter; import org.springframework.web.socket.client.WebSocketClient; import org.springframework.web.socket.client.WebSocketConnectFailureException; import org.springframework.web.util.UriComponents; import org.springframework.web.util.UriComponentsBuilder; /** * @author Rossen Stoyanchev * @since 4.0 */ public class JettyWebSocketClient implements WebSocketClient, SmartLifecycle { private static final Log logger = LogFactory.getLog(JettyWebSocketClient.class); private final org.eclipse.jetty.websocket.client.WebSocketClient client; private boolean autoStartup = true; private int phase = Integer.MAX_VALUE; private final Object lifecycleMonitor = new Object(); public JettyWebSocketClient() { this.client = new org.eclipse.jetty.websocket.client.WebSocketClient(); } // TODO: configure Jetty WebSocketClient properties public void setAutoStartup(boolean autoStartup) { this.autoStartup = autoStartup; } @Override public boolean isAutoStartup() { return this.autoStartup; } public void setPhase(int phase) { this.phase = phase; } @Override public int getPhase() { return this.phase; } @Override public boolean isRunning() { synchronized (this.lifecycleMonitor) { return this.client.isStarted(); } } @Override public void start() { synchronized (this.lifecycleMonitor) { if (!isRunning()) { try { if (logger.isDebugEnabled()) { logger.debug("Starting Jetty WebSocketClient"); } this.client.start(); } catch (Exception e) { throw new IllegalStateException("Failed to start Jetty client", e); } } } } @Override public void stop() { synchronized (this.lifecycleMonitor) { if (isRunning()) { try { if (logger.isDebugEnabled()) { logger.debug("Stopping Jetty WebSocketClient"); } this.client.stop(); } catch (Exception e) { logger.error("Error stopping Jetty WebSocketClient", e); } } } } @Override public void stop(Runnable callback) { this.stop(); callback.run(); } @Override public WebSocketSession doHandshake(WebSocketHandler webSocketHandler, String uriTemplate, Object... uriVariables) throws WebSocketConnectFailureException { UriComponents uriComponents = UriComponentsBuilder.fromUriString(uriTemplate).buildAndExpand(uriVariables) .encode(); return doHandshake(webSocketHandler, null, uriComponents.toUri()); } @Override public WebSocketSession doHandshake(WebSocketHandler webSocketHandler, HttpHeaders headers, URI uri) throws WebSocketConnectFailureException { Assert.notNull(webSocketHandler, "webSocketHandler is required"); Assert.notNull(uri, "uri is required"); if (logger.isDebugEnabled()) { logger.debug("Connecting to " + uri); } // TODO: populate headers JettyWebSocketSessionAdapter session = new JettyWebSocketSessionAdapter(); session.setUri(uri); session.setRemoteHostName(uri.getHost()); JettyWebSocketListenerAdapter listener = new JettyWebSocketListenerAdapter(webSocketHandler, session); try { // TODO: do not block this.client.connect(listener, uri).get(); return session; } catch (Exception e) { throw new WebSocketConnectFailureException("Failed to connect to " + uri, e); } } }