Node.js的socket通信和C++、Java的非常相像,学过这两种语言的socket通信的同学可以很快就掌握好Node.js的socket通信。下面我们以实现一个Echo服务器的服务端和客户端为目的,学习一下Node.js的socket通信。
所谓的Echo服务器指的是这样一种服务器:客户端发送一条消息给服务端,服务端就把这条消息原封不动地返回给客户端。
服务端
服务端的实现分为三步:
- 通过createServer创建一个server服务端
- 使用server的listen方法监听指定端口,等待客户端接入
- 通过socket对象来监听data、close等事件,用以完成与客户端的交互
下面是服务端代码:
/**
* Created by Administrator on 2015/9/8.
*/
var net = require('net');
// 服务器IP
var HOST = '127.0.0.1';
// 端口号
var PORT = 7001;
// 创建一个TCP服务器实例,调用listen函数开始监听指定端口
// 传入net.createServer()的回调函数将作为”connection“事件的处理函数
// 在每一个“connection”事件中,该回调函数接收到的socket对象是唯一的
net.createServer(function(sock) {
// 获得了一个socket连接,将客户端输出来
console.log('CONNECTED: ' +
sock.remoteAddress + ':' + sock.remotePort);
// 为这个socket实例添加一个"data"事件处理函数
sock.on('data', function(data) {
console.log('DATA ' + sock.remoteAddress + ': ' + data);
// 回发该数据,客户端将收到来自服务端的数据,实现ECHO服务器
sock.write('' + data );
});
// 为这个socket实例添加一个"close"事件处理函数
sock.on('close', function(data) {
console.log('CLOSED: ' +
sock.remoteAddress + ' ' + sock.remotePort);
});
}).listen(PORT, HOST);
console.log('Server listening on ' + HOST +':'+ PORT);
使用createListener的时候会传入一个回调函数,当客户端连接上服务端时调用。在回调函数里,完成了对data、close事件的监听。
- data事件:当客户端有消息发送过来时触发此事件
- close事件:当客户端断开连接时触发此事件
客户端
因为要实现的是Echo服务器,所以自然要实现输入与输出,这里,我使用了上一篇介绍的Readline模块来实现控制台的输入与输出,对这一模块不太熟悉的同学,可以先去看一下Readline模块的介绍。
与服务端相似,客户端的实现也分为三步:
- 创建一个socket
- 连接上指定IP指定端口号的服务端
- 监听data、close等事件完成与服务端的交互
客户端代码如下:
/**
* Created by Administrator on 2015/9/8.
*/
var net = require('net');
var readline = require('readline');
// 创建Readline接口
rl = readline.createInterface(process.stdin, process.stdout);
var HOST = '127.0.0.1';
var PORT = 7001;
var client = new net.Socket();
client.connect(PORT, HOST, function() {
console.log('CONNECTED TO: ' + HOST + ':' + PORT);
// 连接建立后,开始执行命令行输入
rl.setPrompt('Test> ');
rl.prompt();
});
// 为客户端添加“data”事件处理函数
// data是服务器发回的数据
client.on('data', function(data) {
// 输出服务器端传来的消息
console.log(""+data);
rl.prompt();
});
// 为客户端添加“close”事件处理函数
client.on('close', function() {
console.log('Connection closed');
process.exit(0);
});
rl.on('line', function(line) {
switch(line.trim()) {
case 'close':
rl.close();
return;
}
client.write(line.trim());
rl.prompt();
}).on('close', function() {
console.log('Good bye!');
client.end();
});
这里应该注意,客户端socket监听了data和close事件、我们的输入输出模块也监听了它们自己的line、close事件,前者是client变量,后者是rl变量,要区别开来。
当我们输入close后,会发生以下过程
- line事件里调用Readline.close()触发了Readline的close事件;
- 在Readline的close事件里,调用client.end()结束客户端,从而触发客户端的close事件;
- 客户端的close事件监听器里执行程序退出操作 process.exit(0).
通信结果
客户端:
服务端:
通信涉及三个过程:
- 客户端接收用户输入信息后将发往服务端;
- 服务端接收客户端传来信息后将原封返回给客户端;
- 客户端接收服务端信息后将其显示出来。
这里只是简单介绍了一下Node.js的基本的用法,如果想要了解更多用法,可以看官方的API文档。