应校验补全GraphQL订阅定义、注入WebSocket上下文绑定、修复前端自动重连缺陷、强制启用GraphQL over WebSocket协议。具体包括：一、补全schema中subscription声明及类型一致性；二、在WebSocket连接中绑定flowId并精准广播；三、增强useSubscription的断线重试逻辑；四、后端启用graphql-ws中间件，前端改用WebSocketLink。
☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 多模态理解力帮你轻松跨越从0到1的创作门槛☜☜☜
如果您在使用CodeBuddy开发涉及GraphQL Subscription和WebSocket的实时功能时遇到生成代码不完整、类型不匹配或连接生命周期管理缺失等问题，则可能是由于AI对订阅上下文建模不足或对底层协议细节理解偏差所致。以下是解决此问题的步骤：
一、校验并补全GraphQL订阅模式定义
CodeBuddy可能生成基础schema但遗漏subscription类型声明或参数校验逻辑，需人工确认schema.graphqls中是否正确定义了taskUpdated、flowUpdated等订阅字段及其返回类型，并确保其与resolver签名一致。
1、打开backend/graph/schema.graphqls文件，定位到subscription块。
2、检查是否存在类似subscription { taskUpdated(flowId: ID!): Task! }的声明，且flowId参数为非空ID类型。
3、确认返回类型Task已在type定义中声明，且包含id、status、updatedAt等必要字段。
4、若缺失，手动补充完整subscription定义，并在CodeBuddy中以指令“根据现有Task类型，补全taskUpdated和flowUpdated订阅声明”重新触发生成。
二、注入WebSocket连接上下文绑定逻辑
CodeBuddy默认生成的WebSocket初始化代码常忽略客户端身份与流程ID的绑定机制，导致广播时无法精准路由至目标订阅者，必须显式将conn与flowId关联并注册进manager。
1、在backend/websocket/websocket.go中，于Upgrade后立即调用conn.SetAttribute("flowId", flowId)保存上下文。
2、修改subscriptions/manager.go中的Register方法，从conn.Attribute("flowId")提取值，并将其存入flowSubscriptions映射表。
3、在broadcast.go的FlowUpdated广播分支中，仅向flowSubscriptions[flowId]中的所有conn推送消息，避免全局广播造成数据泄露或性能浪费。
CodeBuddy Code CLI 的安装、配置与使用指南。CodeBuddy Code 是腾讯推出的 AI 驱动 CLI 编程助手，支持自然语言驱动开发。 - 必备触发词：CodeBuddy, codebuddy, AI CLI, Tencent AI coding, @tencent-ai/codebuddy-code, terminal AI assistant - 适用场景：安装 CodeBuddy CLI、配置 CodeBuddy、使用 CodeBuddy 命令、排查 CodeBuddy 问题
三、修复前端GraphQL订阅查询的自动重连缺陷
CodeBuddy生成的前端订阅代码通常未集成断线重试策略，导致WebSocket关闭后订阅永久失效，需手动增强useSubscription Hook的行为逻辑。
1、在frontend/src/pages/ChatPage/ChatPage.graphql中，确认订阅查询已包含@live或@stream指令（如subscription taskUpdated($flowId: ID!) @live）。
2、在React组件内，使用Apollo Client的onError回调捕获NETWORK_ERROR，并触发client.stop()后延迟3秒调用client.start()。
3、为useSubscription添加retryCount状态变量，限制最大重试次数为5次，超过则显示“实时连接异常，请刷新页面重试”提示。
四、强制启用GraphQL over WebSocket传输协议
CodeBuddy可能默认生成HTTP长轮询式订阅，无法发挥WebSocket低延迟优势，必须显式配置Apollo Server使用subscriptions-transport-ws或graphql-ws中间件。
1、在backend/server.go中，确认已导入github.com/graphql-go/graphql-ws库并初始化wsHandler。
2、将HTTP路由/gql路径替换为wsHandler，确保Upgrade请求被正确拦截并升级为WebSocket连接。
3、前端ApolloClient配置中，将link属性由HttpLink改为WebSocketLink，且uri指向ws://localhost:8080/gql。
4、验证浏览器开发者工具Network标签页中出现ws://协议连接，且状态为101 Switching Protocols。
