| // |
| // ======================================================================== |
| // Copyright (c) 1995-2016 Mort Bay Consulting Pty. Ltd. |
| // ------------------------------------------------------------------------ |
| // All rights reserved. This program and the accompanying materials |
| // are made available under the terms of the Eclipse Public License v1.0 |
| // and Apache License v2.0 which accompanies this distribution. |
| // |
| // The Eclipse Public License is available at |
| // http://www.eclipse.org/legal/epl-v10.html |
| // |
| // The Apache License v2.0 is available at |
| // http://www.opensource.org/licenses/apache2.0.php |
| // |
| // You may elect to redistribute this code under either of these licenses. |
| // ======================================================================== |
| // |
| |
| package org.eclipse.jetty.server; |
| |
| import static org.junit.Assert.assertEquals; |
| |
| import java.io.IOException; |
| import java.net.Socket; |
| import java.util.concurrent.CountDownLatch; |
| import java.util.concurrent.TimeUnit; |
| import java.util.concurrent.atomic.AtomicInteger; |
| |
| import javax.servlet.AsyncContext; |
| import javax.servlet.ServletException; |
| import javax.servlet.http.HttpServletRequest; |
| import javax.servlet.http.HttpServletResponse; |
| |
| import org.eclipse.jetty.io.Connection; |
| import org.eclipse.jetty.server.handler.AbstractHandler; |
| import org.eclipse.jetty.util.IO; |
| import org.junit.Test; |
| |
| public class HalfCloseTest |
| { |
| @Test |
| public void testHalfCloseRace() throws Exception |
| { |
| Server server = new Server(); |
| ServerConnector connector = new ServerConnector(server,1,1); |
| connector.setPort(0); |
| connector.setIdleTimeout(500); |
| server.addConnector(connector); |
| TestHandler handler = new TestHandler(); |
| server.setHandler(handler); |
| |
| server.start(); |
| |
| try(Socket client = new Socket("localhost",connector.getLocalPort());) |
| { |
| int in = client.getInputStream().read(); |
| assertEquals(-1,in); |
| |
| client.getOutputStream().write("GET / HTTP/1.0\r\n\r\n".getBytes()); |
| |
| Thread.sleep(200); |
| assertEquals(0,handler.getHandled()); |
| } |
| |
| } |
| |
| @Test |
| public void testCompleteClose() throws Exception |
| { |
| Server server = new Server(); |
| ServerConnector connector = new ServerConnector(server,1,1); |
| connector.setPort(0); |
| connector.setIdleTimeout(5000); |
| final AtomicInteger opened = new AtomicInteger(0); |
| final CountDownLatch closed = new CountDownLatch(1); |
| connector.addBean(new Connection.Listener() |
| { |
| @Override |
| public void onOpened(Connection connection) |
| { |
| opened.incrementAndGet(); |
| } |
| |
| @Override |
| public void onClosed(Connection connection) |
| { |
| closed.countDown(); |
| } |
| |
| }); |
| server.addConnector(connector); |
| TestHandler handler = new TestHandler(); |
| server.setHandler(handler); |
| |
| server.start(); |
| |
| try(Socket client = new Socket("localhost",connector.getLocalPort());) |
| { |
| client.getOutputStream().write("GET / HTTP/1.0\r\n\r\n".getBytes()); |
| IO.toString(client.getInputStream()); |
| assertEquals(1,handler.getHandled()); |
| assertEquals(1,opened.get()); |
| } |
| assertEquals(true,closed.await(1,TimeUnit.SECONDS)); |
| } |
| |
| @Test |
| public void testAsyncClose() throws Exception |
| { |
| Server server = new Server(); |
| ServerConnector connector = new ServerConnector(server,1,1); |
| connector.setPort(0); |
| connector.setIdleTimeout(5000); |
| final AtomicInteger opened = new AtomicInteger(0); |
| final CountDownLatch closed = new CountDownLatch(1); |
| connector.addBean(new Connection.Listener() |
| { |
| @Override |
| public void onOpened(Connection connection) |
| { |
| opened.incrementAndGet(); |
| } |
| |
| @Override |
| public void onClosed(Connection connection) |
| { |
| closed.countDown(); |
| } |
| |
| }); |
| server.addConnector(connector); |
| AsyncHandler handler = new AsyncHandler(); |
| server.setHandler(handler); |
| |
| server.start(); |
| |
| try(Socket client = new Socket("localhost",connector.getLocalPort());) |
| { |
| client.getOutputStream().write("GET / HTTP/1.0\r\n\r\n".getBytes()); |
| IO.toString(client.getInputStream()); |
| assertEquals(1,handler.getHandled()); |
| assertEquals(1,opened.get()); |
| } |
| assertEquals(true,closed.await(1,TimeUnit.SECONDS)); |
| } |
| |
| public static class TestHandler extends AbstractHandler |
| { |
| transient int handled; |
| |
| public TestHandler() |
| { |
| } |
| |
| @Override |
| public void handle(String target, Request baseRequest, HttpServletRequest request, HttpServletResponse response) throws IOException, ServletException |
| { |
| baseRequest.setHandled(true); |
| handled++; |
| response.setContentType("text/html;charset=utf-8"); |
| response.setStatus(HttpServletResponse.SC_OK); |
| response.getWriter().println("<h1>Test</h1>"); |
| } |
| |
| public int getHandled() |
| { |
| return handled; |
| } |
| } |
| |
| public static class AsyncHandler extends AbstractHandler |
| { |
| transient int handled; |
| |
| public AsyncHandler() |
| { |
| } |
| |
| @Override |
| public void handle(String target, Request baseRequest, final HttpServletRequest request, final HttpServletResponse response) throws IOException, ServletException |
| { |
| baseRequest.setHandled(true); |
| handled++; |
| |
| final AsyncContext async = request.startAsync(); |
| new Thread() |
| { |
| @Override |
| public void run() |
| { |
| try |
| { |
| response.setContentType("text/html;charset=utf-8"); |
| response.setStatus(HttpServletResponse.SC_OK); |
| response.getWriter().println("<h1>Test</h1>"); |
| } |
| catch (Exception ex) |
| { |
| System.err.println(ex); |
| } |
| finally |
| { |
| async.complete(); |
| } |
| } |
| }.start(); |
| } |
| |
| public int getHandled() |
| { |
| return handled; |
| } |
| } |
| } |