使用@kubernetes/client-node时，如何将Stream.Writable数据转为字符串变量

要直接收集@kubernetes/client-node的exec方法输出到变量，你可以自定义一个Writable流子类，在流的写入过程中把数据追加到变量里，无需写入文件。

完整实现代码 #

const { Writable } = require('stream');
const k8s = require('@kubernetes/client-node');

// 自定义Writable流，用于收集命令输出
class OutputCollector extends Writable {
  constructor(options) {
    super(options);
    this.output = ''; // 存储输出的字符串变量
  }

  // 重写_write方法，处理每个数据块
  _write(chunk, encoding, callback) {
    this.output += chunk.toString(encoding);
    callback(); // 通知流已处理完当前数据块
  }
}

async function runPodCommand() {
  // 初始化K8s客户端配置
  const kc = new k8s.KubeConfig();
  kc.loadFromDefault(); // 加载默认配置（~/.kube/config）

  const exec = new k8s.Exec(kc);
  const outputStream = new OutputCollector();
  const commandStream = new k8s.Readable();

  try {
    // 建立exec连接
    const podExec = await exec.exec(
      "myNamespace",
      "myPod",
      "myContainer",
      ["/bin/sh", "-c"],
      outputStream, // 收集stdout
      outputStream, // 收集stderr（如果不需要可以传null）
      commandStream,
      true
    );

    // 发送要执行的命令，末尾加换行符表示输入结束
    commandStream.push("ls -l\n");
    // 关闭输入流，告诉容器命令已全部发送
    commandStream.push(null);

    // 等待输出流完成+exec连接关闭，确保所有输出都被收集
    await new Promise(resolve => {
      outputStream.on('finish', resolve);
      podExec.on('close', resolve);
    });

    // 此时outputStream.output就是完整的命令输出字符串
    const commandOutput = outputStream.output;
    console.log('命令执行结果:\n', commandOutput);
    // 这里可以对commandOutput做后续处理

  } catch (error) {
    console.error('执行命令失败:', error);
  } finally {
    // 清理资源
    outputStream.destroy();
    commandStream.destroy();
  }
}

// 执行函数
runPodCommand();

关键说明 #

1. 自定义OutputCollector流：
   • 继承Writable类，重写_write方法，每次写入数据块时，将其转为字符串追加到output属性中。
2. 正确结束输入流：
   • 发送命令后必须调用commandStream.push(null)，否则容器会一直等待输入，无法结束命令执行。
3. 等待输出完成：
   • 通过Promise监听outputStream的finish事件和podExec的close事件，确保所有输出数据都被收集完毕后再读取结果。
4. 分开处理stdout和stderr：
   • 如果需要分别收集标准输出和错误输出，可以创建两个OutputCollector实例，分别传入exec方法的第5、6个参数。

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