| package fi.iki.elonen.samples.echo; |
| |
| /* |
| * #%L |
| * NanoHttpd-Websocket |
| * %% |
| * Copyright (C) 2012 - 2015 nanohttpd |
| * %% |
| * Redistribution and use in source and binary forms, with or without modification, |
| * are permitted provided that the following conditions are met: |
| * |
| * 1. Redistributions of source code must retain the above copyright notice, this |
| * list of conditions and the following disclaimer. |
| * |
| * 2. Redistributions in binary form must reproduce the above copyright notice, |
| * this list of conditions and the following disclaimer in the documentation |
| * and/or other materials provided with the distribution. |
| * |
| * 3. Neither the name of the nanohttpd nor the names of its contributors |
| * may be used to endorse or promote products derived from this software without |
| * specific prior written permission. |
| * |
| * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND |
| * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED |
| * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. |
| * IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, |
| * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, |
| * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF |
| * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE |
| * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED |
| * OF THE POSSIBILITY OF SUCH DAMAGE. |
| * #L% |
| */ |
| |
| import java.util.ArrayList; |
| import java.util.List; |
| import java.util.concurrent.CountDownLatch; |
| import java.util.concurrent.Future; |
| import java.util.concurrent.TimeUnit; |
| |
| import org.eclipse.jetty.websocket.api.Session; |
| import org.eclipse.jetty.websocket.api.StatusCode; |
| import org.eclipse.jetty.websocket.api.annotations.OnWebSocketClose; |
| import org.eclipse.jetty.websocket.api.annotations.OnWebSocketConnect; |
| import org.eclipse.jetty.websocket.api.annotations.OnWebSocketMessage; |
| import org.eclipse.jetty.websocket.api.annotations.WebSocket; |
| |
| /** |
| * Basic Echo Client Socket |
| */ |
| @WebSocket(maxTextMessageSize = 64 * 1024) |
| public class SimpleEchoSocket { |
| |
| private final List<String> receivedMessages = new ArrayList<String>(); |
| |
| private final List<String> toSendMessages = new ArrayList<String>(); |
| |
| private final CountDownLatch closeLatch; |
| |
| public SimpleEchoSocket() { |
| this.closeLatch = new CountDownLatch(1); |
| } |
| |
| public boolean awaitClose(int duration, TimeUnit unit) throws InterruptedException { |
| return this.closeLatch.await(duration, unit); |
| } |
| |
| public List<String> getReceivedMessages() { |
| return this.receivedMessages; |
| } |
| |
| public List<String> getToSendMessages() { |
| return this.toSendMessages; |
| } |
| |
| @OnWebSocketClose |
| public void onClose(int statusCode, String reason) { |
| System.out.printf("Connection closed: %d - %s%n", statusCode, reason); |
| this.closeLatch.countDown(); |
| } |
| |
| @OnWebSocketConnect |
| public void onConnect(Session session) { |
| System.out.printf("Got connect: %s%n", session); |
| try { |
| Future<Void> fut; |
| |
| for (String message : this.toSendMessages) { |
| fut = session.getRemote().sendStringByFuture(message); |
| fut.get(5, TimeUnit.SECONDS); |
| } |
| session.close(StatusCode.NORMAL, "I'm done"); |
| } catch (Throwable t) { |
| t.printStackTrace(); |
| } |
| } |
| |
| @OnWebSocketMessage |
| public void onMessage(String msg) { |
| System.out.printf("Got msg: %s%n", msg); |
| this.receivedMessages.add(msg); |
| } |
| } |