ChatGPT与PlantUML结合：自动化生成架构图的高效工作流

1. 项目概述：当ChatGPT遇上架构图

作为一名在软件开发和架构设计领域摸爬滚打了十多年的老兵，我深知画架构图这件事有多“磨人”。从早期在白板上手绘，到后来用Visio、Draw.io，再到尝试PlantUML、Mermaid这类文本绘图工具，工具在进化，但核心痛点始终没变：构思和绘制是两件高度耦合但又相互消耗精力的事。你脑子里可能已经有了清晰的模块划分和数据流向，但要把它们精准、美观地落到图上，往往需要花费大量时间在调整框线、对齐元素、选择图标上。更别提当架构需要调整时，牵一发而动全身的修改有多痛苦。

直到我开始深度使用ChatGPT，一个想法逐渐成型：能不能让这个强大的语言模型来分担“绘制”这部分的工作，甚至辅助“构思”？答案是肯定的。这个项目，就是关于如何利用ChatGPT，结合PlantUML这类文本化绘图语言，实现架构图的自动化、半自动化生成。这不仅仅是“让AI画图”，而是一套全新的、高效的架构设计与表达工作流。无论你是需要快速绘制系统架构图、业务流程图，还是生成UML类图，这套方法都能让你从繁琐的绘图操作中解放出来，将精力真正聚焦在架构设计本身。

2. 核心思路与工具选型：为什么是“ChatGPT + PlantUML”？

在开始动手之前，我们需要明确核心思路和选择最合适的工具。我们的目标不是让ChatGPT直接输出一张PNG或SVG图片（目前它做不到），而是让它生成一种中间描述语言，我们再通过专门的工具将这种语言渲染成图。这个思路有几个关键优势：首先，描述语言是文本，ChatGPT极其擅长理解和生成文本；其次，文本易于修改、版本控制和复用；最后，成熟的文本绘图工具能保证输出图形的规范性和美观度。

2.1 主流文本绘图工具对比

市面上主流的文本绘图工具主要有PlantUML、Mermaid和Graphviz DOT。我们来快速对比一下，看看为什么PlantUML是当前阶段与ChatGPT配合的最佳选择。

注意：Mermaid近年来发展很快，尤其在网页集成和实时预览方面体验很棒。但对于以“架构设计”为核心，需要频繁输出标准UML图的场景，PlantUML在专业性和语法友好度上依然有明显优势。

2.2 工作流设计：人与AI的协同

确定了核心工具，我们的工作流就清晰了。这不是一个完全自动化的“黑箱”，而是一个增强智能的循环：

1. 需求输入：你用自然语言向ChatGPT描述你的架构想法。例如：“帮我设计一个简单的电商微服务系统，包含用户服务、商品服务、订单服务和支付服务，它们通过一个API网关对外暴露，并共用同一个认证中心和MySQL数据库。”
2. AI生成初稿：ChatGPT理解你的需求，生成对应的PlantUML代码（例如一张组件图）。
3. 渲染与审查：你将这段代码粘贴到PlantUML在线编辑器或本地工具中，渲染出图片，查看效果。
4. 反馈与迭代：如果对布局、元素命名或关系不满意，你可以直接告诉ChatGPT你的修改意见。例如：“把数据库放在所有服务的下方，用虚线连接，并给订单服务到支付服务的箭头加上‘调用支付接口’的标签。” ChatGPT会修改代码，你再次渲染。
5. 定稿与导出：经过几轮快速迭代，得到满意的架构图，导出为PNG、SVG等格式。

这个工作流的核心价值在于，你将最耗时的“从想法到规范图形代码”的转换工作交给了AI，而你则扮演架构师和审核者的角色，专注于高层设计和细节修正。

3. 实战：从零生成你的第一张架构图

理论说再多不如动手一试。我们以生成一个“在线视频转码平台”的简化系统架构图为例，完整走一遍流程。我假设你使用的是ChatGPT（GPT-4模型效果更佳），并有一个PlantUML的渲染环境（推荐使用官方的在线服务器：https://www.plantuml.com/plantuml/uml/）。

3.1 第一步：提出明确的绘图需求

与AI合作，清晰的指令（Prompt）是成功的一半。不要只说“画一个架构图”。一个好的Prompt应包含：

• 图形类型：明确告诉AI你要什么图。是组件图？部署图？还是时序图？
• 核心元素：列出系统的主要组成部分，如服务、数据库、队列、网关等。
• 元素关系：描述这些组件之间如何交互，数据流向如何。
• 风格或约束：如果有特殊要求，比如颜色、分组、备注等，一并提出。

我的初始Prompt：

请为我生成一份PlantUML代码，绘制一个在线视频转码平台的系统组件图。 核心组件包括： 1. 客户端（Web或App） 2. 负载均衡器（Nginx） 3. API网关（Spring Cloud Gateway） 4. 视频上传服务 5. 转码任务管理服务 6. 分布式转码集群（多个转码Worker） 7. 消息队列（RabbitMQ，用于传递转码任务） 8. 对象存储（MinIO，用于存储原始视频和转码后视频） 9. 元数据库（MySQL，存储用户、视频元数据、任务状态） 10. 缓存（Redis，存储热点数据和任务锁） 关键关系： - 客户端请求经过负载均衡器到达API网关。 - 上传服务接收视频后，存入对象存储，并向任务管理服务创建任务。 - 任务管理服务将任务信息放入消息队列。 - 转码Worker从消息队列消费任务，从对象存储读取原始视频，转码后再写回对象存储，并通知任务管理服务更新状态。 - 所有服务都可能需要访问数据库和缓存。 请使用合适的PlantUML组件图语法，布局尽量清晰，可以适当使用颜色区分不同层次的组件（如接入层、业务层、基础设施层）。最后，请只输出PlantUML代码，不要有其他解释。

3.2 第二步：处理AI的初次输出

ChatGPT会返回一段PlantUML代码。第一次生成的结果通常可用，但可能布局拥挤、颜色混乱或关系不精准。这是完全正常的，也是我们工作流中“迭代”环节的价值所在。

首次生成的代码可能类似这样（已简化）：

@startuml !define LAYER_COLOR #DDDDDD skinparam component { BackgroundColor LAYER_COLOR BorderColor Black } package "接入层" { [客户端] as Client [负载均衡器 Nginx] as LB [API网关] as Gateway } package "业务服务层" { [视频上传服务] as UploadService [转码任务管理服务] as TaskService [转码Worker 1] as Worker1 [转码Worker 2] as Worker2 [转码Worker N] as WorkerN } package "基础设施层" { [消息队列 RabbitMQ] as MQ [对象存储 MinIO] as Storage [元数据库 MySQL] as DB [缓存 Redis] as Cache } Client --> LB : HTTP请求 LB --> Gateway : 路由 Gateway --> UploadService : 上传API Gateway --> TaskService : 查询API UploadService --> Storage : 存储原始视频 UploadService --> TaskService : 创建任务 TaskService --> MQ : 发布任务 Worker1 --> MQ : 消费任务 Worker2 --> MQ : 消费任务 WorkerN --> MQ : 消费任务 Worker1 --> Storage : 读取/写入视频 Worker2 --> Storage : 读取/写入视频 WorkerN --> Storage : 读取/写入视频 TaskService --> DB : 读写元数据 UploadService --> DB : 读写元数据 Worker1 --> TaskService : 更新状态 Worker2 --> TaskService : 更新状态 WorkerN --> TaskService : 更新状态 DB --> Cache : 缓存同步（可选） @enduml

将这段代码复制到PlantUML在线编辑器，你会得到一张图。你可能会立刻发现几个问题：所有转码Worker与MQ和Storage的连接线交叉严重，图显得很乱；“基础设施层”的组件与其他层的连接线也纵横交错。

3.3 第三步：进行迭代优化

现在，发挥你作为架构师的判断力，给ChatGPT具体的修改指令。

我的优化Prompt：

上面生成的图逻辑正确，但布局太乱，连线交叉太多。请优化这段PlantUML代码，要求： 1. 使用 `left to right direction` 指令改为从左到右的布局，这样更符合数据流从左（客户端）到右（存储）的直觉。 2. 将“转码集群”抽象为一个整体组件 `[转码集群]`，内部可以用`()`表示包含多个Worker，避免画出一大堆与MQ和Storage重复的连线。只需展示集群与MQ和Storage的关系。 3. 将“基础设施层”的组件（MQ, Storage, DB, Cache）放在最右侧，作为共享资源。 4. 调整连线，尽量减少交叉。可以尝试使用隐藏节点和间接连线（如 `A -> B` 和 `B -> C`，而不是 `A -> C`）来梳理流程。 5. 为不同层的包（package）设置不同的浅背景色以示区分。 请输出优化后的完整代码。

3.4 第四步：获得最终成果

ChatGPT会根据你的反馈生成新版代码。经过一两轮这样的调整，你就能得到一份布局清晰、表达准确的架构图代码。

优化后的核心部分示意：

@startuml left to right direction skinparam packageBackgroundColor #F5F5F5 skinparam packageBorderColor #333 package “客户端” #LightBlue { [Web/App] as Client } package “接入层” #LightGreen { [负载均衡器\nNginx] as LB [API网关] as Gateway } package “业务服务层” #LightYellow { [视频上传服务] as UploadService [转码任务管理服务] as TaskManager component “转码集群” as TranscodeCluster { () “Worker 1” () “Worker 2” () “Worker …” } } package “基础设施层” #LightGray { [消息队列\nRabbitMQ] as MQ [对象存储\nMinIO] as Storage database “元数据库\nMySQL” as DB [缓存\nRedis] as Cache } ‘ 连接关系 Client -> LB : “HTTP/HTTPS” LB -> Gateway Gateway -> UploadService : “/api/upload” Gateway -> TaskManager : “/api/query” UploadService --> Storage : “上传原始视频” UploadService -> TaskManager : “创建转码任务” TaskManager -> MQ : “发布任务消息” TranscodeCluster -> MQ : “消费任务” TranscodeCluster --> Storage : “读取原始视频\n写入转码后视频” TranscodeCluster -> TaskManager : “通知任务完成” TaskManager --> DB : “更新任务状态” UploadService --> DB : “记录元数据” DB <--> Cache : “缓存热点数据” ‘ 使用隐藏节点整理连线，避免交叉 note right of MQ 任务消息驱动 解耦服务与集群 end note @enduml

渲染出的图形结构会清晰很多：从左到右的流向明确，集群被抽象，基础设施集中右侧，连线经过规划后交叉显著减少。整个过程，你只需要用自然语言沟通，无需手动调整一行布局代码。

4. 进阶技巧与场景应用

掌握了基本流程后，我们可以探索更多高阶用法，让这个组合发挥更大威力。

4.1 生成UML类图（Class Diagram）

这是PlantUML的强项，也是ChatGPT能很好完成的。你可以直接粘贴一段代码（或描述类结构）让ChatGPT生成类图。

Prompt示例：

请根据以下Java类的简单描述，生成PlantUML类图代码。 - 有一个抽象类 `BaseEntity`，包含 `Long id` 属性和 `Date createTime`、`Date updateTime` 属性，以及 `save()` 抽象方法。 - `User` 类继承 `BaseEntity`，增加 `String username`、`String email` 属性，并实现 `save()` 方法。 - `Order` 类继承 `BaseEntity`，增加 `BigDecimal amount`、`OrderStatus status` 枚举属性。它与 `User` 是多对一关系（一个用户有多个订单）。 - `OrderItem` 类，与 `Order` 是一对多关系（一个订单有多个订单项），包含 `String productName`、`Integer quantity` 属性。 - 有一个接口 `Auditable`，包含 `getAuditLog()` 方法。`User` 类实现这个接口。 - 注意使用合适的可见性符号（+ for public, - for private, # for protected）和关系箭头（继承、实现、关联）。

ChatGPT生成的PlantUML代码会清晰地展示出继承树（..|>）、实现关系（..|>）、关联（-->）和聚合/组合（o--，*--），你只需渲染即可得到规范的类图。

4.2 绘制时序图（Sequence Diagram）

时序图非常适合描述单个业务流程或API调用链。用自然语言描述交互流程即可。

Prompt示例：

生成一个PlantUML时序图，描述用户通过客户端上传视频的简化流程： 1. 用户从客户端点击上传。 2. 客户端调用【上传服务】的预签名接口，获取一个可直传至对象存储的临时URL。 3. 客户端直接使用该URL将视频文件上传至【对象存储】。 4. 上传成功后，【对象存储】回调【上传服务】的通知接口。 5. 【上传服务】收到回调后，在【数据库】中创建一条视频记录，状态为“待转码”。 6. 【上传服务】向【消息队列】发送一个“新视频待转码”事件。 7. 【转码服务】监听队列，消费该事件。 请为每个参与者（Client, UploadService, ObjectStorage, DB, MQ, TranscodeService）和关键步骤添加说明。

4.3 利用上下文和知识库生成更专业的图

如果你的架构遵循某种特定范式（如Clean Architecture、DDD分层），或者使用了特定的云服务（AWS、Azure图标），你可以提前“教会”ChatGPT。

方法：在对话开始，先发送一段包含专业约定的文本。

在接下来的对话中，当我们讨论系统架构时，请遵循以下约定： 1. 使用C4模型中的“容器图”视角来描述系统。主要元素是“容器”（可独立部署/运行的应用、数据存储等）。 2. 使用AWS的图标集。例如，数据库用 `:Amazon RDS:`，对象存储用 `:Amazon S3:`，消息队列用 `:Amazon SQS:`。 3. 在PlantUML代码开头使用 `!includeurl https://raw.githubusercontent.com/awslabs/aws-icons-for-plantuml/v14.0/dist` 来引入AWS图标。 现在，请为一个使用AWS服务的内容推荐平台生成容器图，包括API Gateway, Lambda函数, S3, DynamoDB, SQS和Personalize服务。

通过这种方式，ChatGPT能生成更贴近行业规范、更具专业感的架构图。

5. 避坑指南与实操心得

在实际使用中，我积累了一些宝贵的经验和需要避开的“坑”。

5.1 常见问题与解决方案

5.2 我的核心实操心得

1. Prompt工程是关键：把你当成一个需求清晰的“产品经理”，把ChatGPT当成一个能力强大但需要精确指引的“工程师”。指令越具体、越结构化，产出质量越高。多使用“首先…然后…最后…”的句式来分解任务。
2. 迭代优于一次成型：不要指望第一个Prompt就得到完美结果。把生成-审查-修正的循环看作正常流程。每次修正指令要具体（“连线交叉”不如“请将组件A移到组件B的左边以减少连线交叉”）。
3. 积累自己的“模板库”：将多次调试后满意的、用于特定场景（如微服务组件图、数据库ER图、API时序图）的PlantUML代码片段保存下来。下次可以直接将模板喂给ChatGPT：“请参考以下代码风格和结构，为XX系统生成一个类似的图。”
4. 结合图形界面工具：对于最终定稿的、需要精美排版的架构图（如用于正式文档、PPT），可以将PlantUML生成的图导出为SVG，再导入Draw.io或Excalidraw中进行微调和美化。这样既利用了AI的快速构思能力，又保留了人工对最终美学的把控。
5. 理解AI的局限性：ChatGPT本质上是在“模仿”它训练数据中见过的PlantUML模式。对于极其复杂、非典型的架构布局，它可能力不从心。此时，你需要具备基本的PlantUML语法知识，对AI生成的代码进行手动调整。这正体现了“增强智能”的意义——人负责创造和决策，AI负责执行和优化。

这套“ChatGPT + PlantUML”的方法，彻底改变了我绘制技术图表的方式。它最大的价值不是百分百的自动化，而是将我从机械的绘图劳动中解放出来，让我能更专注地思考架构本身。当你需要快速绘制草图进行沟通，或者需要维护一套与代码同步更新的架构文档时，试试这个方法，你会感受到那种“动动嘴皮子就把图搞定”的效率飞跃。