一、前言
在大模型RAG检索、智能Agent、知识库问答等AI应用架构中,结构化业务数据与非结构化向量数据的分层存储已成为工业界主流设计方案。多数中大型项目会采用「PostgreSQL存储结构化业务数据 + Milvus存储向量特征数据」的双库架构,而非直接使用PostgreSQL的pgvector插件实现单库存储。
该架构的核心优势在于业务读写负载与向量检索负载的资源隔离 、海量向量数据的高性能检索与水平扩容能力。但双库独立部署的设计,彻底打破了单机数据库的ACID事务边界,衍生出跨库双写数据一致性 核心问题。同时,该架构的技术选型、同步策略、异常兜底逻辑,本质都是分布式CAP理论在AI工程落地中的具体取舍。
本文将从架构选型逻辑、双写一致性痛点、工程落地解决方案、CAP理论底层约束四个维度,系统性解析该分层存储架构的核心原理与最佳实践。
二、架构选型:双库分离 vs 单库向量存储
2.1 两种存储方案核心对比
目前AI应用向量存储存在两种主流方案,二者的适用场景与性能短板存在本质差异:
方案一:PostgreSQL + pgvector(单库一体化存储)
通过pgvector插件为PostgreSQL赋予向量存储与检索能力,将结构化字段(标题、作者、权限、状态)与向量Embedding存储在同一张数据表中,天然继承PostgreSQL完整的ACID事务特性,可实现结构化数据与向量数据的强一致性,无需额外同步逻辑,部署与运维成本极低。
但其存在无法规避的性能瓶颈:仅适配十万至小百万级向量数据,面对千万、亿级海量向量时,存在索引构建缓慢、检索延迟飙升、内存占用过高、无法分布式分片扩容等问题;同时,高负载向量检索会占用数据库核心资源,拖累用户会话、权限管理、业务审核等核心结构化读写接口,引发业务超时风险。
方案二:PostgreSQL + Milvus(双库分层存储)
严格区分数据职责边界:PostgreSQL作为唯一主数据源 ,承载所有结构化业务数据、业务状态、权限体系、日志记录,保障核心业务的稳定读写;Milvus作为专用二级向量索引,仅存储文档ID与对应向量Embedding,专注提供高性能向量相似度检索服务。
Milvus是云原生专用向量数据库,支持向量分片存储、冷热数据分层、增量索引构建、多租户隔离,可支撑亿级向量的低延迟检索,完美解决pgvector的海量数据性能瓶颈。同时实现负载完全隔离,向量检索的高CPU、内存消耗不会影响核心业务数据库,极大提升系统整体稳定性。
2.2 双库分离架构的适用场景
该架构并非通用最优解,而是中大型AI应用的针对性取舍,核心适用场景:
- 知识库向量规模超200万,存在持续扩容需求;
- 业务核心链路可用性优先级高于瞬时数据一致性;
- 需要隔离向量检索与业务读写负载,保障核心接口稳定性;
- 存在高频向量更新、重建、批量检索的业务场景。
三、双库架构的核心痛点:跨库双写一致性缺失
PostgreSQL与Milvus属于异构分布式存储组件,二者无统一的分布式事务协议,无法实现跨库ACID强事务保障。
在数据写入、更新、删除流程中,必须执行双写操作:结构化数据写入PostgreSQL,向量数据写入Milvus。由于网络波动、服务重启、接口超时等异常,会出现三种数据不一致问题:
- PG写入成功、Milvus写入失败:业务数据存在,但无对应向量,导致文档无法被向量检索召回;
- Milvus写入成功、PG写入失败:向量数据孤立存在,无对应业务结构化信息,产生「孤儿向量」,检索返回无效脏数据;
- 更新/删除不同步:PG数据更新或软删除后,Milvus向量未同步变更,出现新旧数据重叠、无效向量残留问题。
简言之,单库pgvector的强一致性 是天然特性,而双库分离架构的数据不一致风险是架构固有缺陷,必须通过工程方案主动治理。
四、双写一致性工程落地解决方案
行业核心设计原则:以PostgreSQL为绝对主数据源,Milvus为次级索引,所有同步操作围绕主库数据兜底,优先保障主库业务数据可靠,再通过多级机制实现向量数据最终一致。
4.1 基础规范:固定双写顺序
严禁「先写Milvus、后写PG」,统一遵循先落库PG,后同步Milvus的黄金顺序:
- 开启PG事务,完成结构化数据新增/更新,生成唯一文档ID并提交事务;
- 基于文档ID生成向量Embedding,同步写入Milvus;
- PG数据表新增向量同步状态字段,标记
pending_vector/vector_ok/vector_failed,记录同步结果。
该规范可彻底规避「孤儿向量」问题,确保所有向量数据必然依赖有效业务数据。
4.2 核心方案:本地消息表异步同步(行业主流)
针对中小中型Agent、RAG项目,PG本地消息表是性价比最高、最稳定的最终一致性方案,无需引入MQ中间件,依托主库事务实现原子性保障。
实现流程
- 在PG中新建
vector_sync_task同步任务表,记录文档ID、操作类型(新增/更新/删除)、重试次数、任务状态、重试时间; - 在同一个PG事务中,完成「业务数据写入 + 同步任务写入」,保证二者原子成功或失败;
- 独立后台消费者线程定时轮询待处理任务,读取任务信息生成向量并写入Milvus;
- 同步成功则更新任务状态为完成,同步失败则记录失败状态、累加重试次数,等待下次重试。
任务表核心字段设计
sql
CREATE TABLE vector_sync_task (
id bigserial primary key,
doc_id bigint not null,
op_type varchar(20),
retry_count int default 0,
status varchar(20),
create_time timestamp,
last_retry_time timestamp
);
4.3 高并发方案:MQ异步同步(大型项目)
针对高并发、大数据量场景,采用「PG主库写入 + Kafka/RabbitMQ消息投递 + 消费者同步Milvus」架构:
- PG事务完成结构化数据写入并提交;
- 投递可靠同步消息至MQ,携带文档核心信息;
- 消费者消费消息、写入Milvus,成功后手动ACK,失败则消息重试;
- 搭配PG状态字段兜底,防止消息丢失导致的数据不一致。
4.4 终极兜底:定时巡检补偿机制
所有同步方案均无法100%规避极端异常,定时巡检是必备兜底策略:
- 正向巡检:定时扫描PG中
vector_sync_status != vector_ok的异常数据,批量重新生成向量、同步至Milvus; - 反向巡检(低峰期执行):遍历Milvus向量ID集合,比对PG有效数据,删除无对应业务数据的残留孤儿向量。
4.5 删除场景一致性特殊处理
为避免脏数据残留,业务层禁止硬删除PG数据,统一采用软删除:
- PG更新数据
is_deleted=true软删除状态; - 生成删除同步任务,后台消费者异步删除Milvus对应向量;
- 凌晨低峰期批量清理长期软删除的PG历史数据,完成数据归档。
五、CAP理论视角下的架构底层取舍
5.1 CAP理论核心公理
分布式系统三大核心特性一致性©、可用性(A)、分区容错性§ 无法同时满足,仅可三选二:
- 一致性C:所有节点同一时刻读取的数据完全一致;
- 可用性A:系统任意请求均可正常响应,无阻塞、无超时;
- 分区容错性P:网络分区、节点通信中断时,系统仍可正常运行。
在分布式架构中,网络分区是不可规避的客观问题,因此P是分布式系统的必备特性,所有架构只能在「CP强一致」与「AP高可用」之间二选一。
5.2 两种存储架构的CAP选型对照
-
PostgreSQL+pgvector 单库架构(CP模型)
PostgreSQL原生为CP架构,优先保障强一致性与分区容错性。网络分区或写入异常时,系统会牺牲可用性,通过事务锁、阻塞等待避免数据不一致。该模型适配对数据一致性要求极高、吞吐量较低的场景。
-
PostgreSQL+Milvus 双库架构(AP模型)
双库分层架构是典型的AP高可用模型,也是AI应用的最优取舍:
- 优先保障可用性A:网络异常、Milvus服务故障、同步失败时,核心PG业务读写链路完全不受影响,系统持续对外提供服务;
- 保留分区容错性P:双服务独立部署,单组件故障、网络分区不会导致整体系统瘫痪;
- 牺牲瞬时强一致性C:放弃跨库实时一致,接受短时间数据差异,通过异步同步、巡检兜底实现最终一致性。
5.3 架构取舍的工程本质
AI Agent、RAG系统的核心业务诉求是持续可用、低延迟检索、支撑海量数据,而非毫秒级跨库强一致。用户侧无法感知短暂的数据同步延迟,但可以直接感知系统宕机、接口超时、检索无结果等可用性问题。
因此,该架构的所有一致性方案(异步同步、重试机制、巡检兜底),本质都是以极小的瞬时数据不一致代价,换取系统极致的高可用与可扩展性,完全贴合CAP理论的分布式架构设计逻辑。
六、架构选型与一致性方案总结
- 小规模场景(向量<200万):优先选用PostgreSQL+pgvector单库架构,依托原生ACID实现强一致性,降低运维与开发成本;
- 中大规模场景(向量超200万、高并发):采用PostgreSQL+Milvus双库分层架构,负载隔离、支持海量扩容;
- 一致性最优方案:中小项目使用「本地消息表异步同步 + 定时巡检兜底」,大型高并发项目使用「MQ异步同步 + 双向巡检补偿」;
- 底层逻辑:双库架构是典型AP分布式模型,放弃瞬时强一致、保障高可用,通过工程手段实现业务可接受的最终一致性。
该套架构是当前工业界AI应用落地的标准范式,深刻体现了分布式系统「没有完美架构,只有合理取舍」的核心设计思想。
需要我帮你精简成面试速记版,方便快速背诵核心要点吗?