|
1 | 1 | package hello; |
2 | 2 |
|
3 | | -import java.net.InetAddress; |
4 | | -import java.net.InetSocketAddress; |
5 | 3 | import java.nio.ByteBuffer; |
6 | | -import java.util.HashMap; |
| 4 | +import java.util.Collections; |
7 | 5 | import java.util.Map; |
8 | 6 |
|
9 | 7 | import com.fasterxml.jackson.databind.ObjectMapper; |
10 | | -import io.undertow.UndertowOptions; |
11 | | -import io.undertow.io.IoCallback; |
| 8 | +import com.fasterxml.jackson.databind.ObjectWriter; |
| 9 | +import io.undertow.Undertow; |
12 | 10 | import io.undertow.server.HttpHandler; |
13 | | -import io.undertow.server.HttpOpenListener; |
14 | 11 | import io.undertow.server.HttpServerExchange; |
15 | 12 | import io.undertow.util.Headers; |
16 | | -import org.xnio.BufferAllocator; |
17 | | -import org.xnio.ByteBufferSlicePool; |
18 | | -import org.xnio.ChannelListener; |
19 | | -import org.xnio.ChannelListeners; |
20 | | -import org.xnio.OptionMap; |
21 | | -import org.xnio.Options; |
22 | | -import org.xnio.Pool; |
23 | | -import org.xnio.StreamConnection; |
24 | | -import org.xnio.Xnio; |
25 | | -import org.xnio.XnioWorker; |
26 | | -import org.xnio.channels.AcceptingChannel; |
| 13 | + |
| 14 | +import static io.undertow.Undertow.builder; |
27 | 15 |
|
28 | 16 | public class HelloWebServer { |
29 | 17 |
|
30 | | - private static final ObjectMapper mapper = new ObjectMapper(); |
| 18 | + private static final ObjectWriter writer = new ObjectMapper().writerWithType(Map.class); |
| 19 | + |
| 20 | + public static final String HELLO_WORLD = "Hello, World!"; |
31 | 21 |
|
32 | 22 | private final int port; |
| 23 | + private final ByteBuffer buffer; |
33 | 24 |
|
34 | 25 | public HelloWebServer(int port) { |
35 | 26 | this.port = port; |
| 27 | + buffer = ByteBuffer.allocateDirect(HELLO_WORLD.getBytes().length); |
| 28 | + buffer.put(HELLO_WORLD.getBytes()); |
| 29 | + buffer.flip(); |
36 | 30 | } |
37 | 31 |
|
38 | 32 | public void run() throws Exception { |
39 | 33 |
|
40 | | - Xnio xnio = Xnio.getInstance("nio", HelloWebServer.class.getClassLoader()); |
41 | | - XnioWorker worker = xnio.createWorker(OptionMap.builder() |
42 | | - .set(Options.WORKER_IO_THREADS, Runtime.getRuntime().availableProcessors() * 2) |
43 | | - .set(Options.CONNECTION_HIGH_WATER, 1000000) |
44 | | - .set(Options.CONNECTION_LOW_WATER, 1000000) |
45 | | - .set(Options.TCP_NODELAY, true) |
46 | | - .set(Options.CORK, true) |
47 | | - .getMap()); |
48 | | - |
49 | | - OptionMap serverOptions = OptionMap.builder() |
50 | | - .set(Options.TCP_NODELAY, true) |
51 | | - .set(Options.REUSE_ADDRESSES, true) |
52 | | - .getMap(); |
53 | | - |
54 | | - Pool<ByteBuffer> buffers = new ByteBufferSlicePool(BufferAllocator.DIRECT_BYTE_BUFFER_ALLOCATOR, 2048, 2048 * 2048); |
55 | | - |
56 | | - HttpHandler rootHandler = new HttpHandler() { |
57 | | - @Override |
58 | | - public void handleRequest(final HttpServerExchange exchange) throws Exception { |
59 | | - exchange.getResponseHeaders().put(Headers.CONTENT_TYPE, "application/json"); |
60 | | - Map<String, String> data = new HashMap<String, String>(); |
61 | | - data.put("message", "Hello, world"); |
62 | | - String response = mapper.writeValueAsString(data); |
63 | | - exchange.getResponseHeaders().put(Headers.CONTENT_LENGTH, response.length()); |
64 | | - exchange.getResponseSender().send(response, IoCallback.END_EXCHANGE); |
65 | | - } |
66 | | - }; |
67 | | - |
68 | | - HttpOpenListener openListener = new HttpOpenListener(buffers, OptionMap.create(UndertowOptions.BUFFER_PIPELINED_DATA, true), 2048); |
69 | | - openListener.setRootHandler(rootHandler); |
70 | | - ChannelListener<AcceptingChannel<StreamConnection>> acceptListener = ChannelListeners.openListenerAdapter(openListener); |
71 | | - AcceptingChannel<? extends StreamConnection> server = worker.createStreamConnectionServer(new InetSocketAddress(InetAddress.getByAddress(new byte[]{0, 0, 0, 0}), port), acceptListener, serverOptions); |
72 | | - server.resumeAccepts(); |
73 | | - |
| 34 | + Undertow undertow = builder() |
| 35 | + .addListener(port, "0.0.0.0") |
| 36 | + .setBufferSize(1024 * 16) |
| 37 | + .setHandler(new HttpHandler() { |
| 38 | + @Override |
| 39 | + public void handleRequest(final HttpServerExchange exchange) throws Exception { |
| 40 | + if (exchange.getRelativePath().equals("/plaintext")) { |
| 41 | + exchange.getResponseHeaders().put(Headers.CONTENT_TYPE, "text/plain"); |
| 42 | + exchange.getResponseSender().send(buffer.duplicate()); |
| 43 | + } else { |
| 44 | + exchange.getResponseHeaders().put(Headers.CONTENT_TYPE, "application/json"); |
| 45 | + Map<String, String> data = Collections.singletonMap("message", "Hello, world"); |
| 46 | + String response = writer.writeValueAsString(data); |
| 47 | + exchange.getResponseSender().send(response); |
| 48 | + } |
| 49 | + } |
| 50 | + }).build(); |
74 | 51 |
|
| 52 | + undertow.start(); |
75 | 53 | } |
76 | 54 |
|
77 | 55 | public static void main(String[] args) throws Exception { |
|
0 commit comments