读书笔记-《了不起的nodejs》-TCP

var net = require('net');
var count = 0;
// 用户
var users={};


// 创建服务器，每次有新的连接建立时都会被执行
var server = net.createServer(function(conn){

    // tcp是面向字节的协议，若不进行编码，获得的客户端数据是个buffer,
    // 在这里我们只获取utf8的数据，所以通过net.Stream#setCncoding来设置编码
    conn.setEncoding('utf8');

    // 当有连接建立时在终端输出，并把连接数加一    
    conn.write(
        '\n > 欢迎来到 \033[90m node-chat \033[39m! '
        +'\n 现在还有'+count + '个别人在线'
        +'\n 请输入你的名字然后按回车键: '
    );
    count++;
    var nickName;
    
    // 监听客户端发来的数据
    conn.on('data',function(data){
        console.log(data);
        // 清楚按下回车的字符
        data = data.replace('\r\n','');
        // 在本次链接之后第一次输入nickname
        console.log("nickName",!nickName);
        if(!nickName){
            if(users[data]){
                conn.write('\033[96m> 用户名已存在请重试 \033[39m ');
            }else{
                // console.log('conn',conn);
                // 如果用户名还不存在
                nickName = data;
                users[nickName] = conn;
                for(var i in users){
                    users[i].write('\033[96m'+nickName +'\033[39m 进入聊天室\n' );
                }

            }
        }else{
            for(var i in users){
                // 给除去自己以外的人发送消息
                if(i!= nickName){
                    users[i].write('\033[96m'+nickName +':\033[39m '+data+'\n' );
                }
            }

        }
    })

    // 当有连接中断时连接数减一
    conn.on('close',function(){
        count--;
    })
})

server.listen(3000,function(){
    console.log('\033[90m server listening on *:3000 \033[39m ')
})