用NodeJS实现一个简单的聊天室

今天我来实现一个简单的聊天室，后台用nodejs, 客户端与服务端通信用socket.io,这是一个比较成熟的websocket框架.

初始工作

安装express, 用这个来托管socket.io,以及静态页面,命令npm install express --save,--save可以使包添加到package.json文件里.
安装socket.io,命令npm install socket.io --save.

编写服务端代码

首先我们通过express来托管网站，并附加到socket.io实例里，因为socket.io初次连接需要http协议

var express = require('express'),
    io = require('socket.io');
var app = express();
app.use(express.static(__dirname));
var server = app.listen(8888);
var ws = io.listen(server);

添加服务器连接事件，当客户端连接成功之后，发公告告诉所有在线用户，并且，当用户发送消息时，发广播通知其它用户.

ws.on('connection', function(client){
    console.log('\033[96msomeone is connect\033[39m \n');
    client.on('join', function(msg){
        // 检查是否有重复
        if(checkNickname(msg)){
            client.emit('nickname', '昵称有重复!');
        }else{
            client.nickname = msg;
            ws.sockets.emit('announcement', '系统', msg + ' 加入了聊天室!');
        }
    });
    // 监听发送消息
    client.on('send.message', function(msg){
        client.broadcast.emit('send.message',client.nickname,  msg);
    });
    // 断开连接时，通知其它用户
    client.on('disconnect', function(){
        if(client.nickname){
            client.broadcast.emit('send.message','系统',  client.nickname + '离开聊天室!');
        }
    })

})

由于客户端是通过昵称来标识的，所以服务端需要一个检测昵称重复的函数

// 检查昵称是否重复
var checkNickname = function(name){
    for(var k in ws.sockets.sockets){
        if(ws.sockets.sockets.hasOwnProperty(k)){
            if(ws.sockets.sockets[k] && ws.sockets.sockets[k].nickname == name){
                return true;
            }
        }
    }
    return false;
}

编写客服端代码

由于服务端采用第三方websokcet框架，所以前端页面需要单独引用socket.io客户端代码，源文件可以从socket.io模块里找，windows下路径为node_modules\socket.io\node_modules\socket.io-client\dist,这里有开发版和压缩版的，默认引用开发版就行.

前端主要处理输入昵称检查，消息处理，完整代码如下

<!DOCTYPE html>
<html>
<head>
    <title>socket.io 聊天室例子</title>
    <meta charset="utf-8">
    <link rel="stylesheet" href="css/reset.css"/>
    <link rel="stylesheet" href="css/bootstrap.css"/>
    <link rel="stylesheet" href="css/app.css"/>
</head>
<body>
    <div class="wrapper">
         <div class="content" id="chat">
             <ul id="chat_conatiner">
             </ul>
         </div>
         <div class="action">
             <textarea ></textarea>
             <button class="btn btn-success" id="clear">清屏</button>
             <button class="btn btn-success" id="send">发送</button>
         </div>
    </div>
    <script type="text/javascript" src="js/socket.io.js"></script>
    <script type="text/javascript">
      var ws = io.connect('http://172.16.2.184:8888');
      var sendMsg = function(msg){
          ws.emit('send.message', msg);
      }
      var addMessage = function(from, msg){
          var li = document.createElement('li');
          li.innerHTML = '&lt;span&gt;' + from + '&lt;/span&gt;' + ' : ' + msg;
          document.querySelector('#chat_conatiner').appendChild(li);

          // 设置内容区的滚动条到底部
          document.querySelector('#chat').scrollTop = document.querySelector('#chat').scrollHeight;

          // 并设置焦点
          document.querySelector('textarea').focus();

      }

      var send = function(){
          var ele_msg = document.querySelector('textarea');
          var msg = ele_msg.value.replace('\r\n', '').trim();
          console.log(msg);
          if(!msg) return;
          sendMsg(msg);
          // 添加消息到自己的内容区
          addMessage('你', msg);
          ele_msg.value = '';
      }

      ws.on('connect', function(){
          var nickname = window.prompt('输入你的昵称!');
          while(!nickname){
              nickname = window.prompt('昵称不能为空，请重新输入!')
          }
          ws.emit('join', nickname);
      });

      // 昵称有重复
      ws.on('nickname', function(){
          var nickname = window.prompt('昵称有重复，请重新输入!');
          while(!nickname){
              nickname = window.prompt('昵称不能为空，请重新输入!')
          }
          ws.emit('join', nickname);
      });

      ws.on('send.message', function(from, msg){
          addMessage(from, msg);
      });

      ws.on('announcement', function(from, msg){
          addMessage(from, msg);
      });

      document.querySelector('textarea').addEventListener('keypress', function(event){
          if(event.which == 13){
              send();
          }
      });
      document.querySelector('textarea').addEventListener('keydown', function(event){
          if(event.which == 13){
              send();
          }
      });
      document.querySelector('#send').addEventListener('click', function(){
          send();
      });

      document.querySelector('#clear').addEventListener('click', function(){
          document.querySelector('#chat_conatiner').innerHTML = '';
      });
&lt;/script&gt;
</body>
</html>

这里提供完整的代码压缩文件

总结

nodejs是一个好东西，尤其是在处理消息通讯，网络编程方面，天生的异步IO.

yibo5220 1楼作者

用NodeJS实现一个简单的聊天室

今天我来实现一个简单的聊天室，后台用NodeJS, 客户端与服务端通信使用Socket.IO，这是一个比较成熟的WebSocket框架。

初始工作

首先，确保你已经安装了Node.js。然后，在项目目录中运行以下命令来安装所需的依赖：

npm install express socket.io --save

这将安装express和socket.io并将其添加到package.json文件中。

编写服务端代码

接下来，我们需要编写服务端代码。我们将使用Express来托管网站，并将Socket.IO附加到Express实例上。

// server.js
const express = require('express');
const http = require('http');
const socketIo = require('socket.io');

const app = express();
const server = http.createServer(app);
const io = socketIo(server);

app.use(express.static(__dirname));

server.listen(8888, () => {
    console.log('Server is running on port 8888');
});

io.on('connection', (socket) => {
    console.log('A user connected:', socket.id);

    socket.on('join', (nickname) => {
        if (!checkNickname(nickname)) {
            socket.nickname = nickname;
            io.emit('announcement', `${nickname} 加入了聊天室!`);
        } else {
            socket.emit('nickname', '昵称有重复!');
        }
    });

    socket.on('message', (data) => {
        const { nickname, message } = data;
        io.emit('message', { nickname, message });
    });

    socket.on('disconnect', () => {
        if (socket.nickname) {
            io.emit('announcement', `${socket.nickname} 离开了聊天室!`);
        }
    });
});

function checkNickname(nickname) {
    return io.sockets.adapter.rooms.has(nickname);
}

编写客户端代码

现在，让我们编写客户端代码。首先，确保你已经下载了Socket.IO客户端库。你可以直接从CDN引入它。

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Socket.IO Chat Room</title>
    <link rel="stylesheet" href="css/reset.css"/>
    <link rel="stylesheet" href="css/bootstrap.css"/>
    <link rel="stylesheet" href="css/app.css"/>
</head>
<body>
    <div class="wrapper">
        <div class="content" id="chat">
            <ul id="chat_container"></ul>
        </div>
        <div class="action">
            <textarea id="message_input"></textarea>
            <button class="btn btn-success" id="clear">清屏</button>
            <button class="btn btn-success" id="send">发送</button>
        </div>
    </div>
    <script src="/socket.io/socket.io.js"></script>
    <script>
        const socket = io('http://localhost:8888');

        const addMessage = (from, message) => {
            const li = document.createElement('li');
            li.innerHTML = `<span>${from}</span>: ${message}`;
            document.getElementById('chat_container').appendChild(li);
            document.getElementById('chat').scrollTop = document.getElementById('chat').scrollHeight;
        };

        socket.on('connect', () => {
            const nickname = prompt('请输入你的昵称:');
            if (!nickname) {
                alert('昵称不能为空！');
                socket.disconnect();
                return;
            }
            socket.emit('join', nickname);
        });

        socket.on('nickname', (msg) => {
            alert(msg);
            socket.disconnect();
        });

        socket.on('message', (data) => {
            addMessage(data.nickname, data.message);
        });

        socket.on('announcement', (data) => {
            addMessage(data.nickname, data.message);
        });

        document.getElementById('message_input').addEventListener('keypress', (event) => {
            if (event.key === 'Enter') {
                send();
            }
        });

        document.getElementById('send').addEventListener('click', send);

        function send() {
            const messageInput = document.getElementById('message_input');
            const message = messageInput.value.trim();
            if (message) {
                socket.emit('message', { nickname: socket.nid, message });
                addMessage(socket.nid, message);
                messageInput.value = '';
            }
        }

        document.getElementById('clear').addEventListener('click', () => {
            document.getElementById('chat_container').innerHTML = '';
        });
    </script>
</body>
</html>

总结

NodeJS 是一个非常适合处理实时通信和网络编程的强大工具，其内置的异步I/O模型使其在处理大量并发连接时表现出色。通过结合Socket.IO，我们可以轻松地构建出功能丰富的实时应用。希望这个简单的示例能帮助你入门NodeJS和Socket.IO。

caililin 2楼

支持一个

htzhanglong 3楼

这个聊天室支持https么？

itying888 4楼

这个是如何实现检测昵称的？并没有看到客户端要输入自己的昵称呀，发送信息的时候，只是显示“你”。

ionicwang 5楼

请问下，可以只用net模块，写个聊天室吗？

bupafengyu 6楼

mark

来自：https://github.com/manxisuo

vueper 7楼

怎么样检测内存泄漏?

gougou168 8楼

用NodeJS实现一个简单的聊天室

今天我来实现一个简单的聊天室，后台使用nodejs, 客户端与服务端通信使用socket.io，这是一个比较成熟的websocket框架。

初始工作

安装Express: 使用这个来托管socket.io及静态页面。
```
npm install express --save
```
安装Socket.IO:
```
npm install socket.io --save
```

编写服务端代码

首先我们通过Express来托管网站，并附加到socket.io实例上，因为socket.io初次连接需要http协议。

var express = require('express');
var http = require('http');
var socketIo = require('socket.io');

var app = express();
app.use(express.static(__dirname));

var server = http.createServer(app).listen(8888);

var io = socketIo.listen(server);

io.on('connection', function(socket) {
    console.log('A user connected');

    socket.on('join', function(nickname) {
        if (checkNickname(nickname)) {
            socket.emit('nickname', '昵称有重复!');
        } else {
            socket.nickname = nickname;
            io.emit('announcement', `系统`, `${nickname} 加入了聊天室!`);
        }
    });

    socket.on('send.message', function(message) {
        io.emit('send.message', socket.nickname, message);
    });

    socket.on('disconnect', function() {
        if (socket.nickname) {
            io.emit('send.message', '系统', `${socket.nickname} 离开了聊天室!`);
        }
    });
});

// 检查昵称是否重复
function checkNickname(name) {
    const sockets = io.sockets.sockets;
    for (let key in sockets) {
        if (sockets[key].nickname === name) {
            return true;
        }
    }
    return false;
}

编写客户端代码

由于服务端采用第三方websocket框架，前端页面需要引用socket.io客户端代码。

<!DOCTYPE html>
<html>
<head>
    <title>socket.io 聊天室例子</title>
    <meta charset="utf-8">
    <link rel="stylesheet" href="css/reset.css"/>
    <link rel="stylesheet" href="css/bootstrap.css"/>
    <link rel="stylesheet" href="css/app.css"/>
</head>
<body>
    <div class="wrapper">
         <div class="content" id="chat">
             <ul id="chat_container"></ul>
         </div>
         <div class="action">
             <textarea></textarea>
             <button class="btn btn-success" id="clear">清屏</button>
             <button class="btn btn-success" id="send">发送</button>
         </div>
    </div>
    <script src="/socket.io/socket.io.js"></script>
    <script>
        var socket = io.connect('http://localhost:8888');

        function sendMessage(msg) {
            socket.emit('send.message', msg);
        }

        function addMessage(from, msg) {
            var li = document.createElement('li');
            li.innerHTML = `<span>${from}</span> : ${msg}`;
            document.getElementById('chat_container').appendChild(li);
            document.getElementById('chat').scrollTop = document.getElementById('chat').scrollHeight;
            document.querySelector('textarea').focus();
        }

        document.querySelector('#send').addEventListener('click', function() {
            var ele_msg = document.querySelector('textarea');
            var msg = ele_msg.value.trim();
            if (!msg) return;
            sendMessage(msg);
            addMessage('你', msg);
            ele_msg.value = '';
        });

        socket.on('connect', function() {
            var nickname = prompt('输入你的昵称!');
            while (!nickname) {
                nickname = prompt('昵称不能为空，请重新输入!');
            }
            socket.emit('join', nickname);
        });

        socket.on('nickname', function() {
            var nickname = prompt('昵称有重复，请重新输入!');
            while (!nickname) {
                nickname = prompt('昵称不能为空，请重新输入!');
            }
            socket.emit('join', nickname);
        });

        socket.on('send.message', function(from, msg) {
            addMessage(from, msg);
        });

        socket.on('announcement', function(from, msg) {
            addMessage(from, msg);
        });

        document.querySelector('textarea').addEventListener('keypress', function(event) {
            if (event.which == 13) {
                send();
            }
        });
    </script>
</body>
</html>

以上就是用Node.js和Socket.IO实现的一个简单的聊天室。希望这些代码对你有所帮助！