Node.js+websocket+mongodb实现即时聊天室

nodejs简介:Node.js是一个可以让javascript运行在服务器端的平台,它可以让javascript脱离浏览器的束缚运行在一般的服务器下面,你可以用Node.js轻松地进行服务器端应用的开发。Node.js是一个为实时Web应用开发而诞生的平台,它充分考虑了在实时响应和超大规模数据下架构的可扩展性,这使得它摒弃了传统的平台依靠多线程来实现高并发的的设计思路,而采用了单线程,异步式I/O和事件驱动的设计模式。这些特性不仅带来了巨大的性能提升,还减少了多线程程序设计的复杂性,从而提高了开发效率。

websocket:websocket协议是html5中的一种新的协议,它实现了浏览器与服务器的全双工通信。传统的我们通过http协议实现即时通讯时是这样做的,首先由浏览器对服务器发出http request(请求),然后服务器响应客户端的浏览器。这种模式所带来的弊端就是浏览器需要不断的向服务器发出请求。但是我们知道http request 的头部(header)非常长,有时客户端请求的只是很小的数据量却要附带这么长的头部信息,这样似乎在浪费网络带宽。而是用websocket协议,这时浏览器和服务器只需要进行一次握手的过程,之后,它们之间便形成了一条快速通道,接着就可以随时互相发送数据。这样不但响应速度快,而且避免了每次都发送请求头。

mongodb数据库:mongodb是一个面向文档的非关系型数据库,它具有高性能,易部署,易使用,存储数据方便等优点。它支持的数据结构很松散类似json格式。它也是面向集合的,数据被分组存放在数据集中,每个数据集就是一个集合,每个数据库包含若干个集合。

实现和效果:下面要介绍的是使用nodejs实现的即时聊天室,主要用到的是websocket协议,数据库中存放用户民和密码。首先服务器打开一个socket端口3000开始监听客户端的连接,接着客户端浏览器建立socket连接,用户登录时填写用户名和密码,服务器端查询mongodb数据库验证用户名和密码是否正确,用户登录聊天室后可以开始发送消息给其它在线的用户。以上是基本的逻辑。

下面主要贴出服务器端代码:

var express = require('express')
     ,path = require('path')
     ,app = express.createServer().listen(3000)
     ,io = require('socket.io').listen(app)
     ,mongodb = require('mongodb');
var users = [];//保存在线用户
var socketlist = [];//保存当前建立的所有socket
var socketnum = 0, usernum = 0;
//监听socket连接
io.on('connection', function(socket)
{
     socketlist[socketnum++] = socket;
     console.log('someone connected!');
     socket.send(toUserList(users));
    
     socket.on('message', function(data)
     {
              var result = eval_r('('+ data +')');
              var username = result.username;
              var content = result.content;
              var wordsize = result.wordsize;
              var wordcolor = result.wordcolor;
              var sendcontent = "
"+username+":"+content+"
"
             
              for(var i = 0; i < socketlist.length; ++i)
              {
                            if(socketlist[i])
                            {
                                          var jsonstr = '{type:"info",content:"'+sendcontent+'"}';
                                          console.log(jsonstr);
                                          socketlist[i].send(jsonstr);
                            }
              }
     });
    
});
//express基本配置
app.configure(function()
{
     app.use(express.bodyParser());//添加解释文档的中间件
     //app.use(express.methodOverride);
     app.use(app.router);
     app.use(express.static(path.join(__dirname, 'public')));//指定静态文件目录
});
app.get('/', function(req, res)
{
     res.sendfile('views/login.html', {root:__dirname});
     console.log('please login first.');
});
app.post('/login', function(req, res)
{
     var username = req.body.username;
     var password = req.body.password;
     //query the database          
    
     var dbserver = new mongodb.Server('127.0.0.1', 27017);
     new mongodb.Db('myqq', dbserver).open(function(err, client)
     {
              if(err) throw err;
              console.log('connect mongodb success!!!');
              app.user = new mongodb.Collection(client ,'users');
              app.user.findOne({name:req.body.username, password:req.body.password}, function(err, doc)
              {
                            if(doc)
                            {
                                          if(isNewer(username))
                                          {
                                                        users[usernum++] = username;
                                                        console.log(username + " login success!");
                                          }
                                          else
                                          {
                                                        console.log(username + " welcome back!");
                                          }
                                          res.setHeader("Set-Cookie", 'username='+username);
                                          res.sendfile('views/chartRoom.html', {root:__dirname});
                            }
                            else
                            {
                                          res.sendfile('views/login.html', {root:__dirname});
                                          console.log('login failure!');
                            }
              });
             
     });
});
function toUserList(userslist)
{
     var temp, list = '';
     for(var i = 0; i < userslist.length; ++i)
     {
              temp = "
" + userslist[i] + "
";
              list += temp;
     }       
     var jsonstr = '{type:"ulist",content:"'+list+'"}';
     return jsonstr;
}
function isNewer(username)
{
     for(var i = 0; i < users.length; ++i)
     {
              if(username == users[i])
                            return false;
     }
     return true;
}
console.log('server is running at port 3000...');

完整的代码请见github,后续有时间上传完整作品

原文地址:https://www.cnblogs.com/both-eyes/p/10108651.html