zod

安装依赖

npm
yarn
pnpm
bun
npm i drizzle-orm@rc zod

选择模式

定义从数据库查询的数据形状,可用于验证 API 响应。

import { int, mssqlTable, text } from 'drizzle-orm/mssql-core';
import { createSelectSchema } from 'drizzle-orm/zod';

const users = mssqlTable('users', {
  id: int().primaryKey().identity(),
  name: text().notNull(),
  age: int().notNull()
});

const userSelectSchema = createSelectSchema(users);

const rows = await db.select({ id: users.id, name: users.name }).top(1).from(users);
const parsed: { id: number; name: string; age: number } = userSelectSchema.parse(rows[0]); // 错误:上面的查询没有返回 `age`

const rows = await db.select().top(1).from(users);
const parsed: { id: number; name: string; age: number } = userSelectSchema.parse(rows[0]); // 将 सफल地解析

也支持视图。

import { createSelectSchema } from 'drizzle-orm/zod';
import { mssqlView } from 'drizzle-orm/mssql-core';
 
const usersView = mssqlView('users_view').as((qb) => qb.select().from(users).where(gt(users.age, 18)));
const usersViewSchema = createSelectSchema(usersView);
const parsed: { id: number; name: string; age: number } = usersViewSchema.parse(...);

插入模式

定义要插入数据库的数据形状,可用于验证 API 请求。

import { int, mssqlTable, text } from 'drizzle-orm/mssql-core';
import { createInsertSchema } from 'drizzle-orm/zod';

const users = mssqlTable('users', {
  id: int().primaryKey().identity(),
  name: text().notNull(),
  age: int().notNull()
});

const userInsertSchema = createInsertSchema(users);

const user = { name: 'John' };
const parsed: { name: string, age: number } = userInsertSchema.parse(user); // 错误:未定义 `age`

const user = { name: 'Jane', age: 30 };
const parsed: { name: string, age: number } = userInsertSchema.parse(user); // 将 सफल地解析
await db.insert(users).values(parsed);

更新模式

定义要在数据库中更新的数据形状,可用于验证 API 请求。

import { int, mssqlTable, text } from 'drizzle-orm/mssql-core';
import { createUpdateSchema } from 'drizzle-orm/zod';
import { eq } from 'drizzle-orm';

const users = mssqlTable('users', {
  id: int().primaryKey().identity(),
  name: text().notNull(),
  age: int().notNull()
});

const userUpdateSchema = createUpdateSchema(users);

const user = { age: 35 };
const parsed: { name?: string | undefined, age?: number | undefined } = userUpdateSchema.parse(user); // 将 সফল地解析
await db.update(users).set(parsed).where(eq(users.name, 'Jane'));

精炼

每个 create schema 函数都接受一个额外的可选参数,你可以用它来扩展、修改或完全覆盖某个字段的 schema。传入回调函数时会进行扩展或修改;传入 Zod schema 时则会覆盖它。

import { int, mssqlTable, text } from 'drizzle-orm/mssql-core';
import { createSelectSchema } from 'drizzle-orm/zod';
import { z } from 'zod/v4';

const users = mssqlTable('users', {
  id: int().primaryKey(),
  name: text().notNull(),
  bio: text(),
  preferences: text()
});

const userSelectSchema = createSelectSchema(users, {
  name: (schema) => schema.max(20), // 扩展 schema
  bio: (schema) => schema.max(1000), // 在变为可空/可选之前扩展 schema
  preferences: z.object({ theme: z.string() }) // 覆盖该字段,包括其可空性
});

const parsed: {
  id: number;
  name: string,
  bio: string | null;
  preferences: {
    theme: string;
  };
} = userSelectSchema.parse(...);

工厂函数

对于更高级的用例,你可以使用 createSchemaFactory 函数。

用例:使用扩展的 Zod 实例

import { int, mssqlTable, text } from 'drizzle-orm/mssql-core';
import { createSchemaFactory } from 'drizzle-orm/zod';
import { z } from '@hono/zod-openapi'; // 扩展的 Zod 实例

const users = mssqlTable('users', {
  id: int().primaryKey().identity(),
  name: text().notNull(),
  age: int().notNull()
});

const { createInsertSchema } = createSchemaFactory({ zodInstance: z });

const userInsertSchema = createInsertSchema(users, {
  // 现在我们可以使用扩展实例
  name: (schema) => schema.openapi({ example: 'John' })
});

用例:类型强制转换

import { mssqlTable, datetime2 } from 'drizzle-orm/mssql-core';
import { createSchemaFactory } from 'drizzle-orm/zod';
import { z } from 'zod/v4';

const users = mssqlTable('users', {
  ...,
  createdAt: datetime2().notNull()
});

const { createInsertSchema } = createSchemaFactory({
  // 此配置只会强制转换日期。将 `coerce` 设为 `true` 可强制转换所有数据类型,或指定其他类型
  coerce: {
    date: true
  }
});

const userInsertSchema = createInsertSchema(users);
// 上面的效果等同于:
const userInsertSchema = z.object({
  ...,
  createdAt: z.coerce.date()
});

数据类型参考

MSSQL 数据类型映射遵循 MSSQL 列构建器。完整的列参考请参见 MSSQL 数据类型 页面。