6 import java.util.function.*;
7 import java.util.concurrent.*;
8 import java.util.logging.*;
11 import java.nio.channels.*;
13 public class EventServer implements Runnable {
14 private static final double timeout = 5;
15 private static final Logger log = Logger.getLogger("jagi.server");
16 private final ServerSocketChannel sk;
17 private final Function handler;
18 private final Driver ev = Driver.get();
19 private final ExecutorService handlers = new ThreadPoolExecutor(0, Runtime.getRuntime().availableProcessors() * 2,
20 5, TimeUnit.SECONDS, new LinkedBlockingQueue<Runnable>(64),
21 tgt -> new Thread(tgt, "Request handler thread"));
23 public EventServer(ServerSocketChannel sk, Function handler) {
25 this.handler = handler;
28 public static class Request {
29 public final Map<Object, Object> env;
30 public final SocketChannel sk;
32 public Request(Map<Object, Object> env, SocketChannel sk) {
38 ArrayList<Object> cleanup = new ArrayList<>((Collection<?>)env.get("jagi.cleanup"));
40 RuntimeException ce = null;
41 for(Object obj : cleanup) {
42 if(obj instanceof AutoCloseable) {
44 ((AutoCloseable)obj).close();
45 } catch(Exception e) {
47 ce = new RuntimeException("error(s) occurred during cleanup");
57 protected void error(Request req, Throwable error) {
58 log.log(Level.WARNING, "uncaught exception while handling request", error);
61 public static abstract class ChainWatcher implements Watcher {
62 private Runnable then;
63 public ChainWatcher then(Runnable then) {this.then = then; return(this);}
71 public static class BufferedOutput extends ChainWatcher {
72 public final SocketChannel sk;
73 public final ByteBuffer buf;
74 private double lastwrite;
76 public BufferedOutput(SocketChannel sk, ByteBuffer buf) {
81 public void added(Driver d) {lastwrite = d.time();}
82 public SelectableChannel channel() {return(sk);}
83 public int events() {return((buf.remaining() > 0) ? SelectionKey.OP_WRITE : -1);}
84 public double timeout() {return(lastwrite + timeout);}
86 public void handle(int events) throws IOException {
87 double now = Driver.current().time();
88 if((events & SelectionKey.OP_WRITE) != 0) {
92 if(now > lastwrite + timeout)
93 buf.position(buf.limit());
97 public static class TransferOutput extends ChainWatcher {
98 public final SocketChannel sk;
99 public final ReadableByteChannel in;
100 private final ByteBuffer buf;
101 private boolean eof = false;
102 private double lastwrite;
104 public TransferOutput(SocketChannel sk, ReadableByteChannel in) {
107 buf = ByteBuffer.allocate(65536);
111 public void added(Driver d) {lastwrite = d.time();}
112 public SelectableChannel channel() {return(sk);}
113 public int events() {return((eof && (buf.remaining() == 0)) ? -1 : SelectionKey.OP_WRITE);}
114 public double timeout() {return(lastwrite + timeout);}
116 public void handle(int events) throws IOException {
117 if(!eof && (buf.remaining() == 0)) {
119 while(buf.remaining() > 0) {
124 double now = Driver.current().time();
125 if((events & SelectionKey.OP_WRITE) != 0) {
126 if(sk.write(buf) > 0)
129 if(now > lastwrite + timeout) {
131 buf.position(buf.limit());
135 public void close() {
138 } catch(IOException e) {
139 log.log(Level.WARNING, "failed to close transfer channel: " + in, e);
146 public static class TransferInput extends ChainWatcher {
147 public final SocketChannel sk;
148 public final WritableByteChannel out;
149 private final ByteBuffer buf;
150 private final long max;
151 private boolean eof = false;
152 private double lastread;
153 private long cur = 0;
155 public TransferInput(SocketChannel sk, WritableByteChannel out, long max) {
159 buf = ByteBuffer.allocate(65536);
163 public void added(Driver d) {lastread = d.time();}
164 public SelectableChannel channel() {return(sk);}
165 public int events() {return(eof ? -1 : SelectionKey.OP_READ);}
166 public double timeout() {return(lastread + timeout);}
168 public void handle(int events) throws IOException {
169 double now = Driver.current().time();
170 if((events & SelectionKey.OP_READ) != 0) {
172 if(buf.remaining() > max - cur)
173 buf.limit(buf.position() + (int)Math.min(max - cur, Integer.MAX_VALUE));
174 int rv = sk.read(buf);
182 while(buf.remaining() > 0)
185 if(now > lastread + timeout) {
187 buf.position(buf.limit());
191 public void close() {
194 } catch(IOException e) {
195 log.log(Level.WARNING, "failed to close transfer channel: " + out, e);
202 protected void respond(Request req, String status, Map resp) {
203 Object output = resp.get("jagi.output");
204 ByteArrayOutputStream buf = new ByteArrayOutputStream();
206 Writer head = new OutputStreamWriter(buf, Utils.UTF8);
207 head.write("Status: ");
210 for(Iterator it = resp.entrySet().iterator(); it.hasNext();) {
211 Map.Entry ent = (Map.Entry)it.next();
212 Object val = ent.getValue();
213 if((ent.getKey() instanceof String) && (val != null)) {
214 String key = (String)ent.getKey();
215 if(key.startsWith("http.")) {
216 String nm = key.substring(5);
217 if(nm.equalsIgnoreCase("status"))
219 if(val instanceof Collection) {
220 for(Object part : (Collection)val) {
223 head.write(part.toString());
229 head.write(val.toString());
237 } catch(IOException e) {
238 throw(new RuntimeException("cannot happen"));
242 out = new BufferedOutput(req.sk, ByteBuffer.allocate(0));
243 } else if(output instanceof byte[]) {
244 out = new BufferedOutput(req.sk, ByteBuffer.wrap((byte[])output));
245 } else if(output instanceof ByteBuffer) {
246 out = new BufferedOutput(req.sk, (ByteBuffer)output);
247 } else if(output instanceof String) {
248 out = new BufferedOutput(req.sk, ByteBuffer.wrap(((String)output).getBytes(Utils.UTF8)));
249 } else if(output instanceof CharSequence) {
250 out = new BufferedOutput(req.sk, Utils.UTF8.encode(CharBuffer.wrap((CharSequence)output)));
251 } else if(output instanceof InputStream) {
252 out = new TransferOutput(req.sk, Channels.newChannel((InputStream)output));
253 } else if(output instanceof ReadableByteChannel) {
254 out = new TransferOutput(req.sk, (ReadableByteChannel)output);
256 throw(new IllegalArgumentException("response-body: " + output));
258 out.then(() -> submit(req::close));
259 ev.add(new BufferedOutput(req.sk, ByteBuffer.wrap(buf.toByteArray())).then(() -> ev.add(out)));
262 @SuppressWarnings("unchecked")
263 protected void handle(Request req, Function handler) {
264 boolean handoff = false;
266 Throwable error = null;
268 Map resp = (Map)handler.apply(req.env);
270 if((st = (String)resp.get("jagi.status")) != null) {
271 Function next = (Function)resp.get("jagi.next");
274 Object sink = resp.get("jagi.input-sink");
275 Object clen = req.env.get("HTTP_CONTENT_LENGTH");
277 if(clen instanceof String) {
279 max = Long.parseLong((String)clen);
280 } catch(NumberFormatException e) {
283 if(sink instanceof WritableByteChannel) {
284 ev.add(new TransferInput(req.sk, (WritableByteChannel)sink, max).then(() -> submit(() -> handle(req, next))));
285 } else if(sink instanceof OutputStream) {
286 ev.add(new TransferInput(req.sk, Channels.newChannel((OutputStream)sink), max).then(() -> submit(() -> handle(req, next))));
288 throw(new IllegalArgumentException("input-sink: " + sink));
293 submit(() -> handle(req, next));
297 throw(new IllegalArgumentException("jagi.status: " + st));
299 } else if((st = (String)resp.get("http.status")) != null) {
300 respond(req, st, resp);
303 throw(new IllegalArgumentException("neither http.status nor jagi.status set"));
305 } catch(Throwable t) {
312 } catch(Throwable ce) {
316 error.addSuppressed(ce);
321 } catch(Throwable t) {
326 protected void submit(Runnable task) {
327 handlers.submit(task);
330 class Client implements Watcher {
331 final SocketChannel sk;
333 boolean eof = false, handoff = false;
335 ByteBuffer head = null;
336 Map<Object, Object> env = null;
339 Client(SocketChannel sk) {
343 public void added(Driver d) {lastread = d.time();}
344 public SelectableChannel channel() {return(sk);}
345 public double timeout() {return(lastread + timeout);}
346 public int events() {
350 return(SelectionKey.OP_READ);
354 boolean readhead() throws IOException {
356 ByteBuffer buf = ByteBuffer.allocate(1);
359 int rv = sk.read(buf);
366 lastread = Driver.current().time();
368 if((c >= '0') && (c <= '9')) {
369 headlen = (headlen * 10) + (c - '0');
370 } else if(c == ':') {
371 head = ByteBuffer.allocate(headlen + 1);
381 if(head.remaining() == 0) {
382 if(head.get(head.limit() - 1) != ',') {
383 /* Unterminated netstring */
387 head.limit(head.limit() - 1);
388 env = Jagi.mkenv(Scgi.splithead(head), sk);
391 int rv = sk.read(head);
401 public void handle(int events) throws IOException {
402 if((events & SelectionKey.OP_READ) != 0) {
403 if((env == null) && !readhead())
405 req = new Request(env, sk);
408 if(Driver.current().time() > (lastread + timeout))
412 public void close() {
414 submit(() -> EventServer.this.handle(req, handler));
418 } catch(IOException e) {
424 class Accepter implements Watcher {
425 boolean closed = false;
427 public SelectableChannel channel() {return(sk);}
428 public int events() {return(SelectionKey.OP_ACCEPT);}
430 public void handle(int events) throws IOException {
431 if((events & SelectionKey.OP_ACCEPT) != 0)
432 Driver.current().add(new Client(sk.accept()));
435 public void close() {
444 Accepter main = new Accepter();
448 while(!main.closed) {
452 } catch(InterruptedException e) {
457 } catch(IOException e) {
458 throw(new RuntimeException(e));