koa2 長連接 koa-websocket

引入npm依賴

npm i -S koa2
npm i -S koa-route
npm i -S koa-websocket

　　

1.app.js

const port = 9501;
const Koa = require('koa2')
// 路由
const route = require('koa-route')
// koa封裝的websocket這是官網（很簡單有時間去看一下https://www.npmjs.com/package/koa-websocket）
const websockify = require('koa-websocket')
const app = websockify(new Koa());
app.ws.use(function (ctx, next) {
    ctx.websocket.send("連接成功");
    return next(ctx)
})
app.ws.use(route.all('/', function (ctx) {
    /**接收消息*/
    ctx.websocket.on('message', function (message) {
        console.log(message);
        // 返回給前端的數據
        let data = JSON.stringify({
            id: Math.ceil(Math.random()*1000),
            time: parseInt(new Date()/1000)
        })
        ctx.websocket.send(data);
    })
}));

app.listen(port, () => {
    console.log("localhost:" + port);
});