socketIo 搭建长连接(2)

今天我们在这里要说的是wehsocket和node开发长连接问题,我们在真正的项目中,可能要实现的功能不知是简单的聊天功能,我们现在要整合Redis,rabbitMQ,等实现o2o的提醒功能:

首先,整合一次redis:

我们建立一个chat文件夹,在其中写入一个package.json文件,用于生成我们的node类库

{  
  "name": "zefun",  
  "version": "1.0.0",  
  "description": "",  
  "main": "index.js",  
  "dependencies": {  
    "apn": "^1.5.2",  
    "amqp": "^0.2.4",  
    "co": "^3.0.6",  
    "co-redis": "^1.1.1",  
    "log4js": "^0.6.28",  
    "express": "^4.13.3",  
    "redis": "^0.10.3",  
    "redis-sentinel": "^0.1.3",  
    "socket.io": "^1.3.7"  
  },  
  "devDependencies": {},  
  "author": "",  
  "license": "ISC"  
}  

这时,我们通过运行node的命令进行安装类库,npm install

安装结束后,我们会生成一个node_modules的文件,还要在本层的目录创建一个logs的文件夹,因为我们一会会使用一个日志系统进行输出,但是,node只有写文件的权限,没有输出文件夹的权限

新建一个redisClient.js,作为redis的客户端连接

var PORT = 6379;  
var HOST = '120.25.254.164';  
var redis = require('redis');  
var chatClient = redis.createClient(PORT,HOST);  
var logger = require("./log").logger("redis");  
chatClient.on('error', function(err){  
    logger.error(err);  
});  
var chat_sadd = function(key, field) {  
    chatClient.sadd(key, field);  
    logger.info("redis sadd --> key : " + key + ", field : " + field);  
}  
var chat_srem = function(key, field) {  
    chatClient.srem(key, field);  
    logger.info("redis srem --> key : " + key + ", field : " + field);  
}  
var chat_smembers = function(key) {  
    return chatClient.smembers(key);  
}  
var chat_hset = function(key, userId, sockedId) {  
    chatClient.hset(key, userId, sockedId);  
    logger.info("redis hset --> key : " + key + ", field : " + userId + ", value :" + sockedId);  
}   
var chat_hget = function(key, userId) {  
    return chatClient.hget(key, userId);  
}   
  
exports.chat_sadd = chat_sadd;  
exports.chat_srem = chat_srem;  
exports.chat_smembers = chat_smembers;  
exports.chat_hset = chat_hset;  
exports.chat_hget = chat_hget;  

下面我们在主js中进行引用和传值即可

var co = require('co');  
var wrapper = require('co-redis');  
var redis = wrapper(require("./redisClient")); 

我们看到上面我们使用了一个叫co的类库,并使用wrapper对redis进行了包装,只有经过包装,在操作redis的时候,才能处理了node中异步的程序,将redis取值变成了同步的方式,即取到值后才能进行下面程序的执行

[html]  view plain  copy

 print ?

1. co(function* () {  
2. <span style="white-space:pre">    </span>var userSID = yield redis.chat_hget(USER_SOCKET_KEY, toUser);  
3.     io.sockets.connected[userSID].emit('getMessage',{msg:msg});  
4. })();  

对于存值不需要这么同步

[html]  view plain  copy

 print ?

1. redis.chat_sadd(redis_key,data.userId);  

redis就算是结束了,下面就开始rabbitMQ的搭建和操作:

此处我们将node作为了rabbit的一个client进行监听和处理程序,首先新建一个mqClient.js作为队列的客户端进行监听

[html]  view plain  copy

 print ?

1. conn.queue('queue_chat_notify', { autoDelete: false, durable: true }, function(queue) {  
2.     queue.subscribe(function (msg) {  
3.       receiveNotify(msg.data);  
4.     });  
5.   });  
6.   exports.createConnection = function(r1, r2) {  
7.     receiveLogout = r1;  
8.     receiveNotify = r2;  
9.   }  

这是我们最核心的进行监听的程序,是要subscribe这个方法,就可以进行监听了,请注意receiveLogout这个方法,我们这个方式是需要在主js中作为参数传入的,当客户端进行传入一个方法作为替代方法的时候,就可以进行读取消息,但是如果没有传入,那么该条数据丢失.下面是主js中的代码

[html]  view plain  copy

 print ?

1. var mq = require("./mqClient");  
2. mq.createConnection(receiveNotify);  
3. //rabbitmq监听到通知的回调操作  
4. var receiveNotify = function(msg) {  
5.     logger.info("\r\n\r\n<-- receiveNotify begin -->");  
6.     sendMsg(msg);  
7. }  

这样,在客户端中判定登陆后,启用该方法就可以读取队列中的消息,进行发送消息了,下面给出整个mqClient.js的全部内容

[html]  view plain  copy

 print ?

1. var amqp = require("amqp");  
2.   
3. var exchName = "directExchange";  
4.   
5. var connOptions = {  
6.       host: '120.25.254.164',  
7.       port: 5672,  
8.       login: 'zefun',  
9.       password: 'zefun'  
10.     };  
11.   
12. var exchOption = {  
13.       type: 'direct',  
14.       durable: true,  
15.       autoDelete: false,  
16.       confirm: false  
17.     };  
18.   
19. var conn;  
20. var receiveNotify  = null;  
21. var logger = require('./log').logger("rabbitmq");  
22. conn = amqp.createConnection(connOptions);  
23.   
24. conn.on('ready',function() {  
25.   logger.info("rabbitmq is ready ... ");  
26.   conn.queue('queue_chat_notify', { autoDelete: false, durable: true }, function(queue) {  
27.     queue.subscribe(function (msg) {  
28.       logger.info("queue_chat_notify consumer msg : " + msg);  
29.       receiveNotify(msg.data);  
30.     });  
31.   });  
32. });  
33.   
34. conn.on('close', function(){  
35.   logger.error("rabbitmq is close");  
36. });  
37.   
38. conn.on('error', function (error) {  
39.   logger.error('Connection error : ' + error);  
40. });  
41.   
42. exports.createConnection = function(r1) {  
43.   logger.info("createConnection r1 : " + r1);  
44.   receiveNotify = r1;  
45. }  
46. exports.publish = function(routeKey, message) {  
47.   conn.publish(routeKey, message);  
48.   
49. };  

下面就是我们最后一步,看一下servier.js中的内容了

[html]  view plain  copy

 print ?

1. var app = require('express')();  
2. var http = require('http').Server(app);  
3. var io = require('socket.io')(http);  
4.   
5. app.get('/', function(req, res){  
6.     res.send('<h1>Welcome Realtime Server</h1>');  
7. });  
8.   
9. http.listen(3000, function(){  
10.     console.log('listening on *:3000');  
11. });  
12.   
13. var mq = require("./mqClient");  
14.   
15. var co = require('co');  
16. var wrapper = require('co-redis');  
17. var redis = require("./redisClient");  
18. var redisClient = wrapper(redis.chatClient);  
19.   
20. var logger = require("./log").logger("server");  
21.   
22. var STORE_USER_KEY = "store_to_chat_user_set_";  
23. var USER_SOCKET_KEY = "chat_user_to_socket_hash";  
24.   
25. //rabbitmq监听到通知的回调操作  
26. var receiveNotify = function(msg) {  
27.     logger.info("\r\n\r\n<-- receiveNotify begin -->");  
28.     sendMsg(msg);  
29. }  
30.   
31. mq.createConnection(receiveNotify);  
32.   
33. io.on('connection', function (socket) {  
34.   
35.   io.sockets.emit('connect',{hello:'connection success'});  
36.   socket.on('sendMsg', function (from,toUser,msg,msgType) {  
37.      co(function* () {  
38.          var userSID = yield redisClient.hget(USER_SOCKET_KEY, toUser);  
39.          io.sockets.connected[userSID].emit('getMessage',{msg:msg});  
40.      })();   
41.   });  
42.   
43.   socket.on('initUser',function(data){  
44.      var redis_key = STORE_USER_KEY + data.storeId;  
45.   
46.      //redis插入数据   
47.      redisClient.sadd(redis_key,data.userId);  
48.      redisClient.hset(USER_SOCKET_KEY, data.userId, socket.id);  
49.   
50.      logger.info('initUser --->> 门店 :' + data.storeId + ', 用户 : ' + data.userId);  
51.   });  
52.   
53.   socket.on('disconnect', function () {  
54.         
55.   });  
56. });  
57.   
58. var sendMsg = function(obj){  
59.     co(function* () {  
60.          var socketId = yield redisClient.hget(USER_SOCKET_KEY, obj.toUser);  
61.          logger.info("socketId : " + socketId);  
62.          io.sockets.connected[socketId].emit("getMessage", obj);  
63.     })();   
64. }  

下面,我们写一个客户端,其实就是一个jsp文件

[html]  view plain  copy

 print ?

1. <%@ page language="java" contentType="text/html; charset=UTF-8"  
2.     pageEncoding="UTF-8"%>  
3. <%  
4.     String chatPath = request.getContextPath();  
5.             String chatPasePath = request.getScheme() + "://" + request.getServerName() + ":" + request.getServerPort()  
6.                     + chatPath + "/";  
7. %>  
8. <script src="<%=chatPath%>/js/common/socket.io.js"></script>  
9. <script type="text/javascript">  
10. var userId = '${session_key_user_id}';  
11. var storeId = '${session_key_store_id}';  
12. var user = {"userId" : userId, "storeId" : storeId};  
13. if(!isEmpty(userId)){  
14.     var socket = io.connect('ws://localhost:3000');  
15.     socket.on('connect', function(data) {  
16.         //登录聊天室  
17.         socket.emit('initUser', user);  
18.           
19.         //接收消息  
20.         socket.on('getMessage', function(data) {  
21.             console.log("" + data);  
22.             var fid = data.fid;  
23.             //PC通知类处理  
24.             if (fid == 2) {  
25.                 var type = data.data.type;  
26.                 //新预约  
27.                 if (type == 2) {  
28.                     //播放语音  
29.                     textToVoice(0, data.data.msg);  
30.                 }  
31.             }  
32.         });  
33.     });  
34. }  
35. </script>  

可以进行触发登陆,然后进行接收消息即可了.

记得在Linux上启动的时候 , 使用 命令进行启动，可作为守护进程，这样node就不会随着终端的关闭而down掉。。。

到此我们的整套就写完了,最后注意一下日志,如果我们想将现在的类库导入成为package.json的话,在文件夹中使用npm init,自动生成json文件,就可以进行移植 

给出一个该项目的下载链接地址 http://download.csdn.NET/detail/u014201191/9303347

#特别注意:

在本案例中 io.sockets.connected[socket.id].emit("method",msg) 和 io.sockets.sockets[socket.id].emit("method",msg) 这两个方法都不可用,可以换成: io.to[socket.id].emit("method",msg) 触发传送事件了，好像是TM因为版本的问题。。。