1 files changed, 150 insertions, 0 deletions
diff --git a/core/http_bench.js b/core/http_bench.js
new file mode 100644
index 000000000..b9615e689
--- /dev/null
+++ b/core/http_bench.js
@@ -0,0 +1,150 @@
+// This is not a real HTTP server. We read blindly one time into 'requestBuf',
+// then write this fixed 'responseBuf'. The point of this benchmark is to
+// exercise the event loop in a simple yet semi-realistic way.
+const shared32 = new Int32Array(libdeno.shared);
+
+const INDEX_NUM_RECORDS = 0;
+const INDEX_RECORDS = 1;
+const RECORD_OFFSET_PROMISE_ID = 0;
+const RECORD_OFFSET_OP = 1;
+const RECORD_OFFSET_ARG = 2;
+const RECORD_OFFSET_RESULT = 3;
+const RECORD_SIZE = 4;
+const OP_LISTEN = 1;
+const OP_ACCEPT = 2;
+const OP_READ = 3;
+const OP_WRITE = 4;
+const OP_CLOSE = 5;
+
+const NUM_RECORDS = (shared32.length - INDEX_RECORDS) / RECORD_SIZE;
+if (NUM_RECORDS != 100) {
+  throw Error("expected 100 entries");
+}
+
+const requestBuf = new Uint8Array(64 * 1024);
+const responseBuf = new Uint8Array(
+  "HTTP/1.1 200 OK\r\nContent-Length: 12\r\n\r\nHello World\n"
+    .split("")
+    .map(c => c.charCodeAt(0))
+);
+
+const promiseMap = new Map();
+let nextPromiseId = 1;
+
+function createResolvable() {
+  let methods;
+  const promise = new Promise((resolve, reject) => {
+    methods = { resolve, reject };
+  });
+  return Object.assign(promise, methods);
+}
+
+/** Returns Promise<number> */
+function sendAsync(op, arg, zeroCopyData) {
+  const id = nextPromiseId++;
+  const p = createResolvable();
+  shared32[INDEX_NUM_RECORDS] = 1;
+  setRecord(0, RECORD_OFFSET_PROMISE_ID, id);
+  setRecord(0, RECORD_OFFSET_OP, op);
+  setRecord(0, RECORD_OFFSET_ARG, arg);
+  setRecord(0, RECORD_OFFSET_RESULT, -1);
+  promiseMap.set(id, p);
+  libdeno.send(null, zeroCopyData);
+  return p;
+}
+
+/** Returns u32 number */
+function sendSync(op, arg) {
+  shared32[INDEX_NUM_RECORDS] = 1;
+  setRecord(0, RECORD_OFFSET_PROMISE_ID, 0);
+  setRecord(0, RECORD_OFFSET_OP, op);
+  setRecord(0, RECORD_OFFSET_ARG, arg);
+  setRecord(0, RECORD_OFFSET_RESULT, -1);
+  libdeno.send();
+  return getRecord(0, RECORD_OFFSET_RESULT);
+}
+
+function setRecord(i, off, value) {
+  if (i >= NUM_RECORDS) {
+    throw Error("out of range");
+  }
+  shared32[INDEX_RECORDS + RECORD_SIZE * i + off] = value;
+}
+
+function getRecord(i, off) {
+  if (i >= NUM_RECORDS) {
+    throw Error("out of range");
+  }
+  return shared32[INDEX_RECORDS + RECORD_SIZE * i + off];
+}
+
+function handleAsyncMsgFromRust() {
+  for (let i = 0; i < shared32[INDEX_NUM_RECORDS]; i++) {
+    let id = getRecord(i, RECORD_OFFSET_PROMISE_ID);
+    const p = promiseMap.get(id);
+    promiseMap.delete(id);
+    p.resolve(getRecord(i, RECORD_OFFSET_RESULT));
+  }
+}
+
+/** Listens on 0.0.0.0:4500, returns rid. */
+function listen() {
+  return sendSync(OP_LISTEN, -1);
+}
+
+/** Accepts a connection, returns rid. */
+async function accept(rid) {
+  return await sendAsync(OP_ACCEPT, rid);
+}
+
+/**
+ * Reads a packet from the rid, presumably an http request. data is ignored.
+ * Returns bytes read.
+ */
+async function read(rid, data) {
+  return await sendAsync(OP_READ, rid, data);
+}
+
+/** Writes a fixed HTTP response to the socket rid. Returns bytes written. */
+async function write(rid, data) {
+  return await sendAsync(OP_WRITE, rid, data);
+}
+
+function close(rid) {
+  return sendSync(OP_CLOSE, rid);
+}
+
+async function serve(rid) {
+  while (true) {
+    const nread = await read(rid, requestBuf);
+    if (nread <= 0) {
+      break;
+    }
+
+    const nwritten = await write(rid, responseBuf);
+    if (nwritten < 0) {
+      break;
+    }
+  }
+  close(rid);
+}
+
+async function main() {
+  libdeno.recv(handleAsyncMsgFromRust);
+
+  libdeno.print("http_bench.js start");
+
+  const listener_rid = listen();
+  libdeno.print(`listening http://127.0.0.1:4544/ rid = ${listener_rid}`);
+  while (true) {
+    const rid = await accept(listener_rid);
+    // libdeno.print(`accepted ${rid}`);
+    if (rid < 0) {
+      libdeno.print(`accept error ${rid}`);
+      return;
+    }
+    serve(rid);
+  }
+}
+
+main();