1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
// Copyright 2018-2020 the Deno authors. All rights reserved. MIT license.import { serve } from "../http/server.ts";import {  acceptWebSocket,  isWebSocketCloseEvent,  isWebSocketPingEvent,  WebSocket,} from "./mod.ts";
/** websocket echo server */const port = Deno.args[0] || "8080";console.log(`websocket server is running on :${port}`);for await (const req of serve(`:${port}`)) {  const { headers, conn } = req;  acceptWebSocket({    conn,    headers,    bufReader: req.r,    bufWriter: req.w,  })    .then(      async (sock: WebSocket): Promise<void> => {        console.log("socket connected!");        const it = sock.receive();        while (true) {          try {            const { done, value } = await it.next();            if (done) {              break;            }            const ev = value;            if (typeof ev === "string") {              // text message              console.log("ws:Text", ev);              await sock.send(ev);            } else if (ev instanceof Uint8Array) {              // binary message              console.log("ws:Binary", ev);            } else if (isWebSocketPingEvent(ev)) {              const [, body] = ev;              // ping              console.log("ws:Ping", body);            } else if (isWebSocketCloseEvent(ev)) {              // close              const { code, reason } = ev;              console.log("ws:Close", code, reason);            }          } catch (e) {            console.error(`failed to receive frame: ${e}`);            await sock.close(1000).catch(console.error);          }        }      }    )    .catch((err: Error): void => {      console.error(`failed to accept websocket: ${err}`);    });}