Way of Life Welcome to Anho's World

简洁的陌生人聊天nodejs实现

最近写了个陌生人聊天应用，原型是omegle，本来是要在人人上上线的，结果因为UI的各种原因审核一直没有通过
整个应用是nodejs + socket.io实现的，一开始用的websocket，结果顾忌到强大的某浏览器用户，还是使用了socket.io中ajax部分的api
服务端实现就是存下每个用户的socket信息，匹配，发送信息。匹配的部分还有待改进，感觉没有随机性= =

var users=[];//存当前接入用户
var avail=[];//存正在等待聊天的用户
var abled=[];//存users中空缺的位置

io.sockets.on('connection', function(socket) {
    //socket连接时存入用户信息，并在用户socket中存入其在users中的位置
    if (abled.length == 0) {
        socket.set('index', users.length);
        users.push({
            socket : socket,
            available : false,
            stranger : null
        });
    } else {
        var index = abled.pop();
        socket.set('index', index);
        users[index] = {
            socket : socket,
            available : false,
            stranger : null
        }
    }

    //收到开始要求，若没有在等待用户就加入等待用户队列，有就匹配聊天
    socket.on('start', function() {
        var my_index = socket.store.data.index;

        if (avail.length == 0) {
            avail.push(my_index);
        } else {
            var index = r.rand(avail.length);
            var stranger_index = avail[index];
            avail.splice(index);
            if (users[stranger_index]) {
                if (stranger_index != my_index) {
                    users[my_index].stranger = stranger_index;
                    socket.emit('started');
                    users[stranger_index].stranger = my_index;
                    users[stranger_index].available = false;
                    users[stranger_index].socket.emit('started');
                }
            }
        }
    });

    //收到结束要求，更新用户信息
    socket.on('end', function() {
        var my_index = socket.store.data.index;
        var stranger_index = users[my_index].stranger;
        socket.emit('ended', true);
        if (users[stranger_index]) {
            users[stranger_index].socket.emit('ended', false);
			users[stranger_index].stranger = null;
        }
		users[my_index].stranger = null;
    });

    //收到发消息要求，发送给相应用户
    socket.on('send_message', function(message) {
        var my_index = socket.store.data.index;
        var stranger_index = users[my_index].stranger;
        if (users[stranger_index]) {
            users[stranger_index].socket.emit('get_message', message);
        }
    });

    //socket断开连接
    socket.on('disconnect', function() {
        var my_index = socket.store.data.index;
        var stranger_index = users[my_index].stranger;
        if (users[stranger_index]) {
            users[stranger_index].socket.emit('ended', false);
			users[stranger_index].stranger = null;
        }
        users[my_index] = null;
        abled.push(my_index);
    });
});

客户端也是一样，发送匹配、消息、断开请求

var socket = io.connect('http://' + window.location.host);

function send_message(){
    if ($('#change').attr('class') == 'start') {
        return false;
    }
    var message = $('textarea').val();
    if (message) {
        show_message('you', message);
        socket.emit('send_message', message);
        $('textarea').attr('value', '');
    }
}

function show_message(say, msg){
    $('#chat').append(message_html(say, msg))
        .scrollTop($('#chat')[0].scrollHeight);;
}

$(document).ready(function() {

    show_message('stranger','点击左下角遇见有缘的陌生人吧(￣ˇ￣)');

    $('#change').click(function() {
        var method = $(this).attr('class');
        if (method == 'start') {
            socket.emit('start');
            $('#chat').empty();
            show_message('stranger','寻找中......');
        } else {
            socket.emit('end');
        }
    });

    socket.on('started', function() {
        $button = $('#change');
        $button.text('disconnect');
        $button.attr('class', 'disconnect');
        $('#chat').empty();
        show_message('stranger','找到TA了，开始聊天吧∩ˍ∩，聊得不happy可以点左下角断开哦');
    });

    socket.on('ended', function(is_you) {
        $button = $('#change');
        $button.text('start');
        $button.attr('class', 'start');
        if (is_you) {
            show_message('stranger', '你和我断开连接了（＞﹏＜）');
        } else {
            show_message('stranger', '他离开了，再找个吧');
        }
    });

    $("#send").click(function(){
        send_message();
        $('textarea').focus();
    });

    $('textarea').keydown(function(e){
        if(e.keyCode == 13){
            send_message();
            return false;
        }
    });

    socket.on('get_message', function(message) {
        show_message('stranger', message);
    });
});

话说最近看到个什么文章，说node.js可以写桌面程序了。。。我又想起了我以前说的，很多人都想发明一种语言，这种语言什么都能干，结果JavaScript办到了。。。