Node.jsのhttp2モジュールを使ってServer Pushの送信/受信を試す

const http2 = require("http2");
const fs = require("fs");
const path = require("path");

const { HTTP2_HEADER_PATH } = http2.constants;

const server = http2.createServer();

server.on("error", console.error);

server.on("stream", (stream, headers) => {
  (async () => {
    let disconnected = false;
    stream.on("close", () => {
      disconnected = true;
    });
    for (let i = 1; i <= 10; i++) {
      if (disconnected || !stream.pushAllowed) {
        break;
      }
      stream.pushStream(
        { [HTTP2_HEADER_PATH]: "/random" + i },
        (err, pushStream) => {
          console.log(`push ${i}`);
          pushStream.respond({ ":status": 200 });
          pushStream.end(`push ${i}`);
        }
      );
      await new Promise((resolve) => setTimeout(resolve, Math.random() * 2000));
    }
    if (!disconnected) stream.end("end stream");
  })();
});

server.listen(3000);

const http2 = require("http2");

const client = http2.connect("http://localhost:3000");

client.on("stream", (pushedStream, requestHeaders) => {
  pushedStream.on("push", (responseHeaders) => {
    console.log("push", responseHeaders);
  });
  let data = "";
  pushedStream.on("data", (chunk) => {
    data += chunk;
  });
  pushedStream.on("end", () => {
    console.log(`Pushed: ${data}`);
  });
});

const req = client.request({ ":path": "/" });
req.setEncoding("utf8");

let data = "";
req.on("data", (chunk) => {
  data += chunk;
});

req.on("end", () => {
  console.log(`finish: ${data}`);
  client.close();
});
req.end();