Node.js中使用socket建立私聊和公共聊天室_node.js

先給大家看效果圖：

在上篇文章跟大家介紹使用Angular和Nodejs、socket.io搭建聊天室及多人聊天室，本文繼續介紹Node.js中使用socket創建私聊和公聊聊天室，具體詳情請看下文吧。

nodejs的應用中，關於socket應該是比較出彩的了，socket.io在github上有幾萬人的star，它的成功應該是不輸於express的，為了方便了解整個socket.io的使用.

例子請點選http://chat.lovewebgames.com/

原始碼下載https://github.com/tianxiangbing/chat

由於本人太窮，所以伺服器和資料庫都是使用的國外免費的，存取速度上可以會稍慢。

先說下我對socket.io的理解，websocket更像是開啟了一個連接埠服務，來監視過往的通訊。所以我們可以依賴目前站點80埠啟動socket服務，也可以放在其他連接埠上，例如：

複製程式碼 程式碼如下:

 require('socket.io').listen(3000);

這樣就是監視3000端口了，由於我用的免費伺服器，沒有權限打開其他端口，所以，我還是使用80了，由於80已經被express使用了，所以我只好在express使用的時候傳進來了。

複製程式碼 程式碼如下:

 var server = http.createServer(app);
 var socket = require(‘./socket/msg')(server);

然後 我在msg.js裡是這樣寫的

複製程式碼 程式碼如下:

var db = require('../db/mysql');
var sio = require('socket.io');
var IO = function(server) {
var io = sio.listen(server)

這樣就和諧了，db是建立mysql連線的方法，不在本節內容裡，略。

在socket.io裡是這樣的，先創建一個io通道的連接，然後監視裡面的socket的事件，nodejs是事件驅動嘛。程式碼如下：

複製程式碼 程式碼如下:

io.on('connection', function(socket) {
        console.log('a user connected.');
        socket.on('disconnect', function() {
            console.log('user disconnected.');
        });
})

這時只要有使用者連線上，就會進入connection中了，然後它的參數是個socket,如果是公聊，我們可以直接用

複製程式碼 程式碼如下:

io.emit('chat message', {});

這種形式了。但我們這裡是私聊，所以我們要臨時的把這個socket對象保存在全局裡，供與你私聊的對象使用找到你的socket,很繞口，其實這裡的私聊，不算完全的點對點，它還是經過了伺服器的，訊息傳給伺服器，伺服器再找到你要傳達給的那個人的socket對象，發給他。這就是整個的過程了。這裡我使用的是一個類別數組物件來儲存的.

複製程式碼 程式碼如下:

var users = {},
usocket = {};
socket.on('user join', function(data) {
    users[username] = username;
    usocket[username] = socket;
})

由於我這裡需要用戶名登錄，所以我就把用戶名作為了唯一的標識（這只是一個例子，不要跟我談用戶名重複的情況），這裡用類數組的形式的好處就是我不用循環也能夠很快的找到它。再我寄私聊給A時，我會先在這個uscoket裡面找到它，然後呼叫它的emit。

function sendUserMsg(data) {
 if (data.to in usocket) {
 console.log('================')
 console.log('to' + data.to, data);
 usocket[data.to].emit('to' + data.to, data);
 usocket[data.user].emit('to' + data.user, data);
 console.log('================')
 }
}

這裡我emit了兩次的原因是，我發給對方訊息的同時，我自己也要收到這個訊息，然後把它顯示出來，為什麼這樣？其一，介面統一了，聊天裡的內容全是伺服器過來的，其二，證明我發送成功了。

然後我在客戶端監聽時，也用我自己的用戶名起了一個to 用戶名的事件監聽。

socket.on('to' + user, function(data) {
 //console.log(data);
 formatMsg(data);
})

這樣，不管是我發的訊息，還是我收到訊息，都會進入這個事件了。最後，在使用者離開的時候別忘記delete掉這個物件。

socket.on('disconnect', function() {
 console.log('disconnect')
 if (username) {
 counter--;
 delete users[username];
 delete usocket[username];
 if (home.name == username) {
  homeLeave(username);
 }
 sendmsg({
  type: 0,
  msg: "用户<b>" + username + "</b>离开聊天室",
  counter: counter,
  users: users
 })
 }
});

好了，這樣就大功告成了。