From 05ebf81122468a09593a13c037f2e6644c5efda2 Mon Sep 17 00:00:00 2001
From: bugyaluwang <jiabowang@moego.pet>
Date: Fri, 13 Sep 2024 18:07:58 +0800
Subject: [PATCH] feat: add websocket demo to koa

---
 apps/koa/src/index.js | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/apps/koa/src/index.js b/apps/koa/src/index.js
index 19f10f0..451e2f1 100644
--- a/apps/koa/src/index.js
+++ b/apps/koa/src/index.js
@@ -1,3 +1,5 @@
+const WebSocket = require('ws')
+
 const run = () => {
   let register
   try {
@@ -70,6 +72,20 @@ const run = () => {
     ctx.body = data
   })
 
+  router.get('/ws', async (ctx) => {
+    const ws = new WebSocket('wss://echo.websocket.org/')
+    ws.onopen = () => {
+      ws.send('Hello from Koa')
+    }
+    ws.onmessage = (event) => {
+      console.log('WebSocket message:', event.data)
+    }
+    ws.onclose = () => {
+      console.log('WebSocket connection closed')
+    }
+    ctx.body = 'WebSocket connection established'
+  })
+
   app.use(router.routes())
   app.listen(3000)