使用node+vue實現簡單的WebSocket聊天功能

最近學習了一下websocket的即時通信，感覺非常的強大，這里我用node啟動了一個服務進行websocket鏈接，然后再vue的view里面進行了鏈接，進行通信，廢話不多說，直接上代碼吧，

首先，我需要用到node的nodejs-websocket模塊

使用yarn進行安裝

yarn add nodejs-websocket --save

當然，你也可以用npm進行安裝

npm i nodejs-websocket --save

安裝完畢之后，我們開始寫服務端的代碼，首先，我用node在本地起了一個node服務器用來開啟websocket服務

sock.js:

let ws = require("nodejs-websocket");
console.log("開始建立鏈接");
ws.createServer(function (conn) {
  conn.on("text", function (str) {
    console.log("收到的信息為", str);
      conn.send(`${str}（機器人`)
  });
  conn.on("close", function (code, reason) {
    console.log("關閉連接")
  });
  conn.on("error", function (code, reason) {
    console.log("異常關閉")
  })
}).listen(8001);
console.log("鏈接建立完畢");

服務端主要是用nodejs-websocket用來開啟服務，以及返回前端需要的值，這里我只是做了一個簡單的處理，在接受值得后面加了一個‘機器人’的string,

然后，我們需要開啟這個node服務，

命令后面的路徑一定要找對，我是把sock.js放在了根目錄的socket文件夾下面

執行

yarn socket

　　

最后，看我們的客戶端，客戶端我是想有一個輸入框，然后有個聊天框：

<template>
  <div class="test3">
    <div class="msg" ref="box">
      <div v-for="item in list" :class="[item.type,'msg-item']">
        <p>
          {{item.content}}
        </p>
      </div>
    </div>
    <div class="input-group">
      <input type="text" v-model="contentText">
      <button @click="sendText">發送</button>
    </div>
  </div>
</template>

<script>
  export default {
    name: "index3",
    data() {
      return {
        list: [],//聊天記錄的數組
        contentText: "",//input輸入的值
      }
    },
    methods: {
      //發送聊天信息
       sendText() {
        let that = this;
        this.list = [...this.list, {type: "mine", content: this.contentText}];//通過type字段進行區分是自己（mine）發的還是系統（robot）返回的
        this.backText(function () {
          that.contentText = "";//加回調在得到返回數據的時候清除輸入框的內容
        });
      },
      backText(callback) {
        let that = this;
        if (window.WebSocket) {
          let ws = new WebSocket("ws://192.168.11.169:8001");
          ws.onopen = function (e) {
            console.log("鏈接服務器成功");
            console.log("that.contentText is", that.contentText);
            ws.send(that.contentText);
            callback();
          };
          ws.onclose = function (e) {
            console.log("服務器關閉")
          };
          ws.onerror = function () {
            console.log("服務器出錯")
          };
          ws.onmessage = function (e) {
            that.list = [...that.list, {type: "robot", content: e.data}]
          }
        }
      }
    },
    watch: {
      //監聽list,當有修改的時候進行div的屏幕滾動，確保能看到最新的聊天
      list: function () {
        let that = this;
        setTimeout(() => {
          that.$refs.box.scrollTop = that.$refs.box.scrollHeight;
        }, 0);
        //加setTimeout的原因：由於vue采用虛擬dom，我每次生成新的消息時獲取到的div的scrollHeight的值是生成新消息之前的值，所以造成每次都是最新的那條消息被隱藏掉了
      }
    },
    mounted() {
    }
  };


</script>

<style scoped lang="scss">
  .test3 {
    text-align: center;
  }

  .msg {
    width: 100px;
    height: 100px;
    overflow: auto;
    padding-top: 5px;
    border: 1px solid red;
    display: inline-block;
    margin-bottom: 6px;

    .msg-item {
      position: relative;
      overflow: hidden;
      p {
        display: inline-block;
        border-radius: 40px;
        background: #3C3D5A;
        color: white;
        float: left;
        padding: 2px 12px;
        margin: 0 0 2px 0;
        max-width: 70%;
        text-align: left;
        box-sizing: border-box;
      }

      &.mine {
        p {
          float: right;
          background: aquamarine;
          color: white;
        }
      }
    }
  }

</style>

　　看一下最終效果：