1 /* 2 * Copyright (c) 2017, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. 8 * 9 * This code is distributed in the hope that it will be useful, but WITHOUT 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 12 * version 2 for more details (a copy is included in the LICENSE file that 13 * accompanied this code). 14 * 15 * You should have received a copy of the GNU General Public License version 16 * 2 along with this work; if not, write to the Free Software Foundation, 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 18 * 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 20 * or visit www.oracle.com if you need additional information or have any 21 * questions. 22 */ 23 24 import jdk.incubator.http.HttpClient; 25 import jdk.incubator.http.WebSocket; 26 27 import java.io.IOException; 28 import java.net.URI; 29 30 /* 31 * @test 32 * @bug 8164625 33 * @summary Verifies HttpClient yields the connection to the WebSocket 34 * @run main/othervm -Djdk.httpclient.HttpClient.log=trace ConnectionHandover 35 */ 36 public class ConnectionHandover { 37 38 static { 39 LoggingHelper.setupLogging(); 40 } 41 42 /* 43 * An I/O channel associated with the connection is closed by WebSocket.abort(). 44 * If this connection is returned to the connection pool, then the second 45 * attempt to use it would fail with a ClosedChannelException. 46 * 47 * The assumption is that since the WebSocket client is connecting to the 48 * same URI, the pooled connection is to be used. 49 */ 50 public static void main(String[] args) throws IOException { 51 try (DummyWebSocketServer server = new DummyWebSocketServer()) { 52 server.open(); 53 URI uri = server.getURI(); 54 WebSocket.Builder webSocketBuilder = 55 HttpClient.newHttpClient().newWebSocketBuilder(uri, new WebSocket.Listener() { }); 56 57 WebSocket ws1 = webSocketBuilder.buildAsync().join(); 58 try { 59 ws1.abort(); 60 } catch (IOException ignored) { } 61 62 WebSocket ws2 = webSocketBuilder.buildAsync().join(); // Exception here if the connection was pooled 63 try { 64 ws2.abort(); 65 } catch (IOException ignored) { } 66 } 67 } 68 }