Pyodide标准输入(stdin)正确处理方案技术咨询

我是Pyodide新手，正在将其应用到开源教育软件TALightDesktop中。已经参照loadPyodide文档配置了stdin回调，但stdin: ()=>{return this.onStdin()}的运行机制要求输入结果必须在stdin请求生成前准备就绪，尝试用Promise但Pyodide不支持这种方式。现在想知道怎么实现暂停Pyodide执行、提示用户输入、等待回复后恢复执行的流程，怀疑轮询是不是唯一的办法。

以下是当前的实现代码：

class PyodideWorker{
  let pyodide:any
  let stdinBuffer:string[]
  ...
  async initPydiode(){
    let options = {
      stdin: ()=>{return this.onStdin()},
      stdout: (msg:string)=>{this.onStdout(msg)},
      stderr: (msg:string)=>{this.onStderr(msg)},
    }
    this.pyodide = await loadPyodide(options);
  }
  ...
  onStdin(){
    let cnt = this.stdinBuffer.length
    let msg = "";
    if(cnt > 0){
      let items = this.stdinBuffer.splice(0,cnt)
      msg = items.join("")
    }
    return msg
  }
  ...
  sendStdin(request:PyodideRequest){
    let response = this.responseFromRequest(request); 
    let data = request.message.contents[0];
    console.log("sendStdin:\n",data)//,res)
    if (this.stdinResolver){
      this.stdinResolver(this.toString(data))
    }else{
      this.stdinBuffer.push(this.toString(data))
    }
    response.message.args = ['true']
    return response;
  }
}

Pyodide的stdin回调确实不支持异步返回Promise，但可以通过基于共享内存的同步等待实现交互式输入，比轮询更高效，核心是用Atomics.wait和Atomics.notify实现线程间同步。

具体实现步骤 #

1. 新增同步状态变量：在PyodideWorker中添加共享内存信号量和输入存储变量，替代原有的stdinBuffer。
2. 修改stdin回调逻辑：当需要输入时，阻塞等待用户输入信号，直到外部触发唤醒。
3. 调整sendStdin方法：接收用户输入后，触发信号唤醒等待的Pyodide线程。

修改后的关键代码示例 #

class PyodideWorker{
  let pyodide:any
  let stdinInput: string = "";
  // 创建共享内存信号量，用于同步等待
  let stdinReady = new Int32Array(new SharedArrayBuffer(4));
  
  async initPydiode(){
    let options = {
      stdin: ()=>{
        // 等待输入信号，0表示持续等待直到被唤醒
        Atomics.wait(this.stdinReady, 0, 0);
        // 重置信号量
        Atomics.store(this.stdinReady, 0, 0);
        // 返回用户输入（添加换行符匹配Python input()的预期）
        const input = this.stdinInput + "\n";
        this.stdinInput = "";
        return input;
      },
      stdout: (msg:string)=>{this.onStdout(msg)},
      stderr: (msg:string)=>{this.onStderr(msg)},
    }
    this.pyodide = await loadPyodide(options);
  }
  
  sendStdin(request:PyodideRequest){
    let response = this.responseFromRequest(request); 
    let data = request.message.contents[0];
    this.stdinInput = this.toString(data);
    // 发送信号唤醒等待的stdin回调
    Atomics.store(this.stdinReady, 0, 1);
    Atomics.notify(this.stdinReady, 0);
    
    response.message.args = ['true']
    return response;
  }
}

注意事项 #

• 确保SharedArrayBuffer可用：需要在服务器配置跨域隔离（COOP/COEP）响应头，这在教育类Web应用中通常可实现。
• 可添加超时逻辑：如果担心用户长时间不输入导致线程挂起，可给Atomics.wait设置超时参数。
• 无需轮询：这种方式是真正的同步等待，不会占用多余CPU资源。

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