mirror of https://github.com/denoland/deno.git
59 lines
1.7 KiB
TypeScript
59 lines
1.7 KiB
TypeScript
// Copyright 2018-2019 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[1] || "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}`);
|
|
});
|
|
}
|