example for cloudflare worker server-sent events

sse-worker.js

/**
 * Welcome to Cloudflare Workers! This is your first worker.
 *
 * - Run "npm run dev" in your terminal to start a development server
 * - Open a browser tab at http://localhost:8787/ to see your worker in action
 * - Run "npm run deploy" to publish your worker
 *
 * Learn more at https://developers.cloudflare.com/workers/
 */

// Cloudflare Worker code

/**
 * listen for fetch event
 */
addEventListener('fetch', event => {
  event.respondWith(fetchAndApply(event.request));
})

/**
 * apply event-stream to reqsponse
 * @param {Request} request 
 * @returns Response
 */
async function fetchAndApply(request) {

  const { readable, writable } = new TransformStream()
  let headers = new Headers();
  headers.append('Content-Type', 'text/event-stream');
  headers.append('Cache-Control', 'no-cache');
  headers.append('Connection', 'keep-alive');
  headers.append('Access-Control-Allow-Origin', '*');
  headers.append('Access-Control-Allow-Headers', 'Origin, X-Requested-With, Content-Type, Accept');
  const init = { "status": 200, "statusText": "ok", "headers": headers };

  writeToStream(writable);
  return new Response(readable, init);
}

/**
 * write data to writer stream
 * @param {WritableStream} writable 
 */
async function writeToStream(writable) {

  const writer = writable.getWriter();
  
  let count = 0;
  let eventId = `id-${++count}`;
  let msgBody = `{status: ${true}, text: "Hello to SSE message", time: "${new Date().toISOString()}"}`;

  // send hello message
  await constructSSE(writer, eventId, "userConnected", msgBody);

  // send message every 5 second
  setInterval(function () {
    eventId = `id-${++count}`;
    msgBody = `{status: ${true}, text: "Repeat message: ${++count}", time: "${new Date().toISOString()}"}`;
    constructSSE(writer, eventId, "userMessage", msgBody);
  }, 5000);

}

/**
 * construct message in SSE format
 * @param {WritableStreamDefaultWriter} writer 
 * @param {string|number} eventId 
 * @param {string} eventType 
 * @param {string} msgBody 
 */
async function constructSSE(writer, eventId, eventType, msgBody) {
  const encoder = new TextEncoder();

  await writer.write(encoder.encode(`id: ${eventId}` + '\n'));
  if (eventType) {
    await writer.write(encoder.encode(`event: ${eventType}` + '\n'));
  }
  await writer.write(encoder.encode(`data: ${msgBody}`+ '\n\n'));
}