NodeJs 大文件拷貝

NodeJS提供了基本的文件操作API，但是像文件拷貝這種高級(jí)功能就沒(méi)有提供，因此我們先拿文件拷貝程序練手。與copy命令類似，我們的程序需要能接受源文件路徑與目標(biāo)文件路徑兩個(gè)參數(shù)。

小文件拷貝

我們使用NodeJS內(nèi)置的fs模塊簡(jiǎn)單實(shí)現(xiàn)這個(gè)程序如下。

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)容寫入目標(biāo)路徑。

豆知識(shí)： process是一個(gè)全局變量，可通過(guò)process.argv獲得命令行參數(shù)。由于argv[0]固定等于NodeJS執(zhí)行程序的絕對(duì)路徑，argv[1]固定等于主模塊的絕對(duì)路徑，因此第一個(gè)命令行參數(shù)從argv[2]這個(gè)位置開始。

大文件拷貝

上邊的程序拷貝一些小文件沒(méi)啥問(wèn)題，但這種一次性把所有文件內(nèi)容都讀取到內(nèi)存中后再一次性寫入磁盤的方式不適合拷貝大文件，內(nèi)存會(huì)爆倉(cāng)。對(duì)于大文件，我們只能讀一點(diǎn)寫一點(diǎn)，直到完成拷貝。因此上邊的程序需要改造如下。

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)建了一個(gè)源文件的只讀數(shù)據(jù)流，并使用fs.createWriteStream創(chuàng)建了一個(gè)目標(biāo)文件的只寫數(shù)據(jù)流，并且用pipe方法把兩個(gè)數(shù)據(jù)流連接了起來(lái)。連接起來(lái)后發(fā)生的事情，說(shuō)得抽象點(diǎn)的話，水順著水管從一個(gè)桶流到了另一個(gè)桶。