prisma-driver-adapter-implementation

Name: prisma-driver-adapter-implementation
Author: prisma

Prisma v7 驱动适配器工作所需的必备参考。在实现或修改适配器、添加数据库驱动、或接触 SqlDriverAdapter/Transaction 接口时使用。包含无法从代码示例推断的关键契约细节——包括事务生命周期协议、错误映射要求和验证清单。现有实现不能替代此技能。

44Star

3Fork

更新于 2026/7/14

获取 Skill 源代码

Prisma 7 驱动适配器实现指南

本技能提供实现任何数据库的 Prisma ORM v7 驱动适配器所需的一切。

架构概览

┌─────────────────────────────────────────────────────────────────┐
│                         PrismaClient                            │
│                    (需要适配器工厂)                               │
└─────────────────────────────────────────────────────────────────┘
                                │
                                ▼
┌─────────────────────────────────────────────────────────────────┐
│            SqlMigrationAwareDriverAdapterFactory                │
│   ┌─────────────────────┐    ┌─────────────────────────────┐    │
│   │ connect()           │    │ connectToShadowDb()         │    │
│   │ → SqlDriverAdapter  │    │ → SqlDriverAdapter          │    │
│   └─────────────────────┘    └─────────────────────────────┘    │
└─────────────────────────────────────────────────────────────────┘
                                │
                                ▼
┌─────────────────────────────────────────────────────────────────┐
│                      SqlDriverAdapter                           │
│  ┌──────────────┐ ┌──────────────┐ ┌──────────────────────────┐ │
│  │ queryRaw()   │ │ executeRaw() │ │ startTransaction()       │ │
│  │ → ResultSet  │ │ → number     │ │ → Transaction            │ │
│  └──────────────┘ └──────────────┘ └──────────────────────────┘ │
│  ┌──────────────┐ ┌──────────────┐ ┌──────────────────────────┐ │
│  │executeScript │ │ dispose()    │ │ getConnectionInfo()      │ │
│  └──────────────┘ └──────────────┘ └──────────────────────────┘ │
└─────────────────────────────────────────────────────────────────┘
                                │
                                ▼
┌─────────────────────────────────────────────────────────────────┐
│                        Transaction                              │
│  扩展 SqlQueryable + commit() + rollback() + options            │
│  (仅生命周期钩子 — Prisma 通过 executeRaw 发送 SQL)              │
└─────────────────────────────────────────────────────────────────┘

必需接口

从 @prisma/driver-adapter-utils 导入：

import type {
  ColumnType,
  IsolationLevel,
  SqlDriverAdapter,
  SqlMigrationAwareDriverAdapterFactory,
  SqlQuery,
  SqlQueryable,
  SqlResultSet,
  Transaction,
  TransactionOptions,
  ArgType,
  ConnectionInfo,
  MappedError,
} from "@prisma/driver-adapter-utils";
import {
  ColumnTypeEnum,
  DriverAdapterError,
} from "@prisma/driver-adapter-utils";

接口定义

SqlQuery (queryRaw/executeRaw 的输入)

type SqlQuery = {
  sql: string; // 带占位符的参数化 SQL
  args: Array<unknown>; // 绑定的参数值
  argTypes: Array<ArgType>; // 每个参数的类型提示
};

type ArgType = {
  scalarType: ArgScalarType; // 'string' | 'int' | 'bigint' | 'float' | 'decimal' | 'boolean' | 'enum' | 'uuid' | 'json' | 'datetime' | 'bytes' | 'unknown'
  dbType?: string;
  arity: "scalar" | "list";
};

SqlResultSet (queryRaw 的输出)

interface SqlResultSet {
  columnNames: Array<string>; // 按顺序的列名
  columnTypes: Array<ColumnType>; // 与 columnNames 匹配的列类型
  rows: Array<Array<unknown>>; // 行数据，以数组形式
  lastInsertId?: string; // 用于没有 RETURNING 的 INSERT
}

ColumnTypeEnum 值

const ColumnTypeEnum = {
  Int32: 0,
  Int64: 1,
  Float: 2,
  Double: 3,
  Numeric: 4,
  Boolean: 5,
  Character: 6,
  Text: 7,
  Date: 8,
  Time: 9,
  DateTime: 10,
  Json: 11,
  Enum: 12,
  Bytes: 13,
  Set: 14,
  Uuid: 15,
  Int32Array: 64,
  Int64Array: 65,
  FloatArray: 66,
  DoubleArray: 67,
  NumericArray: 68,
  BooleanArray: 69,
  CharacterArray: 70,
  TextArray: 71,
  DateArray: 72,
  TimeArray: 73,
  DateTimeArray: 74,
  JsonArray: 75,
  EnumArray: 76,
  BytesArray: 77,
  UuidArray: 78,
  UnknownNumber: 128,
} as const;

SqlDriverAdapter

interface SqlDriverAdapter extends SqlQueryable {
  executeScript(script: string): Promise<void>;
  startTransaction(isolationLevel?: IsolationLevel): Promise<Transaction>;
  getConnectionInfo?(): ConnectionInfo;
  dispose(): Promise<void>;
}

Transaction

interface Transaction extends SqlQueryable {
  readonly options: TransactionOptions;
  commit(): Promise<void>;
  rollback(): Promise<void>;
}

type TransactionOptions = { usePhantomQuery: boolean };

SqlMigrationAwareDriverAdapterFactory

interface SqlMigrationAwareDriverAdapterFactory {
  readonly provider: "mysql" | "postgres" | "sqlite" | "sqlserver";
  readonly adapterName: string;
  connect(): Promise<SqlDriverAdapter>;
  connectToShadowDb(): Promise<SqlDriverAdapter>;
}

实现步骤

步骤 1：创建 Queryable 基类

class MyQueryable<TClient> implements SqlQueryable {
  readonly provider = "postgres" as const; // 或 'sqlite' | 'mysql' | 'sqlserver'
  readonly adapterName = "@my-org/adapter-mydb" as const;

  constructor(protected readonly client: TClient) {}

  async queryRaw(query: SqlQuery): Promise<SqlResultSet> {
    try {
      const args = query.args.map((arg, i) =>
        mapArg(arg, query.argTypes[i] ?? { scalarType: "unknown", arity: "scalar" })
      );

      // 使用你的驱动执行查询
      const result = await this.client.query(query.sql, args);

      // 提取列元数据
      const columnNames = /* 从结果中获取 */;
      const columnTypes = /* 映射到 ColumnTypeEnum */;

      // 将行映射为 ResultValue 数组
      const rows = result.map(row => mapRow(row, columnTypes));

      return { columnNames, columnTypes, rows };
    } catch (e) {
      this.onError(e);
    }
  }

  async executeRaw(query: SqlQuery): Promise<number> {
    try {
      const args = query.args.map((arg, i) =>
        mapArg(arg, query.argTypes[i] ?? { scalarType: "unknown", arity: "scalar" })
      );
      const result = await this.client.query(query.sql, args);
      return result.affectedRows ?? 0;
    } catch (e) {
      this.onError(e);
    }
  }

  protected onError(error: unknown): never {
    throw new DriverAdapterError(convertDriverError(error));
  }
}

步骤 2：创建 Transaction 类

关键：commit() 和 rollback() 只是生命周期钩子。它们不能发出 SQL。Prisma 通过事务对象上的 executeRaw 发送 COMMIT/ROLLBACK。

class MyTransaction extends MyQueryable<TClient> implements Transaction {
  readonly options: TransactionOptions;
  readonly #release: () => void;

  constructor(
    client: TClient,
    options: TransactionOptions,
    release: () => void,
  ) {
    super(client);
    this.options = options;
    this.#release = release;
  }

  commit(): Promise<void> {
    // 不要在这里发出 COMMIT SQL — Prisma 通过 executeRaw 完成
    this.#release(); // 释放连接/资源
    return Promise.resolve();
  }

  rollback(): Promise<void> {
    // 不要在这里发出 ROLLBACK SQL — Prisma 通过 executeRaw 完成
    this.#release();
    return Promise.resolve();
  }
}

步骤 3：创建 Adapter 类

class MyAdapter extends MyQueryable<TClient> implements SqlDriverAdapter {
  #transactionDepth = 0;

  constructor(client: TClient) {
    super(client);
  }

  async executeScript(script: string): Promise<void> {
    // 对于 SQLite：按 ';' 分割并逐个执行语句
    // 对于 Postgres：使用多语句执行
    try {
      // 实现取决于驱动能力
    } catch (e) {
      this.onError(e);
    }
  }

  async startTransaction(
    isolationLevel?: IsolationLevel,
  ): Promise<Transaction> {
    // 验证你的数据库的隔离级别
    const validLevels = new Set<IsolationLevel>([
      "READ UNCOMMITTED",
      "READ COMMITTED",
      "REPEATABLE READ",
      "SERIALIZABLE",
    ]);

    if (isolationLevel !== undefined && !validLevels.has(isolationLevel)) {
      throw new DriverAdapterError({
        kind: "InvalidIsolationLevel",
        level: isolationLevel,
      });
    }

    const options: TransactionOptions = { usePhantomQuery: false };

    this.#transactionDepth += 1;
    const depth = this.#transactionDepth;

    try {
      if (depth === 1) {
        // 发出 BEGIN（如果指定了隔离级别）
        const beginSql = isolationLevel
          ? `BEGIN ISOLATION LEVEL ${isolationLevel}`
          : "BEGIN";
        await this.client.query(beginSql);
      } else {
        // 嵌套：使用保存点
        await this.client.query(`SAVEPOINT sp_${depth}`);
      }
    } catch (e) {
      this.#transactionDepth -= 1;
      this.onError(e);
    }

    const release = () => {
      this.#transactionDepth -= 1;
    };
    return new MyTransaction(this.client, options, release);
  }

  getConnectionInfo(): ConnectionInfo {
    return { supportsRelationJoins: true };
  }

  async dispose(): Promise<void> {
    await this.client.close();
  }
}

步骤 4：创建 Factory 类

export type MyAdapterConfig = {
  url: string;
};

export type MyAdapterOptions = {
  shadowDatabaseUrl?: string;
};

export class MyAdapterFactory implements SqlMigrationAwareDriverAdapterFactory {
  readonly provider = "postgres" as const;
  readonly adapterName = "@my-org/adapter-mydb" as const;

  constructor(
    private readonly config: MyAdapterConfig,
    private readonly options?: MyAdapterOptions,
  ) {}

  connect(): Promise<SqlDriverAdapter> {
    return Promise.resolve(new MyAdapter(openConnection(this.config.url)));
  }

  connectToShadowDb(): Promise<SqlDriverAdapter> {
    const url = this.options?.shadowDatabaseUrl ?? this.config.url;
    return Promise.resolve(new MyAdapter(openConnection(url)));
  }
}

转换辅助函数

参数映射（输入）

将 Prisma 参数值转换为驱动原生类型：

function mapArg(arg: unknown, argType: ArgType): unknown {
  if (arg === null || arg === undefined) return null;

  // String → number 用于 int 列
  if (typeof arg === "string" && argType.scalarType === "int")
    return Number.parseInt(arg, 10);

  // String → number 用于 float 列
  if (typeof arg === "string" && argType.scalarType === "float")
    return Number.parseFloat(arg);

  // String → BigInt 用于 bigint 列
  if (typeof arg === "string" && argType.scalarType === "bigint")
    return BigInt(arg);

  // Base64 字符串 → Buffer 用于 bytes 列
  if (typeof arg === "string" && argType.scalarType === "bytes")
    return Buffer.from(arg, "base64");

  // Boolean → 0/1 用于 SQLite
  if (typeof arg === "boolean" && /* SQLite */)
    return arg ? 1 : 0;

  return arg;
}

行映射（输出）

将驱动结果值转换为 Prisma 期望的类型：

function mapRow(row: unknown[], columnTypes: ColumnType[]): ResultValue[] {
  const result: ResultValue[] = [];

  for (let i = 0; i < row.length; i++) {
    const value = row[i] ?? null;
    const colType = columnTypes[i];

    if (value === null) {
      result.push(null);
      continue;
    }

    // bigint → string 用于 Int64（JSON 安全）
    if (typeof value === "bigint") {
      result.push(value.toString());
      continue;
    }

    // Date → ISO 8601 字符串用于 DateTime
    if (value instanceof Date) {
      result.push(value.toISOString());
      continue;
    }

    // JSON 对象 → 字符串化
    if (colType === ColumnTypeEnum.Json && typeof value === "object") {
      result.push(JSON.stringify(value));
      continue;
    }

    result.push(value as ResultValue);
  }

  return result;
}

列类型推断

当驱动不提供类型元数据时，从 JS 值推断：

function inferColumnType(value: NonNullable<unknown>): ColumnType {
  if (typeof value === "boolean") return ColumnTypeEnum.Boolean;
  if (typeof value === "bigint") return ColumnTypeEnum.Int64;
  if (value instanceof Uint8Array) return ColumnTypeEnum.Bytes;
  if (value instanceof Date) return ColumnTypeEnum.DateTime;
  if (Array.isArray(value)) return ColumnTypeEnum.Text; // 回退
  if (typeof value === "object") return ColumnTypeEnum.Json;
  if (typeof value === "number") return ColumnTypeEnum.UnknownNumber;
  return ColumnTypeEnum.Text;
}

错误处理

将驱动错误映射为 MappedError，以便 Prisma 正确处理：

function convertDriverError(error: unknown): MappedError {
  if (error instanceof Error) {
    // 数据库特定的错误映射
    const dbError = error as Error & { code?: string; errno?: number };

    // PostgreSQL 示例
    if (dbError.code === "23505") {
      return { kind: "UniqueConstraintViolation" };
    }
    if (dbError.code === "23502") {
      return { kind: "NullConstraintViolation" };
    }
    if (dbError.code === "23503") {
      return { kind: "ForeignKeyConstraintViolation" };
    }
    if (dbError.code === "42P01") {
      return { kind: "TableDoesNotExist" };
    }

    // SQLite 示例
    if (error.name === "SQLiteError") {
      return {
        kind: "sqlite",
        extendedCode: dbError.errno ?? 1,
        message: error.message,
      };
    }

    // PostgreSQL 原始错误
    if (dbError.code) {
      return {
        kind: "postgres",
        code: dbError.code,
        severity: "ERROR",
        message: error.message,
        detail: undefined,
        column: undefined,
        hint: undefined,
      };
    }
  }

  return { kind: "GenericJs", id: 0 };
}

数据库特定说明

SQLite

打开数据库时设置 safeIntegers: true 以获取大整数的 bigint
仅 SERIALIZABLE 隔离级别有效
executeScript：按 ; 分割并逐个执行语句
布尔值：存储为 0/1，返回为布尔值

PostgreSQL

所有标准隔离级别均有效
对于连接池（PgBouncer），使用 prepare: false
事务需要专用连接（reserve() 模式）
executeScript：使用多语句执行（某些驱动中的 .simple()）
int8 列可能返回为字符串（已由驱动字符串化）
numeric 列返回为字符串以保持精度

MySQL/MariaDB

支持 READ UNCOMMITTED、READ COMMITTED、REPEATABLE READ、SERIALIZABLE
使用 ? 占位符作为参数
将 BIGINT 作为字符串处理以处理大值

测试策略

单元测试（无 PrismaClient）

直接使用原始数据库驱动测试适配器：

describe("queryRaw", () => {
  test("返回列名和类型", async () => {
    const adapter = new MyAdapter(createTestConnection());
    const result = await adapter.queryRaw({
      sql: "SELECT id, name FROM users",
      args: [],
      argTypes: [],
    });
    expect(result.columnNames).toEqual(["id", "name"]);
    expect(result.columnTypes[0]).toBe(ColumnTypeEnum.Int32);
  });
});

describe("startTransaction", () => {
  test("提交持久化更改", async () => {
    const adapter = new MyAdapter(createTestConnection());
    const tx = await adapter.startTransaction();
    await tx.executeRaw({
      sql: "INSERT INTO users (name) VALUES (?)",
      args: ["Alice"],
      argTypes: [],
    });
    // Prisma 通过 executeRaw 发送 COMMIT
    await tx.executeRaw({ sql: "COMMIT", args: [], argTypes: [] });
    await tx.commit(); // 仅生命周期钩子
    // 验证数据已持久化
  });
});

端到端测试（使用 PrismaClient）

测试完整集成：

describe("E2E", () => {
  let prisma: PrismaClient;

  beforeEach(async () => {
    const factory = new MyAdapterFactory({ url: TEST_DB_URL });
    prisma = new PrismaClient({ adapter: factory });
  });

  test("CRUD 操作", async () => {
    const user = await prisma.user.create({ data: { name: "Alice" } });
    expect(user.id).toBeGreaterThan(0);

    const found = await prisma.user.findUnique({ where: { id: user.id } });
    expect(found?.name).toBe("Alice");
  });

  test("事务在出错时回滚", async () => {
    await expect(
      prisma.$transaction(async (tx) => {
        await tx.user.create({ data: { name: "Bob" } });
        throw new Error("回滚！");
      }),
    ).rejects.toThrow();

    expect(await prisma.user.count()).toBe(0);
  });
});

使用示例

import { PrismaClient } from "./generated/prisma/client";
import { MyAdapterFactory } from "@my-org/adapter-mydb";

const factory = new MyAdapterFactory({
  url: process.env.DATABASE_URL!,
});

const prisma = new PrismaClient({ adapter: factory });

// 正常使用 prisma
const users = await prisma.user.findMany();

检查清单

在认为适配器完成之前：

[ ] SqlMigrationAwareDriverAdapterFactory 实现了 connect() 和 connectToShadowDb()
[ ] SqlDriverAdapter 实现了 queryRaw、executeRaw、executeScript、startTransaction、dispose
[ ] Transaction 实现了 queryRaw、executeRaw、commit、rollback，并带有 options: { usePhantomQuery: false }
[ ] commit() 和 rollback() 仅是生命周期钩子（不发出 SQL）
[ ] startTransaction 发出 BEGIN（深度 1）或 SAVEPOINT sp_N（嵌套）
[ ] 参数映射处理：string→int、string→bigint、string→float、base64→bytes
[ ] 行映射处理：bigint→string、Date→ISO 字符串、JSON→字符串
[ ] 列类型正确映射到 ColumnTypeEnum
[ ] 错误包装在 DriverAdapterError 中，带有正确的 MappedError 类型
[ ] 针对目标数据库的隔离级别验证
[ ] 单元测试通过 queryRaw、executeRaw、executeScript、事务
[ ] 端到端测试通过真实的 PrismaClient

你可能还喜欢

things-mac

384Kdatabase

Add, update, list, search, or inspect Things 3 todos, inbox, today, projects, areas, and tags on macOS.

steipete

获取

plan-orchestrate

237Kdatabase

Read a plan document, decompose it into steps, design a per-step agent chain from the ECC catalogue, and emit ready-to-paste /orchestrate custom prompts. Generative only — never invokes /orchestrate itself. Use when the user has a multi-step plan and wants to drive it through orchestrate without composing chains by hand.

affaan-m

获取

project-overview

81Kdatabase

LobeHub open-source monorepo architecture map. Use when locating code layers, understanding apps/packages/src layout, business stubs, project structure, or onboarding to the repository.

lobehub

获取

继续探索

浏览 database 热门 Skills 最近更新