如何在Node.js中对输入流应用正则表达式并返回结果流？

要实现过滤输入流中匹配指定正则的块并返回结果流，你需要创建一个Transform流（Node.js流API中的转换流），它能一边读取输入数据，一边处理并输出符合条件的内容。以下是完整实现：

const stream = require('stream');

function processStream(is, expression) {
    // 创建Transform流，处理数据过滤
    const transformStream = new stream.Transform({
        readableObjectMode: true,
        writableObjectMode: true,
        transform(chunk, encoding, callback) {
            // 测试当前chunk是否匹配正则
            if (expression.test(chunk)) {
                // 将匹配的chunk推送到输出流
                this.push(chunk);
            }
            // 通知流当前chunk处理完成
            callback();
        }
    });

    // 将输入流导入Transform流
    is.pipe(transformStream);

    // 返回处理后的输出流
    return transformStream;
}

代码说明 #

• Transform流的选择：Transform流是Duplex流的子类，专门用于数据转换/过滤场景，天然支持"输入-处理-输出"的流程，比手动监听data事件更规范、更符合Node.js流的设计模式。
• _transform方法：这是Transform流的核心处理逻辑，每个输入的chunk都会进入这个方法。我们在这里用正则测试chunk，匹配的就通过this.push()推送到输出流。
• 对象模式设置：因为输入流是用Readable.from()创建的字符串数组流，开启objectMode可以让流直接处理字符串对象，不需要额外的编码转换。

修正后的测试代码 #

注意原测试代码里的filterStream应改为processStream，consolo.log是笔误，另外需要实现streamToArray来收集流的结果：

const stream = require('stream');
const { promisify } = require('util');
const concat = promisify(require('concat-stream'));

// 实现streamToArray：将流转换为数组的Promise
async function streamToArray(readableStream) {
    const result = await concat({ encoding: 'object' }, readableStream);
    return result;
}

// 测试逻辑
const is = stream.Readable.from(['aaa', 'aab']).setEncoding('utf-8');
const os = processStream(is, /aaa/i);

streamToArray(os).then(console.log, console.error); // 输出: [ 'aaa' ]

关键注意事项 #

• 如果你的输入流是二进制流（非字符串），需要先将chunk转换为字符串再进行正则匹配，比如chunk.toString(encoding)。
• 若正则需要匹配跨chunk的内容（比如一个匹配项被分割在两个chunk里），当前实现无法处理这种场景。如果需要支持这种情况，你需要在Transform流中维护一个缓冲区，累积未完成的匹配内容，直到能完整匹配或流结束。

内容的提问来源于stack exchange，提问作者Learner