NodeJS服务器退出:完成任务,优雅退出

上一篇文章,我们通过一个简单的例子,学习了NodeJS中对客户端的请求(request)对象的解析和处理,整个文件共享的功能已经完成。但是,纵观整个过程,还有两个地方明显需要改进:

首先,不能共享完毕之后,都通过Ctrl+C来关闭NodeJS服务器。

其次,如果仅仅能向客户端提供d:\ilinkit_logo.png文件的下载,是没有意义的,共享哪个文件,应该可以通过传入的参数来指定。

老规矩,先上一个图:

NodeJS服务器退出:完成任务,优雅退出

我们首先来实现退出功能,如果客户端向服务器提交了http://localhost:8000/exit的请求,我们就执行服务器的退出操作。

上一篇文章我们已经能够识别出/xiaohong的请求,所以这个实现起来很简单,代码如下:

 var http = require( 'http' );
var fs = require('fs');
var url = require( 'url' ); var file_path = "D:\\ilinkit_logo.png" ;
var file_stream ;
var buffer_box = [] ;
var file_length = 0 ; var file_name = file_path.substr( file_path.lastIndexOf('\\')+1 ); fs.stat( file_path , function ( err , stat ){
if (err) {
if ('ENOENT' == err.code) {
console.log( 'File does not exist...' );
} else {
console.log( 'Read file exception...' );
}
} else {
file_stream = fs.createReadStream( file_path );
file_stream.on( 'data' , function( chunk ){
buffer_box.push( chunk ) ;
file_length += chunk.length ;
} );
file_stream.on( 'end' , function( ){
console.log( "文件读取完毕" );
} );
file_stream.on('error', function(err){
console.log( "文件读取失败!" );
}); var server =http.createServer( function ( request ,response ){
var h_name = request.headers.host ;
var h_path = url.parse( request.url ).pathname ; if( h_path === '/xiaohong' ){
response.setHeader( 'Content-Type' , 'application/octet-stream' );
response.setHeader( 'Content-Disposition' , 'attachment; filename=' + encodeURIComponent(file_name) ); for( var buffer_index = 0 ; buffer_index<buffer_box.length ; buffer_index++ )
{
response.write( buffer_box[buffer_index] );
}
response.end();
}
else if( h_name === 'localhost:8000' && h_path === '/exit' ){
response.end('Bye!');
server.close() ;
console.log( 'Bye!' );
}
else{
response.end( 'Hello, iLinkIT' );
} } );
server.listen( 8000 );
console.log( 'HTTP服务器启动中,端口:8000.....' ); }//end else,读取文件没有发生错误
});

关键的代码解析如下:

第33行,我们通过request对象获取客户端请求的主机及端口内容。

第46行~第50行,我们判断客户端提交的请求信息,如果是http://localhost:8000/exit,则调用server.close()关闭服务器。为什么要判断是不是localhost提交的请求?因为我们希望仅仅在服务器本地提交的请求才能关闭NodeJS服务器。

验证方式如下:

1. 启动服务器:打开命令行,进入js脚本所在的位置,执行:node h_ilinkit_1.js

2. 打开浏览器,输入:http://localhost:8000,显示如下:

NodeJS服务器退出:完成任务,优雅退出

说明当前服务器启动正常。

3. 打开浏览器,输入:http://localhost:8000/exit。

NodeJS服务器退出:完成任务,优雅退出

提示NodeJS服务器已经关闭,我们把浏览器关闭之后,发现服务器已经正常退出,如下所示。

NodeJS服务器退出:完成任务,优雅退出

这样,我们就没必要每次为了退出服务器,都去按Ctrl + C了。

通过请求退出服务器就介绍到这里,接下来我们再看一下,如果在启动NodeJS服务器的时候,给它传入参数。对应到我们爱莲(iLinkIT)的场景,希望能够将要共享的文件的路径作为参数传递给NodeJS服务器,服务器根据传入的文件路径读取数据到缓冲区,接受客户端的响应。

代码如下:

 var http = require( 'http' );
var fs = require('fs');
var url = require( 'url' ); var args = process.argv.splice( 2 );
var file_path = args.join( '' ) ;
var file_stream ;
var buffer_box = [] ;
var file_length = 0 ; var file_name = file_path.substr( file_path.lastIndexOf('\\')+1 ); fs.stat( file_path , function ( err , stat ){
if (err) {
if ('ENOENT' == err.code) {
console.log( 'File does not exist...' );
} else {
console.log( 'Read file exception...' );
}
} else {
file_stream = fs.createReadStream( file_path );
file_stream.on( 'data' , function( chunk ){
buffer_box.push( chunk ) ;
file_length += chunk.length ;
} );
file_stream.on( 'end' , function( ){
console.log( "文件读取完毕" );
} );
file_stream.on('error', function(err){
console.log( "文件读取失败!" );
}); var server =http.createServer( function ( request ,response ){
var h_name = request.headers.host ;
var h_path = url.parse( request.url ).pathname ; if( h_path === '/xiaohong' ){
response.setHeader( 'Content-Type' , 'application/octet-stream' );
response.setHeader( 'Content-Disposition' , 'attachment; filename=' + encodeURIComponent(file_name) ); for( var buffer_index = 0 ; buffer_index<buffer_box.length ; buffer_index++ )
{
response.write( buffer_box[buffer_index] );
}
response.end();
}
else if( h_name === 'localhost:8000' && h_path === '/exit' ){
response.end('Bye!');
server.close() ;
console.log( 'Bye!' );
}
else{
response.end( 'Hello, iLinkIT' );
} } );
server.listen( 8000 );
console.log( 'HTTP服务器启动中,端口:8000.....' ); }//end else,读取文件没有发生错误
});

关键代码解析如下:

第5行,通过process.argv.splice( 2 )获得了传入的命令行参数。

之前我们启动NodeJS服务器的命令为:node h_ilinkit_1.js,而要传入参数之后,执行的命令为node h_ilinkit_2.js d:\ilinkit_logo.rar

process.argv会将输入命令行的所有的内容都获取到,包括node h_ilinkit_2.js部分,我们通过调用splice( 2 ),获得传入的第3个参数的内容,将前面的两个字符串剔除掉。

第6行,将输入命令行的内容中,除了node h_ilinkit_2.js之外的内容合并在一起,作为文件路径。

验证方式如下:

1. 启动服务器:打开命令行,进入js脚本所在的位置,执行:node h_ilinkit_2.js d:\ilinkit_logo.rar,如下图所示:

NodeJS服务器退出:完成任务,优雅退出

3. 打开浏览器,输入:http://localhost:8000/xiaohong,显示如下:

NodeJS服务器退出:完成任务,优雅退出

可见,我们已经实现了通过命令行传入参数的功能,因为我们传入的是d:\ilinkit_logo.rar,所以,客户端提交请求后,下载到的文件也是ilinkit_logo.rar。

简单回顾一下:

1. 借助NodeJS的服务器响应机制,通过给服务器提交/exit的请求,实现服务器的退出操作。

2. 通过在启动NodeJS时,向服务器传入共享文件的路径,实现共享文件的自定义,这样,想共享哪个文件,就可以共享哪个文件。

关于通过爱莲(iLinkIT)这个小项目讲解NodeJS的特性的文章,到此已经全部结束,不知道您是否已经对NodeJS产生爱恋^_^?更深入的了解NodeJS,推荐您根据业务需要,找专门的专著来阅读,另外,泡NodeJS社区论坛,也是一个扩大认识,解决问题的好办法,没错,就是这个社区:https://cnodejs.org/

感谢您的捧场,晚安^_^~~

-----------------------爱莲(iLinkIT)系列文章------------------------------------------

缘起爱莲:我要的,现在就要!

爱莲(iLinkIT)的架构与原理

遇见NodeJS:JavaScript的贵人

NodeJS服务器:一行代码 = 一个的HTTP服务器

NodeJS文件读取:感恩常在--抓把糖果,愉悦客人

NodeJS缓存机制:畅销货,就多囤一点呗

NodeJS安全设计:好吃的草莓味糖果,只给好朋友小红

NodeJS服务器退出:完成任务,优雅退出

上一篇:win32自绘按钮,使用GDI+(三)


下一篇:常量指针(const X*)和指针常量(X* const)