基于Kubernetes的Linux Playground终端状态保持问题求助

核心问题根源 #

当前实现每次执行命令时，都会在Kubernetes Pod中启动新的独立进程，没有维护持续的交互式Shell会话。像cd这类修改Shell环境状态的命令，仅对当前进程有效，后续命令启动新进程后会回到初始环境，导致状态丢失。

具体解决方案 #

1. 为每个用户维护持久的交互式Shell会话 #

每个用户建立WebSocket连接时，在其专属Pod中启动一个长期运行的bash/zsh进程，全程复用该进程处理所有命令输入：

• 启动Shell进程时，配置stdin=subprocess.PIPE、stdout=subprocess.PIPE、stderr=subprocess.STDOUT，保持进程处于打开状态
• 将Shell进程实例绑定到Channels的scope对象中，确保在整个WebSocket会话周期内可访问
• 用户输入命令时，直接将命令（追加换行符\n）写入Shell进程的标准输入，异步捕获标准输出/错误并返回前端
• WebSocket连接断开时，立即终止Shell进程并清理对应Kubernetes Pod

代码示例（Channels Consumer）

import asyncio
import subprocess
from channels.generic.websocket import AsyncWebsocketConsumer

class TerminalConsumer(AsyncWebsocketConsumer):
    async def connect(self):
        await self.accept()
        # 启动用户专属Pod内的交互式bash会话
        self.shell_proc = await asyncio.create_subprocess_exec(
            "/bin/bash",
            stdin=subprocess.PIPE,
            stdout=subprocess.PIPE,
            stderr=subprocess.STDOUT,
            cwd="/home/playground-user"  # 初始工作目录
        )
        # 启动异步任务持续读取Shell输出
        asyncio.create_task(self._stream_shell_output())

    async def _stream_shell_output(self):
        while self.shell_proc.returncode is None:
            output_chunk = await self.shell_proc.stdout.read(1024)
            if output_chunk:
                await self.send(text_data=output_chunk.decode("utf-8", errors="replace"))
            await asyncio.sleep(0.05)

    async def receive(self, text_data):
        # 将用户命令写入Shell标准输入
        command = f"{text_data.strip()}\n"
        self.shell_proc.stdin.write(command.encode("utf-8"))
        await self.shell_proc.stdin.drain()

    async def disconnect(self, close_code):
        # 终止Shell进程并清理Pod
        self.shell_proc.terminate()
        await self.shell_proc.wait()
        # 此处添加删除对应Kubernetes Pod的逻辑（调用K8s API或使用客户端库）

2. 引入PTY（伪终端）增强终端兼容性 #

如果需要支持终端颜色、命令补全、信号处理（如Ctrl+C）等高级特性，可使用PTY模拟真实终端环境：

• 使用Python标准库pty或第三方库pexpect创建伪终端，绑定到Shell进程
• 前端终端组件（如xterm.js）需传递终端尺寸参数，后端同步更新PTY窗口大小，确保输出格式正确
• PTY会自动维护完整的Shell会话状态，包括当前工作目录、环境变量等

3. Kubernetes侧的会话保障优化 #

• 资源隔离：为每个用户Pod设置CPU/内存请求和限制（resources.requests/resources.limits），防止单个用户占用过多集群资源
• 安全加固：通过securityContext配置Pod使用非root用户运行，禁用特权模式，限制容器的系统调用权限
• 自动清理：WebSocket连接超时或断开后，通过Kubernetes API立即删除对应Pod；可配合TTL控制器自动清理闲置Pod
• 会话稳定性：为用户Pod设置PodDisruptionBudget，避免集群维护时意外终止正在运行的会话

生产级额外考量 #

• 日志审计：记录用户的命令执行日志和会话操作，用于合规审计和故障排查
• 连接重试：前端实现重连逻辑，后端支持会话恢复（需存储会话状态到Redis等缓存）
• 限流：限制单个用户的并发会话数和命令执行频率，防止恶意攻击

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