基於node的tcp客戶端和服務端的簡單通信

  • 2022 年 2 月 27 日
  • 筆記

1.簡單介紹下TCP/IP

TCP/IP是互聯網相關協議的集合,分為以下四層:應用層、傳輸層、網絡層、數據鏈路層。

分成四層的好處是,假如只有一層,某個地方需要改變設計時,就必須把所有整體替換掉,而分層之後只需把變動的層替換掉即可。

2. 服務端編程

node提供了net模塊來實現tcp編程。主要分為服務端編程和客戶端編程兩部分,先來寫服務端的:

監聽客戶端連接

先引入net模塊,再通過 net.createServer 創建一個TCP服務器,服務器可以監聽一個connection事件。

net.createServer([options][, connectionListener])

  • options<Object>

    • allowHalfOpen<boolean> 表明是否允許半開的 TCP 連接。默認值: false
    • pauseOnConnect<boolean> 表明是否應在傳入連接上暫停套接字。默認值: false
  • connectionListener<Function> 自動設置為 'connection' 事件的監聽器

我們先不傳參數:

var net = require('net');
//創建tcp服務器
var server = net.createServer(); 

//監聽connect事件
server.on('connection', (socket) => {
    socket.on('data',(data)=>{
        console.log(data.toString(), 'data')
    })
})

//設置監聽端口
server.listen(8989);

//設置監聽時的回調函數
server.on('listening', (res)=>{
    console.log('in listen...')
})

connection事件當有數據發送過來時會被觸發。監聽函數的參數是一個socket,用戶可以使用它與客戶端進行交互。

通過socket的data事件可以打印出發送過來的數據,在瀏覽器中打開 //localhost:8989,瀏覽器會將請求頭的信息發送到server,所以輸出結果如下:

 上面代碼中,connection事件可以傳入到createServer中,listen和listening可以合併,寫成下面這種形式:

//第二種寫法
var net = require('net');
//創建tcp服務器
var server = net.createServer((socket) => {
    socket.on('data', (data) => {
        console.log(data.toString(), 'data..')
    })
});
//設置監聽端口
server.listen(8989, (res) => {
    console.log('in listen...')
});

除了listening和connect事件外,server還有如下事件:

  • close:TCP服務器關閉的時候觸發,回調函數沒有參數
  • error:TCP服務器發生錯誤的時候觸發,回調函數的參數為error對象

接收數據和發送數據

通過socket的data事件接收數據,write方法發送數據

socket.write(data[, encoding][, callback])

//創建一個TCP服務器
var server = net.createServer((socket) => {
    //'connect'事件的回調函數
    console.log('客戶端已連接');
    socket.on('end', () => {
        console.log('客戶端已斷開');
    })
    //接收來自客戶端的數據
    socket.on('data', (data) => {
        console.log(data.toString(), 'data');
        var readSize = socket.bytesRead;
        console.log('the size of data is ' + readSize);
    })

    //向客戶端寫入數據
    socket.write('hello\r\n');

    //設置連接最大數量,可不設
    server.maxConnection = 3;

    server.getConnections(function (err, count) {

        console.log('the count of clieent is ' + count);

    });

})

server.listen(8989, () => {
    console.log('服務器已啟動');
    //獲取地址信息
    var address = server.address();

    //獲取地址端口
    console.log('the port of server is ' + address.port);
    console.log('the address of server is ' + address.address);
    console.log('the ip family of server is ' + address.family);
})

我們先不寫客戶端的代碼,先用telnet工具來當作客戶端來測試上面代碼

打開命令行工具輸入 telnet localhost 8989

 

 可以看到  socket.write(‘hello\r\n’) 這句話已經生效了,服務端向客戶端返回了hello,但這樣肯定還不夠,

3. 客戶端編程

客戶端編程比服務端簡單,因為不用監聽端口。只要把數據從指定的端口發出去就可以了。

net.createConnection(options[, connectListener]) 用來創建一個socket。第一個參數必填,要寫發送的端口號,第二個參數是這個socket的 'connect'事件的回調函數

//創建一個客戶端
var client = net.connect({ port: 8989 }, () => {
    console.log('連接到服務器');
    //向服務端發送數據
    client.write('hello,i am from client');
})

//監聽事件,當服務端發送數據過來時會觸發該事件
client.on('data', (data) => {
    console.log(data, 'data');
    client.end()
})

client.on('end', () => {
    console.log('已從服務器斷開');
});

 上面代碼的client就是一個socket,所以可以監聽data事件來獲取服務端發送來的數據。然後socket也可以通過write來向服務端發送數據

4. 一個小的登錄系統

利用上面的知識,可以實現一個小型的登錄系統:

/*server.js*/
var net = require('net');
//這兒用一個對象來模擬數據庫,保存用戶名和密碼
var user = {
    admin: '123',
    test: '333',
    lucy: '222',
}
//臨時變量保存用戶輸入的內容
var username = '';

var server = net.createServer((socket)=>{
    console.log('服務器已連接');
    socket.write('請輸入用戶名:');
    socket.on('data',(data)=>{
        //通過stdin輸入的內容是buffer,需要轉成字符串且清除空格
        data = data.toString().trim();
        if(!username){
            if(user[data]){
                socket.write('請輸入密碼:');
                username = data;
            }else{
                socket.write('用戶名不對,請輸入用戶名:');
            }
        }else {
            if(user[username] === data){
                socket.write('登錄成功');
            }else {
                socket.write('密碼不對,請輸入密碼:');
            }
        }
    });
    socket.on('close',data=>{
        console.log(data);
    })
})

server.listen(8899,()=>{
    console.log('服務器監聽8899端口中')
})
/*client.js*/
var net = require('net');
process.stdin.resume(); 

var client = net.createConnection({port: 8899},()=>{
    //process.stdin可以獲取到用戶的輸入
    process.stdin.on('data',input=>{
        client.write(input)
    })
})

//這兒接收服務端的返回數據
client.on('data',data=>{
    console.log(data.toString());
    if(data.toString()==='登錄成功'){
        process.exit();
    }
})