NodeJS提供了基本的文件操作API,但是像文件拷貝這種高級功能就沒有提供,因此我們先拿文件拷貝程序練手。與`copy`命令類似,我們的程序需要能接受源文件路徑與目標文件路徑兩個參數。
## 小文件拷貝
我們使用NodeJS內置的`fs`模塊簡單實現這個程序如下。
~~~
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`從源路徑讀取文件內容,并使用`fs.writeFileSync`將文件內容寫入目標路徑。
> **豆知識:**?`process`是一個全局變量,可通過`process.argv`獲得命令行參數。由于`argv[0]`固定等于NodeJS執行程序的絕對路徑,`argv[1]`固定等于主模塊的絕對路徑,因此第一個命令行參數從`argv[2]`這個位置開始。
## 大文件拷貝
上邊的程序拷貝一些小文件沒啥問題,但這種一次性把所有文件內容都讀取到內存中后再一次性寫入磁盤的方式不適合拷貝大文件,內存會爆倉。對于大文件,我們只能讀一點寫一點,直到完成拷貝。因此上邊的程序需要改造如下。
~~~
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`創建了一個源文件的只讀數據流,并使用`fs.createWriteStream`創建了一個目標文件的只寫數據流,并且用`pipe`方法把兩個數據流連接了起來。連接起來后發生的事情,說得抽象點的話,水順著水管從一個桶流到了另一個桶。