UseStudio.NodejsSocket/uswebsocket.js

//引用http和socket.io模块
var us = {}
//包含http模块
us.http = require('http');
//包含文件模块，以实现能够读取本目录下面任意文件的功能
var fs = require('fs');
//支持目录文件模块
var path = require('path');
//包含socket.io模块
var socketio = require('socket.io');
//nodejs的underscore扩展中的findWhere方法，可以是我们很容易在对象集合中，通过对象的属性值找到该对象并返回。
var _ = require('underscore');
//构建文件读取路径。自己实现web服务器功能
us.handler=function(request, response) {
    //console.log('request starting...');
    var filePath = '.' + request.url;
    if (filePath == './')
        filePath = './index.html';

    var extname = path.extname(filePath);
    var contentType = 'text/html';
    switch (extname) {
        case '.js':
            contentType = 'text/javascript';
            break;
        case '.css':
            contentType = 'text/css';
            break;
        case '.json':
            contentType = 'application/json';
            break;
        case '.png':
            contentType = 'image/png';
            break;      
        case '.jpg':
            contentType = 'image/jpg';
            break;
        case '.wav':
            contentType = 'audio/wav';
            break;
    }

    fs.readFile(filePath, function(error, content) {
        if (error) {
            if(error.code == 'ENOENT'){
                fs.readFile('./404.html', function(error, content) {
                    response.writeHead(200, { 'Content-Type': contentType });
                    response.end(content, 'utf-8');
                });
            }
            else {
                response.writeHead(500);
                response.end('Sorry, check with the site admin for error: '+error.code+' ..\n');
                response.end(); 
            }
        }
        else {
            response.writeHead(200, { 'Content-Type': contentType });
            response.end(content, 'utf-8');
        }
    });
}

//创建httpserver,管理文件路径
us.app = us.http.createServer(us.handler);
//开启888端口，服务开始监听
//us.app.listen(888);
const hostname="127.0.0.1";
const port=888;
us.app.listen(port, hostname);
console.log(`Server running at http://${hostname}:${port}/`);
//也可以采用下面的写法,但代码更多
/*us.app.listen(port, hostname, function() {
    console.log(`Server running at http://${hostname}:${port}/`);
});*/
//输出调试信息
//console.log("Http Server start at 888");

//socketio开始监听http请求
us.socketio = socketio.listen(us.app);

//webSocket协议握手成功
////'connection' 是socket.io 保留的.不能错，结果在变量socket中
us.socketio.sockets.on('connection', function (socket) {
    //用户登录后端,以实现私聊，此处的登录只是把唯一的用户名放入内存。还未未实现cookie认证功能。
    socket.on('socketLogin',function (data) {
        //如果存在相同名称用户，弹出提示
        var toSocket= _.findWhere(us.socketio.sockets.sockets,{name:data.msg});
        if(toSocket){
            //如果用户名重复，自己给自己发送消息,现在的socket是自己的socket。
            socket.emit("UserNameRepeat",data.msg);
            return;
        }
        //以唯一的用户名给该用户设置socket。
        socket.name = data.msg;
        
        //把该用户名广播给其他用户,让其他用户好友列表中出现该用户
        socket.broadcast.emit('addFriend', data);
    });
    //向其他人发送信息
    socket.on('sayTo',function (data) {
        var toName = data.to;
        //查找是否存在该用户的socket。
        var toSocket= _.findWhere(us.socketio.sockets.sockets,{name:toName});
        //如果存在，则发送消息
        if(toSocket){
            //私聊
            toSocket.emit('sayTo',data.msg);
        }
    })

    //前端javascript使用socket.emit('msg', { msg: str }); //字符串msg是自己定义的，
    //socket.on('msg',function (data) {}); 后端需要用msg接收前端发送来的消息。'msg'需要和客户端发送时定义的事件名相同。
    socket.on('broadCast', function (data) {
        console.log('Get a msg from client ...');
        //console.log(data);
        //向客户端javascript发送广播,注意：这里广播消息为bcast，
        //则前端javascript用socket.on('bcast', function (r) {alert(r); });接收消息
        socket.broadcast.emit('broadCast', data);
    });
    //console.log("WebSocket connect ok....");

    
    //断开链接
    socket.on('disconnect', function(){
        console.log(socket.name+' connection is disconnect!');
    });
})