Question

我需要生成一些大小不同的文件来测试应用程序，我认为最简单的方法是编写一个Node脚本来这样做。

我编写了以下代码，但是当文件大小超出内存时，该过程将崩溃。

const fs = require("fs");
const crypto = require('crypto');

const gb = 1024 * 1024 * 1024;

const data = crypto.randomBytes(gb * 5);

fs.writeFile('bytes.bin', data, (err) => {
    if (err) throw err;
    console.log('The file has been saved!');
});

Answer 1

除了内存问题之外，加密模块也会出现问题，因为它可以生成的字节数是有限的。

您将需要使用fs.createWriteStream来生成和写入数据块，而不是一次性生成数据。

这里是Node documentation on streams中某些代码的修改版本，用于将随机字节的大块数据流传输到文件：

const fs = require("fs");
const crypto = require('crypto');

const fileName = "random-bytes.bin";

const fileSizeInBytes = Number.parseInt(process.argv[2]) || 1000;
console.log(`Writing ${fileSizeInBytes} bytes`)

const writer = fs.createWriteStream(fileName)

writetoStream(fileSizeInBytes, () => console.log(`File created: ${fileName}`));

function writetoStream(bytesToWrite, callback) {
    const step = 1000;
    let i = bytesToWrite;
    write();
    function write() {
        let ok = true;
        do {
            const chunkSize = i > step ? step : i;
            const buffer = crypto.randomBytes(chunkSize);

            i -= chunkSize;
            if (i === 0) {
                // Last time!
                writer.write(buffer, callback);
            } else {
                // See if we should continue, or wait.
                // Don't pass the callback, because we're not done yet.
                ok = writer.write(buffer);
            }
        } while (i > 0 && ok);

        if (i > 0) {
            // Had to stop early!
            // Write some more once it drains.
            writer.once('drain', write);
        }
    }
}

还有一些在线工具可以让您generate files of your required size进行更少的设置。这些文件也是在您的系统上生成的，因此不必通过网络下载。

如何向文件中写入大量随机字节

1 个答案: