类型 API

要从表架构中检索用于 select 和 insert 查询的类型，你可以使用我们的类型助手。

PostgreSQL
MySQL
SQLite
SingleStore 
import { serial, text, pgTable } from 'drizzle-orm/pg-core';
import { type InferSelectModel, type InferInsertModel } from 'drizzle-orm'

const users = pgTable('users', {
  id: serial('id').primaryKey(),
  name: text('name').notNull(),
});

type SelectUser = typeof users.$inferSelect;
type InsertUser = typeof users.$inferInsert;
// or
type SelectUser = typeof users._.$inferSelect;
type InsertUser = typeof users._.$inferInsert;
// or
type SelectUser = InferSelectModel<typeof users>;
type InsertUser = InferInsertModel<typeof users>;

日志记录

要启用默认查询日志记录，只需将 { logger: true } 传递给 drizzle 初始化函数：

import { drizzle } from 'drizzle-orm/...'; // driver specific

const db = drizzle({ logger: true });

你可以通过创建 DefaultLogger 实例并为其提供自定义 writer 来更改日志目标：

import { DefaultLogger, LogWriter } from 'drizzle-orm/logger';
import { drizzle } from 'drizzle-orm/...'; // driver specific

class MyLogWriter implements LogWriter {
  write(message: string) {
    // Write to file, stdout, etc.
  }
}

const logger = new DefaultLogger({ writer: new MyLogWriter() });
const db = drizzle({ logger });

你还可以创建自定义日志器：

import { Logger } from 'drizzle-orm/logger';
import { drizzle } from 'drizzle-orm/...'; // driver specific

class MyLogger implements Logger {
  logQuery(query: string, params: unknown[]): void {
    console.log({ query, params });
  }
}

const db = drizzle({ logger: new MyLogger() });

多项目模式

表创建 API 允许你自定义表名。当你需要将不同项目的模式保存在一个数据库中时，它非常有用。

PostgreSQL
MySQL
SQLite
SingleStore
import { serial, text, pgTableCreator } from 'drizzle-orm/pg-core';

const pgTable = pgTableCreator((name) => `project1_${name}`);

const users = pgTable('users', {
  id: serial('id').primaryKey(),
  name: text('name').notNull(),
});
import { defineConfig } from "drizzle-kit";

export default defineConfig({
  schema: "./src/schema/*",
  out: "./drizzle",
  dialect: "mysql", 
  dbCredentials: {
    url: process.env.DATABASE_URL,
  }
  tablesFilter: ["project1_*"],
});

你可以应用多个 or 过滤器：

tablesFilter: ["project1_*", "project2_*"]

打印 SQL 查询

你可以使用 db 实例或 standalone query builder 打印 SQL 查询。

const query = db
  .select({ id: users.id, name: users.name })
  .from(users)
  .groupBy(users.id)
  .toSQL();
// query:
{
  sql: 'select 'id', 'name' from 'users' group by 'users'.'id'',
  params: [],
}

原始 SQL 查询执行

如果你需要执行一些复杂的查询，而 drizzle-orm 尚无法处理它们，你可以使用 db.execute 方法执行原始的 parametrized 查询。

PostgreSQL
MySQL
SQLite
SingleStore 
const statement = sql`select * from ${users} where ${users.id} = ${userId}`;
const res: postgres.RowList<Record<string, unknown>[]> = await db.execute(statement)

独立查询构建器

Drizzle ORM 提供了一个独立的查询构建器，允许你无需创建数据库实例即可构建查询并获取生成的 SQL。

PostgreSQL
MySQL
SQLite
SingleStore 
import { QueryBuilder } from 'drizzle-orm/pg-core';

const qb = new QueryBuilder();

const query = qb.select().from(users).where(eq(users.name, 'Dan'));
const { sql, params } = query.toSQL();

获取类型表列

你可以获得一个类型表列映射，这在你需要在选择时忽略某些列时非常有用。

PostgreSQL
MySQL
SQLite
SingleStore
index.ts
schema.ts
import { getTableColumns } from "drizzle-orm";
import { user } from "./schema";

const { password, role, ...rest } = getTableColumns(user);

await db.select({ ...rest }).from(users);

获取表信息

PostgreSQL
MySQL
SQLite
SingleStore 
import { getTableConfig, pgTable } from 'drizzle-orm/pg-core';

export const table = pgTable(...);

const {
  columns,
  indexes,
  foreignKeys,
  checks,
  primaryKeys,
  name,
  schema,
} = getTableConfig(table);

比较对象类型（实例或替代方案）

你可以使用 is() 函数检查某个对象是否属于特定的 Drizzle 类型。你可以将它与 Drizzle 中的任何可用类型一起使用。

IMPORTANT

你应该始终使用 is() 而不是 instanceof。

一些示例

import { Column, is } from 'drizzle-orm';

if (is(value, Column)) {
  // value's type is narrowed to Column
}

模拟驱动程序

此 API 是未定义的 drizzle({} as any) API 的后继者，我们在 Drizzle 测试中内部使用过该 API，很少推荐给外部开发者。

我们决定构建并公开一个合适的 API，现在每个 drizzle 驱动程序都支持 drizzle.mock()：

import { drizzle } from "drizzle-orm/node-postgres";

const db = drizzle.mock();

你可以根据需要为类型提供架构

import { drizzle } from "drizzle-orm/node-postgres";
import * as schema from "./schema"

const db = drizzle.mock({ schema });