专栏首页web秀Nodejs + WebSocket + Vue 实现多人聊天室WebIM功能 – 第二章

Nodejs + WebSocket + Vue 实现多人聊天室WebIM功能 – 第二章

前言

《Nodejs + WebSocket简单介绍及示例 - 第一章》中简单的介绍了,Nodejs + WebSocket的使用方法及作用,今天就用它来搭建一个简单的聊天室功能。

1、Nodejs+WebSocket创建后台服务器功能 2、Vue视图层,接收后台数据并渲染页面 3、LocalStorage存储会话ID等用户信息

vue + webpack 生成vue项目

脚手架搭建项目也是非常好用,简单命令即可搞定

# vue init webpack web-im

然后一路向下,填写项目名称,描述,作者等等信息,完成安装。

现在都可以自动安装模块了,当然,你可以可以到目录下面执行npm install

# cd web-im
# npm install

这就是整个生成后的项目结构。

WebSocket服务端

在项目根目录下新建server/index.js文件。

var ws = require("nodejs-websocket");
// 这里用到了moment,请大家自行安装
var moment = require('moment');

console.log("开始建立连接...")

let users = [];

// 向所有连接的客户端广播
function boardcast(obj) {
  server.connections.forEach(function(conn) {
      conn.sendText(JSON.stringify(obj));
  })
}

function getDate(){
  return moment().format('YYYY-MM-DD HH:mm:ss')
}

var server = ws.createServer(function(conn){
  conn.on("text", function (obj) {
    obj = JSON.parse(obj);
    if(obj.type===1){
      users.push({
        nickname: obj.nickname,
        uid: obj.uid
      });
      boardcast({
        type: 1,
        date: getDate(),
        msg: obj.nickname+'加入聊天室',
        users: users,
        uid: obj.uid,
        nickname: obj.nickname
      });
    } else {
      boardcast({
        type: 2,
        date: getDate(),
        msg: obj.msg,
        uid: obj.uid,
        nickname: obj.nickname
      });
    }
  })
  conn.on("close", function (code, reason) {
    console.log("关闭连接")
  });
  conn.on("error", function (code, reason) {
    console.log("异常关闭")
  });
}).listen(8001)
console.log("WebSocket建立完毕")

这里和《Nodejs + WebSocket简单介绍及示例 - 第一章》大体结构相同,不同的是,这里向客户端发送消息是用的一个方法

server.connections.forEach(function(conn) {
  conn.sendText(JSON.stringify(obj));
})

遍历所有连接,发送信息。

这里为什么要JSON.stringify(obj)转换成字符串???

那是sendText方法只能传入字符串,所以我们需要将我们的对象转换一下。

同时,大家应该可以看出,在conn.on("text", ()=>{})的时候判断了一个从客户端传入的type,这个操作是判断用户是否是第一次进入。

WebSocket客户端视图层

<div id="app">
  <c-dialog
    ref="loginDialog"
    title='请输入你的昵称'
    confirmBtn="开始聊天"
    @confirm="login"
  >
    <input class="nickname" v-model="nickname" type="text" placeholder="请输入你的昵称">
  </c-dialog>

  <div class="web-im">
    <div class="header im-title">聊天室</div>
    <div class="content im-record">
      <div class="li" :class="{user: item.uid == uid}" v-for="item in messageList">
        <template v-if="item.type===1">
          <p class="join-tips">{{item.msg}}</p>
        </template>
        <template v-else>
          <div class="img">{{item.nickname}}</div>
          <p class="message-box">{{item.msg}}</p>
        </template>
      </div>
    </div>
    <div class="footer im-input">
      <input type="text" v-model="msg" placeholder="请输入内容">
      <button @click="send">发送</button>
    </div>
  </div>
</div>

样式方面就不做解释了,都是非常简单的样式,有兴趣的可以点击最下方获取源码查看。

WebSocket客户端

export default {
  ...
  data(){
    return {
      uid: '',
      nickname: '',
      socket: '',
      msg: '',
      messageList: []
    }
  },
  mounted() {
    let vm = this;
    let user = localStorage.getItem('WEB_IM_USER');
    user = user && JSON.parse(user) || {};
    vm.uid = user.uid;
    vm.nickname = user.nickname;

    if(!vm.uid){
      vm.$refs.loginDialog.show()
    } else {
      vm.conWebSocket();
    }
    document.onkeydown = function (event) {
        var e = event || window.event;
        if (e && e.keyCode == 13) { //回车键的键值为13
            vm.send()
        }
    }
  },
  methods: {
    send(){
      if(!this.msg){
        return
      }
      this.sendMessage(2, this.msg)
    },
    sendMessage(type, msg){
      this.socket.send(JSON.stringify({
        uid: this.uid,
        type: type,
        nickname: this.nickname,
        msg: msg
      }));
      this.msg = '';
    },
    conWebSocket(){
      let vm = this;
      if(window.WebSocket){
        vm.socket = new WebSocket('ws://localhost:8001');
        let socket = vm.socket;

        socket.onopen = function(e){
          console.log("连接服务器成功");
          if(!vm.uid){
            // 生成新的用户id,并存入localStorage
            vm.uid = 'web_im_' + moment().valueOf();
            localStorage.setItem('WEB_IM_USER', JSON.stringify({
              uid: vm.uid,
              nickname: vm.nickname
            }))
            vm.sendMessage(1)
          }
        }
        socket.onclose = function(e){
          console.log("服务器关闭");
        }
        socket.onerror = function(){
          console.log("连接出错");
        }
        // 接收服务器的消息
        socket.onmessage = function(e){
          let message = JSON.parse(e.data);
          vm.messageList.push(message);
        }   
      }
    },
    login(){
      this.$refs.loginDialog.hide()
      this.conWebSocket();
    }
  }
}

页面渲染完成后,我们localStorage.getItem('WEB_IM_USER')获取本地存储是否有用户信息 1、没有用户信息,弹框填写昵称,确认开始连接,并生成一个时间戳的用户id,存入localStorage 2、有用户信息,直接连接 3、socket.onmessage监听服务器发送过来的消息,转换成json,push到messageList数组中,然后渲染到页面 4、通过type判断是新加入用户,还是正常发送消息,显示到页面 5、通过uid,判断是否是本人发送的消息,如果是消息内容靠右显示,其他用户发送的消息都靠左显示,并设置不同背景色

这样我们就完成了一个简单的node + websocket群聊功能,你从中学习到了什么了???

最后来一睹风采

总结

Nodejs + WebSocket群聊功能和核心不知道大家有没有get到了??? 其实核心代码就是它:

function boardcast(obj) {
  server.connections.forEach(function(conn) {
      conn.sendText(JSON.stringify(obj));
  })
}

向所有连接者发送消息,这样所有连接者都能接收到消息。

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • 后端小白的 Vue 入门笔记 —— 进阶篇

    出处:https://www.cnblogs.com/ZhuChangwu/p/11325489.html

    IT技术小咖
  • 8 道高频出现的 Vue 面试题及答案

    MVP 模式将 Controller 改名为 Presenter,同时改变了通信方向。

    夜尽天明
  • 后端小白的 Vue 入门笔记 —— 基础篇

    出处:https://www.cnblogs.com/ZhuChangwu/p/11303521.html

    IT技术小咖
  • 使用mpvue开发小程序教程

    网上一搜,解决的方法通常也是五花八门的,什么通过app上的globalData啊、通过存取storage啊、通过一个单独的模块(module)啊、通过Page路...

    极乐君
  • PWA入门:手把手教你制作一个PWA应用

    Web前端的同学是否想过学习app开发,以弥补自己移动端能力的不足?但在面对一众的选择时很多同学略感迷茫,是学习ios还是android开发?是学习原生开发、混...

    Fundebug
  • Vue归纳笔记:Vue 实例如何实现代理 data 对象属性的访问

    对于初学Vue.js的小伙伴而言,可能会认为Vue实例是一个很神奇的东西!因为它除了帮助我们完成双向绑定之外,还在某些细节方面为我们增加了一些理解上面的小烦恼!...

    用户1272076
  • vue/cle3项目运行报错sockjs-node/info解决方案

      继上次将vue项目脚手架工具从vue-cli2.x升级到vue-cli3.x后,除了环境、配置问题外,在运行时出现了一些个新的问题。  启动项目后,在内网非...

    流眸
  • Vue中拆分视图层代码的5点建议

    SPA框架几乎都是基于MVC或MVVM设计模式而建立起来的,这些模式都只是宏观的分层设计,当代码量开始随着项目增大而增多时,问题就会越来越多。许多企业内部的项目...

    大史不说话
  • 2019年要学习的前5个前端开发主题

    TypeScript是2018年最令人惊讶的增长故事之一.npm调查发现,有46%的npm用户使用TypeScript。它现在不仅是使用Angular的默认语言...

    技术小工
  • vscode+eslint让你编程快到飞起

    Ewall

扫码关注云+社区

领取腾讯云代金券