NodeJS提供了基本的文件操作API,但是像文件拷貝這種高級功能就沒有提供,因此我們先拿文件拷貝程序練手。與copy
命令類似,我們的程序需要能接受源文件路徑與目標文件路徑兩個參數(shù)。
我們使用NodeJS內(nèi)置的fs
模塊簡單實現(xiàn)這個程序如下。
var fs = require('fs');
function copy(src, dst) {
fs.writeFileSync(dst, fs.readFileSync(src));
}
function main(argv) {
copy(argv[0], argv[1]);
}
main(process.argv.slice(2));
以上程序使用fs.readFileSync
從源路徑讀取文件內(nèi)容,并使用fs.writeFileSync
將文件內(nèi)容寫入目標路徑。
豆知識:
process
是一個全局變量,可通過process.argv
獲得命令行參數(shù)。由于argv[0]
固定等于NodeJS執(zhí)行程序的絕對路徑,argv[1]
固定等于主模塊的絕對路徑,因此第一個命令行參數(shù)從argv[2]
這個位置開始。
上邊的程序拷貝一些小文件沒啥問題,但這種一次性把所有文件內(nèi)容都讀取到內(nèi)存中后再一次性寫入磁盤的方式不適合拷貝大文件,內(nèi)存會爆倉。對于大文件,我們只能讀一點寫一點,直到完成拷貝。因此上邊的程序需要改造如下。
var fs = require('fs');
function copy(src, dst) {
fs.createReadStream(src).pipe(fs.createWriteStream(dst));
}
function main(argv) {
copy(argv[0], argv[1]);
}
main(process.argv.slice(2));
以上程序使用fs.createReadStream
創(chuàng)建了一個源文件的只讀數(shù)據(jù)流,并使用fs.createWriteStream
創(chuàng)建了一個目標文件的只寫數(shù)據(jù)流,并且用pipe
方法把兩個數(shù)據(jù)流連接了起來。連接起來后發(fā)生的事情,說得抽象點的話,水順著水管從一個桶流到了另一個桶。
更多建議: