Postgres.js 多租户架构实现方案探讨

多租户架构背景

在现代SaaS应用中，多租户架构是一个常见需求。Postgres.js作为PostgreSQL的Node.js客户端，开发者经常需要在其基础上实现多租户支持。本文探讨了基于Postgres.js实现多租户架构的几种技术方案。

共享表模式实现方案

最常见的多租户实现方式是在共享表中添加tenant_id字段。这种方式下，每个SQL查询都需要包含租户过滤条件。

方案一：高阶函数封装

通过创建高阶函数来统一添加租户条件：

function tenant(tenantId) {
  return (query) => sql`${query} and tenant_id = ${tenantId}`
}

// 使用示例
const getRooms = (roomIds) => sql`select * from room where room_id in ${roomIds}`;
await tenant(1)(getRooms([1,2,3]));

这种方案简单直接，但需要在每个查询处显式调用tenant函数。

方案二：事务封装

通过Postgres.js的事务机制，在事务开始时设置租户上下文：

export async function sql<T extends readonly object[]>(strings: any, ...args: any[]) {
  const tenantId = '1';
  return sqlOriginal.begin(async (pg_sql) => {
    pg_sql`select set_config('myapp.current_tenant', ${tenantId}, false);`;
    return pg_sql<T>(strings as TemplateStringsArray, ...args);
  });
}

这种方式利用了PostgreSQL的会话级配置(set_config)，配合RLS(Row Level Security)可以实现自动过滤。

行级安全(RLS)方案

PostgreSQL原生支持RLS，可以创建策略自动过滤数据：

CREATE POLICY tenant_isolation_policy ON room
    USING (tenant_id = current_setting('myapp.current_tenant')::integer;

配合事务封装方案，可以实现透明的多租户数据隔离。

架构选择建议

1. 共享表+tenant_id字段：适合租户数量多但数据量不大的场景
2. Schema隔离：每个租户独立schema，隔离性更好
3. 独立数据库：最高隔离级别，适合大型企业级应用

Postgres.js作为轻量级客户端，与这些架构都能良好配合。开发者需要根据业务规模、性能要求和维护成本进行权衡选择。

性能优化建议

1. 确保tenant_id字段有索引
2. 考虑使用连接池为不同租户分配不同连接
3. 对于高频查询，可以使用视图预先过滤
4. 监控多租户环境下的查询性能

通过合理设计，Postgres.js完全可以支撑企业级的多租户应用需求。