router.js
'use strict';/** * @param {Egg.Application} app - egg application */ module.exports=app=>{const{ router, controller}= app; router.get('/', controller.home.index);// 用户注册 router.post('/reg',controller.user.reg);// 登录 router.post('/login',controller.user.login);// 退出登录 router.post('/logout',controller.user.logout);// 搜索用户 router.post('/search/user',controller.search.user);// 申请添加好友 router.post('/apply/addfriend',controller.apply.addFriend);// 获取好友申请列表 router.get('/apply/:page',controller.apply.list);// 处理好友申请 router.post('/apply/handle/:id',controller.apply.handle);// 通讯录好友申请 router.get('/friend/list',controller.friend.list);// 查看好友资料 router.get('/friend/read/:id',controller.friend.read);// 移入/移除黑名单 router.post('/friend/setblack/:id', controller.friend.setblack);// 设置/取消星标好友 router.post('/friend/setstar/:id', controller.friend.setstar);// 设置朋友圈权限 router.post('/friend/setmomentauth/:id', controller.friend.setMomentAuth);// 举报投诉好友/群组 router.post('/report/save', controller.report.save);// 设置好友备注和标签 router.post('/friend/setremarktag/:id',controller.friend.setremarkTag);// websocket app.ws.route('/ws', controller.chat.connect); app.ws.use(async(ctx, next)=>{// 获取参数 ws://localhost:7001/ws?token=123456// ctx.query.token// 验证用户tokenlet user={};let token= ctx.query.token;try{ user= ctx.checkToken(token);// 验证用户状态let userCheck=await app.model.User.findByPk(user.id);if(!userCheck){ ctx.websocket.send(JSON.stringify({ msg:"fail", data:'用户不存在'}));return ctx.websocket.close();}if(!userCheck.status){ ctx.websocket.send(JSON.stringify({ msg:"fail", data:'你已被禁用'}));return ctx.websocket.close();}// 用户上线 app.ws.user= app.ws.user? app.ws.user:{};// 下线其他设备if(app.ws.user[user.id]){ app.ws.user[user.id].send(JSON.stringify({ msg:"fail", data:'你的账号在其他设备登录'})); app.ws.user[user.id].close();}// 记录当前用户id ctx.websocket.user_id= user.id; app.ws.user[user.id]= ctx.websocket;awaitnext();}catch(err){ console.log(err);let fail= err.name==='TokenExpiredError'?'token 已过期! 请重新获取令牌':'Token 令牌不合法!'; ctx.websocket.send(JSON.stringify({ msg:"fail", data: fail}))// 关闭连接 ctx.websocket.close();}});};
chat.js
// app/controller/chat.jsconst Controller=require('egg').Controller;classChatControllerextendsController{// 连接socketasyncconnect(){const{ ctx, app}=this;if(!ctx.websocket){ ctx.throw(400,'非法访问');}// console.log(`clients: ${app.ws.clients.size}`);// 监听接收消息和关闭socket ctx.websocket.on('message',msg=>{// console.log('接收消息', msg);}).on('close',(code, reason)=>{// 用户下线 console.log('用户下线', code, reason);let user_id= ctx.websocket.user_id;if(app.ws.user&& app.ws.user[user_id]){delete app.ws.user[user_id];}});}} module.exports= ChatController;