nodejs 基于socket.io实现聊天室

标签：nodejs   socket.io   聊天室   express   

由于之后要做的网页视频直播项目要用到socket.io模块，所以特地花时间研究了下，参照网上的代码做了些改进，自己写了个聊天室代码。不得不承认后端事实推送能力有点厉害，这是以前我用php一直苦恼的事情。下面简单介绍下我的项目，顺带讲解下nodejs。

事实上，在看别人写的代码之前，我一直不知道nodejs是干嘛的，直到真正接触到才明白这也可以算作是服务端代码，丰富的第三方库使其功能极其强大。它可以像golang的beego一样直接通过命令行开启服务器，不过要用到express模块。加载模块的方式类似php，如下：

var express = require('express'); //引用express

var express = require('express'); //引用express
var app = express();
var server = require('http').Server(app);
//监听服务器启动
server.listen(3000, function() {
	console.log("Express server listening on port " + app.get('port'));
});

接下来就该定义路由了，事实上nodejs有路由模块，然而我还没研究过那模块具体实现了哪些功能，只借用了别人的代码来写基本的路由，大致形式如下：

app.set('views', __dirname + '/views');
app.get('/', function(req, res) {
	//res.send('hello world');
	res.sendFile(app.get("views") + '/login.html');
	//res.redirect('/login');
});

介绍完路由，再来介绍模板。既然nodejs能作为服务端语言来开发，那么自然少不了模板模块。当然，nodejs有很多模板模块，这边我只了解了ejs模块，就先只介绍这。

调用方式如下：

app.set("view engine", "ejs");
//聊天室首页
app.get('/index', function(req, res) {
	res.render("index", {
		"user": req.session.user
	});
});

在模板页面调用也很方便，直接<%=user%>即可。如下：

<input type="hidden" value="<%=user%>" id="user" />

var session = require('express-session'); //如果要使用session，需要单独包含这个模块
app.use(session({
	secret: 'ScumVirus',
	name: 'sv_chat', //这里的name值得是cookie的name，默认cookie的name是：connect.sid
	cookie: {
		maxAge: 3600000
	}, //设置maxAge是3600000ms，即1h后session和相应的cookie失效过期
	resave: false,
	saveUninitialized: true,
}));
//设置session
req.session.user = "ScumVirus";
//获取session
var name = req.session.user;

var crypto = require('crypto'); //加密
//获取md5加密后的值
var getMD5 = function(str) {
	var md5 = crypto.createHash('md5');
	md5.update(str);
	var d = md5.digest('hex');
	return d;
}

var mysql = require("mysql"); //数据库模块
//连接数据库
var connPool = mysql.createPool({
	host: '127.0.0.1', //主机
	user: 'root', //MySQL认证用户名
	password: 'admin', //MySQL认证用户密码
	port: '3306', //端口号
	database: 'sv_chat', //数据库
	waitForConnections: true, //当连接池没有连接或超出最大限制时，设置为true且会把连接放入队列
	//connectionLimit：10,//连接数限制
});
//根据用户获取用户信息
var getUserById = function(name, callback) {
	//执行SQL语句
	var sql = 'select * from sv_user where name=?';
	var params = [name];
	connPool.query(sql, params, function(err, result) {
		if (err) {
			console.log('[SELECT ERROR] - ', err.message);
			return;
		}
		return callback(result[0]);
	});
}
//添加新用户
var addUser = function(params, callback) {
	//执行SQL语句
	var sql = 'insert into sv_user(`name`,`pwd`,`email`,`phone`,`create_time`) values(?,?,?,?,?)';
	connPool.query(sql, params, function(err, result) {
		if (err) {
			console.log('[INSERT ERROR] - ', err.message);
			return callback(false);
		} else {
			console.log('INSERT ID:', result.insertId);
			return callback(true);
		}
	});
}

最后是最关键的通信阶段，socket.io模块，后台代码如下：

var io = require('socket.io').listen(server); //socket io模块
//WebSocket连接监听
io.on('connection', function(socket) {
	//socket.emit('open',onlineMember); //通知客户端已连接

	// 打印握手信息
	// console.log(socket.handshake);

	// 构造客户端对象
	var client = {
		name: '',
	}

	// 对message事件的监听
	//登录事件
	socket.on('login', function(name) {
		var time = getTime();
		client.name = name;
		var index = getArrIndex(name,onlineMember);
		if(index == -1){
			onlineMember.push(client.name);
			console.log(time + " " + client.name + " login");
		}
		var obj = {
			time: time,
			author: client.name,
			text: '',
			type: 'login',
			member: onlineMember
		};
		socket.emit('system', obj);
		socket.broadcast.emit('system', obj);
	});
	//消息事件
	socket.on('message', function(msg) {
		var obj = {
			time: getTime(),
		};
		obj['msg'] = msg;
		obj['author'] = client.name;
		obj['type'] = 'message';

		// 返回消息（可以省略）
		socket.emit('message', obj);
		// 广播向其他用户发消息
		socket.broadcast.emit('message', obj);
	});

	//监听退出事件
	socket.on('disconnect', function() {
		var index = getArrIndex(client.name, onlineMember);
		if (index > -1) {
			onlineMember.splice(index, 1);
		}
		var time = getTime();
		var obj = {
			time: time,
			author: client.name,
			text: '',
			type: 'loginout',
			member: onlineMember
		};
		console.log(time + " " + client.name + " loginout");

		// 广播用户已退出
		socket.broadcast.emit('system', obj);
	});
});

<script src="/socket.io/socket.io.js"></script>
<script>
	//建立websocket连接
	socket = io.connect('http://localhost:3000');
	var userName = $("#user").val();
	socket.emit('login', userName);
</script>

最后放上我源代码的下载地址：nodejs聊天室

nodejs 基于socket.io实现聊天室

标签：nodejs   socket.io   聊天室   express   

原文地址：http://blog.csdn.net/u013401219/article/details/47283287