Postgraphile中实现时间版本化数据查询的技术方案

背景介绍

在数据库设计中，有时我们需要实现数据的版本控制功能，即记录数据在不同时间点的状态变化。传统的做法是使用简单的自增ID作为主键，但在版本控制场景下，我们需要采用复合主键(ID+版本号)的设计模式。

问题描述

在使用Postgraphile构建GraphQL API时，如何优雅地实现以下功能：

1. 数据表采用(ID, 版本号)的复合主键
2. 提供"时间机器"功能，可以查询任意时间点的数据状态
3. 在GraphQL中保持数据间的关联关系

解决方案

基础表结构设计

首先创建基础表结构，采用复合主键设计：

CREATE TABLE foo(
  id INTEGER NOT NULL, 
  version INTEGER NOT NULL, 
  bar_id INTEGER NOT NULL,
  PRIMARY KEY (id, version)
);

CREATE TABLE bar(
  id INTEGER NOT NULL, 
  version INTEGER NOT NULL, 
  data TEXT,
  PRIMARY KEY (id, version)
);

时间点查询函数

为每个表创建时间点查询函数，使用DISTINCT ON获取指定版本前的最后一条记录：

CREATE FUNCTION foo_at(max_version INTEGER) 
RETURNS SETOF foo AS $$
  SELECT DISTINCT ON (foo.id) * 
  FROM foo 
  WHERE version <= max_version 
  ORDER BY id, version DESC
$$ LANGUAGE SQL STABLE;

CREATE FUNCTION bar_at(max_version INTEGER) 
RETURNS SETOF bar AS $$
  SELECT DISTINCT ON (bar.id) * 
  FROM bar 
  WHERE version <= max_version 
  ORDER BY id, version DESC
$$ LANGUAGE SQL STABLE;

关联关系处理

为了实现跨表的关联查询，我们需要创建复合类型和关联函数：

-- 定义快照类型
CREATE TYPE foo_snapshot AS (row foo, max_version INTEGER);
CREATE TYPE bar_snapshot AS (row bar, max_version INTEGER);

-- 创建带版本号的查询函数
CREATE FUNCTION foo_snapshot_at(max_version INTEGER) 
RETURNS SETOF foo_snapshot AS $$
  SELECT DISTINCT ON (foo.id) foo, max_version 
  FROM foo 
  WHERE version <= max_version 
  ORDER BY id, version DESC
$$ LANGUAGE SQL STABLE;

CREATE FUNCTION bar_snapshot_at(max_version INTEGER) 
RETURNS SETOF bar_snapshot AS $$
  SELECT DISTINCT ON (bar.id) bar, max_version 
  FROM bar 
  WHERE version <= max_version 
  ORDER BY id, version DESC
$$ LANGUAGE SQL STABLE;

-- 创建关联函数
CREATE FUNCTION foo_snapshot_bar(x foo_snapshot) 
RETURNS SETOF bar_snapshot AS $$
  SELECT * 
  FROM bar_snapshot_at(x.max_version) AS y 
  WHERE (y.row).id = (x.row).bar_id 
$$ LANGUAGE SQL STABLE;

技术要点解析

1. DISTINCT ON子句：PostgreSQL特有的语法，配合ORDER BY可以获取每个ID的最新版本记录。

2. 复合类型定义：通过创建包含原始行和版本号的复合类型，可以保持版本上下文在关联查询中的传递。

3. STABLE函数：标记函数为STABLE表示在相同输入下输出不变，允许查询优化器进行缓存。

4. 关联查询设计：通过将max_version作为快照的一部分，确保关联查询使用相同的版本上下文，避免数据不一致。

实际应用场景

这种设计特别适合以下场景：

• 需要查看历史数据的审计系统
• 支持时间旅行的版本控制系统
• 需要撤销/重做功能的应用
• 法律或合规要求的系统

性能考虑

1. 为(id, version)创建复合索引可大幅提升查询性能
2. 考虑使用分区表管理大量历史数据
3. 对于频繁查询的时间点，可以预计算并缓存结果

通过这种设计，我们可以在Postgraphile中构建出功能完整、关联正确的时间版本化数据查询API。