NodeJS+Redis实现Web实时通讯聊天模块

var redis_host = '127.0.0.1', redis_port = 6379;
var redis = require("redis"), client = redis.createClient(redis_port, redis_host);

var io = require('socket.io').listen(4000);
io.set('log level', 1);
io.set('transports', ['websocket' ,'flashsocket' ,'htmlfile' ,'xhr-polling' ,'jsonp-polling']);

var online_user_chat_key = "chat:";
var online_user_list_key = "user:list";

var sockets = [];
var user_info_hash = {};
var socket_user_hash = {};

io.sockets.on('connection', function (socket) {

socket.on('add-user', function(data) {
var user = JSON.parse(data).user;
var id_user = user.id;
user_info_hash[id_user] = user;
socket_user_hash[socket.id] = id_user;
client.sadd(online_user_list_key, id_user, function(err, res){
update_user_list(id_user, socket);
});
});

socket.on('send-message', function(data){
var msg = JSON.parse(data);
var id_target = msg.id_target;
var id_user = msg.id_user;
var message = msg.message;
var date_string = msg.date;
var id_chat = [id_user, id_target].sort().join("_");
var json = JSON.stringify({ 'id_user': id_user, 'id_target': id_target,  'message': message, 'date': date_string });
client.lpush(online_user_chat_key + id_chat, json, function(err, res){
if (undefined != sockets[id_chat]) {
for (var id_user in sockets[id_chat]){
for (var socket_id in sockets[id_chat][id_user]) {
sockets[id_chat][id_user][socket_id].emit('receive-message', json);
}
}
}
});
});

socket.on("disconnect", function() {
var user_id = socket_user_hash[socket.id];
if(undefined != user_id) {
client.srem(online_user_list_key, user_id, function(){
update_user_list(user_id, socket);
});
for(var id_chat in sockets) {
for(var id_user in sockets[id_chat]) {
if(id_user == user_id) {
delete sockets[id_chat][id_user];
}
}
}
}
});

socket.on("add-socket", function(data) {
var msg = JSON.parse(data);
var id_target = msg.id_target;
var id_user = msg.id_user;
store_socket(id_user, id_target, socket);
});
});

function update_user_list(id_user, socket) {
client.smembers(online_user_list_key, function(error, users) {
if(users.length > 0) {
var user_info = {};
for(var i in users) {
var user_id = users[i];
if(undefined != user_info_hash[user_id]) {
user_info[user_id] = user_info_hash[user_id];
}
}
show_online_users(user_info, id_user, socket);
}
});
}

function show_online_users(user_info, id_user, socket) {
var reply = JSON.stringify({ id_user: id_user, user_info: user_info });
io.sockets.emit('show-online', reply);
for(var id_target in user_info) {
store_socket(id_user, id_target, socket);
}
}

function store_socket(id_user, id_target, socket) {
if(id_user != id_target) {
var id_chat = [id_user, id_target].sort().join("_");
if (undefined == sockets[id_chat]) {
sockets[id_chat] = {};
}
if (undefined == sockets[id_chat][id_user]) {
sockets[id_chat][id_user] = {}
}
sockets[id_chat][id_user][socket.id] = socket;
}
}

<script type="text/javascript" src="http://<?php echo $hostname; ?>:4000/socket.io/socket.io.js"></script>
<script type="text/javascript">

$(document).ready(function(){
$('.chat_icon_here img').each(function(){
$(this).attr('title', $(this).attr('alt'));
})
});

var info;
var month = ['Jan', 'Feb', 'Mar', 'Feb', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec'];
var new_msg_sound;
soundManager.setup({
url: '/sounds/swf/',
flashVersion: 9,
onready: function() {
// Ready to use; soundManager.createSound() etc. can now be called.
new_msg_sound = soundManager.createSound({
id: 'sound_new_msg',
url: '/sounds/msg.mp3'
});
}
});

var chat_logs = {};
var id_user = '<?php echo $userId; ?>';
var socket = io.connect('http://<?php echo $hostname; ?>:4000');
socket.emit('add-user', $.toJSON({ user: <?php echo json_encode($login_user); ?> }));

socket.on('show-online', function (msg) {
var message = $.evalJSON(msg);
info = message.user_info;
$("#user_list").empty();
var user_account = 0;
for(var user in info) {
try {
var username = (info[user].first_name)? (info[user].first_name+ " "+ info[user].last_name): info[user].username;
var avator = (info[user].avator)? info[user].avator : "/images/chat/chat_9.png";
if(user != id_user) {
//var image = "<li rel='" + user + "'><a href='javascript:show_chat(" +user+ ");' title='" +username+ "' class='showChat' user='" +user+ "'><image src='" +avator+ "' alt='" +username+ "' width='28'></a></li>";
var image ="<li rel='" + user + "'><span class='chat_photo'><a href='javascript:show_chat(" +user+ ");' class='showChat' title='Click to Chat with " +username+ "' user='"+ user+ "'><image src='" +avator+ "' alt='" +username+ "' width='28' height='28'></a></span><strong>" +username+ "</strong><span class='cl'></span></li>";
$("#user_list").append(image);
user_account++;
}
}
catch(e){}
}
$("#chat_amount").html(user_account);
var id_target = message.id_user;
socket.emit('add-socket', $.toJSON({ id_user: id_user, id_target: id_target }));
});

$('#message_box').keydown(function(event) {
if (event.keyCode == 13) {
var message = $('#message_box').val().trim();
if(!message) {
return false;
}
var id_target = $("#target").attr("rel");
var date = new Date();
var year = date.getFullYear()
var mon = date.getMonth()
var day = date.getDate()
var date_string = month[mon] + " "+ day + ", " + year;
socket.emit('send-message', $.toJSON({ id_target: id_target, id_user: id_user, message: message, date: date_string }));
if (chat_logs[id_target] == undefined) {
chat_logs[id_target] = {};
}
if (chat_logs[id_target][date_string] == undefined) {
chat_logs[id_target][date_string] = [];
}
chat_logs[id_target][date_string].push(id_user+"@_@"+ message);
$('#message_box').val("");
return false;
}
});

socket.on('receive-message', function (msg) {
var message = $.evalJSON(msg);
//console.log(message);
var target = message.id_user;
var id_target = message.id_target;
var avator = info[target].avator;
if(avator == null) {
avator = "/images/chat/chat_9.png";
}
var date_string = message.date;
var msg_user = info[target].first_name + " " + info[target].last_name;
var msg_content = decode_emotion(message.message);
if (chat_logs[target] == undefined) {
chat_logs[target] = {};
}
if (chat_logs[target][date_string] == undefined) {
chat_logs[target][date_string] = [];
}
chat_logs[target][date_string].push(target+"@_@"+msg_content);

if( ($('#chat_window').css("display") == "block" && $("#target").attr("rel") ==  target) || ($('#chat_window').css("display") == "block" && $("#target").attr("rel") == id_target) ) {
$("#chat_content").append(date_string+ "<hr size='1' /><img width='28' height='28' src='"+ avator+ "' alt='"+ msg_user+ "' align='top' /> " + msg_content);
$("#chat_content").append("<br/>");
$("#user_list a.showChat[user="+ target+ "] img").css({"border": "0px solid red"});
$("#chat_content")[0].scrollTop = $("#chat_content")[0].scrollHeight;
}
else {
$("#user_list a.showChat[user="+ target+ "] img").css({"border": "2px solid red"});
if (undefined != new_msg_sound){
new_msg_sound.play();
}
}
});

function emotion(emo) {
if(emo) {
$("#message_box").val($("#message_box").val() + " [:"+ emo+ "] ");
$('#chat_icon_all').slideToggle();
$("#message_box").focus();
}
}

function decode_emotion(emo) {
var ret_val = emo.replace(/\[:(([^\]]+))\]/gi, "<img  align='absmiddle' width='35' height='35' src='/images/chat/$1.png' title='$1' alt='$1' />");
return ret_val;
}

function show_chat(id) {
var target = id;
var username = info[target].first_name + " " + info[target].last_name;
var avator = info[target].avator;
if(avator == null) {
avator = "/images/chat/chat_9.png";
}
if(target != id_user) {
$("#target").text(username).attr("rel", target);
var chat_content = "";
if (chat_logs[target] != undefined) {
//console.log(chat_logs);
for(date_string in chat_logs[target]) {
var list = chat_logs[target][date_string];
for(i=0; i<list.length; i++)  {
var tmp = list[i].split("@_@");
var message_field = tmp[1];
var avator = info[tmp[0]].avator;
chat_content += date_string+ "<hr size='1' /><img width='28' height='28' src='"+ avator+ "' alt='"+ username+ "' align='top' /> " + decode_emotion(message_field)+ "<br />";
}
}
}
$("#chat_content").html(chat_content);
$("#user_list a.showChat[user="+ target+ "] img").css({"border": "0px solid red"});
$('#chat_window').slideToggle(500, function(){
$("#chat_content")[0].scrollTop = $("#chat_content")[0].scrollHeight;
});
}
}

function logout_user() {
socket.emit('disconnect');
}
</script>