如何配置Prisma实现数据库连接重试？

Prisma本身没有提供直接的配置项来开启连接失败自动重试，但可以通过以下两种方式实现需求：

1. 手动封装重试逻辑 #

针对单个Prisma客户端调用，编写通用的重试函数，捕获连接相关的错误并进行重试。

示例代码（JavaScript/TypeScript） #

首先定义判断连接错误的工具函数，再实现重试逻辑：

import { Prisma } from '@prisma/client';

// 判断是否为连接类错误
function isConnectionError(error: unknown): boolean {
  return (
    // 初始化连接失败
    error instanceof Prisma.PrismaClientInitializationError ||
    // 运行时连接相关错误（如P1001=无法连接数据库，P1008=查询超时，P1010=数据库认证失败）
    (error instanceof Prisma.PrismaClientKnownRequestError && 
      ['P1001', 'P1008', 'P1010'].includes(error.code))
  );
}

// 通用重试函数
async function withRetry<T>(
  fn: () => Promise<T>,
  maxRetries = 3,
  baseDelay = 1000 // 初始重试间隔（毫秒）
): Promise<T> {
  let retries = maxRetries;
  while (retries > 0) {
    try {
      return await fn();
    } catch (error) {
      if (isConnectionError(error)) {
        retries--;
        if (retries === 0) throw error;
        // 指数退避延迟
        await new Promise(resolve => setTimeout(resolve, baseDelay * (2 ** (maxRetries - retries - 1))));
      } else {
        throw error;
      }
    }
  }
  throw new Error('重试次数耗尽');
}

使用方式 #

在调用Prisma查询时包裹重试函数：

// 查询示例
const getUser = async (userId: number) => {
  return await withRetry(() => 
    prisma.user.findUnique({ where: { id: userId } })
  );
};

2. 优化连接池配置辅助稳定 #

Prisma基于pg驱动维护连接池，可以通过调整连接池参数减少连接失效概率：

• 在schema.prisma的datasource中设置connection_limit，根据业务需求合理配置连接数（避免过多连接导致数据库压力）：

  datasource db {
    provider = "postgresql"
    url      = env("DATABASE_URL")
    connection_limit = 10
  }
  

• 在DATABASE_URL中添加pg驱动参数，比如设置连接超时和保持连接：

  postgresql://user:password@host:port/db?connect_timeout=10&keepalives=1&keepalives_idle=30
  

注意：连接池优化只能降低连接失效的概率，无法替代查询级别的重试逻辑。

内容的提问来源于stack exchange，提问作者Paymahn Moghadasian