本文档适用于 drizzle-arktype@0.1.0
及更高版本
你还必须安装 Drizzle ORM v0.36.0 或更高版本和 Arktype v2.0.0 或更高版本。
drizzle-arktype
是 Drizzle ORM 的一个插件,允许你从 Drizzle ORM 模式生成 Arktype 模式。
npm i drizzle-arktype
本文档适用于 drizzle-arktype@0.1.0
及更高版本
你还必须安装 Drizzle ORM v0.36.0 或更高版本和 Arktype v2.0.0 或更高版本。
定义从数据库查询的数据形状 - 可用于验证 API 返回的数据。
import { pgTable, text, integer } from 'drizzle-orm/pg-core';
import { createSelectSchema } from 'drizzle-arktype';
const users = pgTable('users', {
id: integer().generatedAlwaysAsIdentity().primaryKey(),
name: text().notNull(),
age: integer().notNull()
});
const userSelectSchema = createSelectSchema(users);
const rows = await db.select({ id: users.id, name: users.name }).from(users).limit(1);
const parsed: { id: number; name: string; age: number } = userSelectSchema(rows[0]); // 错误:上述查询未返回 `age`
const rows = await db.select().from(users).limit(1);
const parsed: { id: number; name: string; age: number } = userSelectSchema(rows[0]); // 解析成功
视图和枚举类型也支持。
import { pgEnum } from 'drizzle-orm/pg-core';
import { createSelectSchema } from 'drizzle-arktype';
const roles = pgEnum('roles', ['admin', 'basic']);
const rolesSchema = createSelectSchema(roles);
const parsed: 'admin' | 'basic' = rolesSchema(...);
const usersView = pgView('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(...);
定义插入数据库数据的形状 - 可用于验证 API 请求。
import { pgTable, text, integer } from 'drizzle-orm/pg-core';
import { createInsertSchema } from 'drizzle-arktype';
const users = pgTable('users', {
id: integer().generatedAlwaysAsIdentity().primaryKey(),
name: text().notNull(),
age: integer().notNull()
});
const userInsertSchema = createInsertSchema(users);
const user = { name: 'John' };
const parsed: { name: string, age: number } = userInsertSchema(user); // 错误:`age` 未定义
const user = { name: 'Jane', age: 30 };
const parsed: { name: string, age: number } = userInsertSchema(user); // 解析成功
await db.insert(users).values(parsed);
定义更新数据库数据的形状 - 可用于验证 API 请求。
import { pgTable, text, integer } from 'drizzle-orm/pg-core';
import { createUpdateSchema } from 'drizzle-arktype';
import { parse } from 'arktype';
const users = pgTable('users', {
id: integer().generatedAlwaysAsIdentity().primaryKey(),
name: text().notNull(),
age: integer().notNull()
});
const userUpdateSchema = createUpdateSchema(users);
const user = { id: 5, name: 'John' };
const parsed: { name?: string | undefined, age?: number | undefined } = userUpdateSchema(user); // 错误:`id` 是生成列,不能更新
const user = { age: 35 };
const parsed: { name?: string | undefined, age?: number | undefined } = userUpdateSchema(user); // 解析成功
await db.update(users).set(parsed).where(eq(users.name, 'Jane'));
每个 create schema 函数都接受一个附加的可选参数,用于扩展、修改或完全覆盖字段的 schema。定义回调函数将扩展或修改 schema,直接提供 arktype schema 会覆盖原有字段。
import { pgTable, text, integer, json } from 'drizzle-orm/pg-core';
import { createSelectSchema } from 'drizzle-arktype';
import { parse, pipe, maxLength, object, string } from 'arktype';
const users = pgTable('users', {
id: integer().generatedAlwaysAsIdentity().primaryKey(),
name: text().notNull(),
bio: text(),
preferences: json()
});
const userSelectSchema = createSelectSchema(users, {
name: (schema) => pipe(schema, maxLength(20)), // 扩展 schema
bio: (schema) => pipe(schema, maxLength(1000)), // 扩展 schema,注意生效前字段可空/可选
preferences: object({ theme: string() }) // 覆盖字段,包含其可空性
});
const parsed: {
id: number;
name: string,
bio?: string | undefined;
preferences: {
theme: string;
};
} = userSelectSchema(...);
pg.boolean();
mysql.boolean();
sqlite.integer({ mode: 'boolean' });
// Schema
type.boolean;
pg.date({ mode: 'date' });
pg.timestamp({ mode: 'date' });
mysql.date({ mode: 'date' });
mysql.datetime({ mode: 'date' });
mysql.timestamp({ mode: 'date' });
sqlite.integer({ mode: 'timestamp' });
sqlite.integer({ mode: 'timestamp_ms' });
// Schema
type.Date;
pg.date({ mode: 'string' });
pg.timestamp({ mode: 'string' });
pg.cidr();
pg.inet();
pg.interval();
pg.macaddr();
pg.macaddr8();
pg.numeric();
pg.text();
pg.sparsevec();
pg.time();
mysql.binary();
mysql.date({ mode: 'string' });
mysql.datetime({ mode: 'string' });
mysql.decimal();
mysql.time();
mysql.timestamp({ mode: 'string' });
mysql.varbinary();
sqlite.numeric();
sqlite.text({ mode: 'text' });
// Schema
type.string;
pg.bit({ dimensions: ... });
// Schema
type(`/^[01]{${column.dimensions}}$/`);
pg.uuid();
// Schema
type(/^[\da-f]{8}(?:-[\da-f]{4}){3}-[\da-f]{12}$/iu);
pg.char({ length: ... });
mysql.char({ length: ... });
// Schema
type.string.exactlyLength(length);
pg.varchar({ length: ... });
mysql.varchar({ length: ... });
sqlite.text({ mode: 'text', length: ... });
// Schema
type.string.atMostLength(length);
mysql.tinytext();
// Schema
type.string.atMostLength(255); // 无符号 8 位整数范围限制
mysql.text();
// Schema
type.string.atMostLength(65_535); // 无符号 16 位整数范围限制
mysql.mediumtext();
// Schema
type.string.atMostLength(16_777_215); // 无符号 24 位整数范围限制
mysql.longtext();
// Schema
type.string.atMostLength(4_294_967_295); // 无符号 32 位整数范围限制
pg.text({ enum: ... });
pg.char({ enum: ... });
pg.varchar({ enum: ... });
mysql.tinytext({ enum: ... });
mysql.mediumtext({ enum: ... });
mysql.text({ enum: ... });
mysql.longtext({ enum: ... });
mysql.char({ enum: ... });
mysql.varchar({ enum: ... });
mysql.mysqlEnum(..., ...);
sqlite.text({ mode: 'text', enum: ... });
// Schema
type.enumerated(...enum);
mysql.tinyint();
// Schema
type.keywords.number.integer.atLeast(-128).atMost(127); // 8 位有符号整数上下限
mysql.tinyint({ unsigned: true });
// Schema
type.keywords.number.integer.atLeast(0).atMost(255); // 8 位无符号整数上下限
pg.smallint();
pg.smallserial();
mysql.smallint();
// Schema
type.keywords.number.integer.atLeast(-32_768).atMost(32_767); // 16 位有符号整数上下限
mysql.smallint({ unsigned: true });
// Schema
type.keywords.number.integer.atLeast(0).atMost(65_535); // 16 位无符号整数上下限
pg.real();
mysql.float();
// Schema
type.number.atLeast(-8_388_608).atMost(8_388_607); // 24 位整数上下限
mysql.mediumint();
// Schema
type.keywords.number.integer.atLeast(-8_388_608).atMost(8_388_607); // 24 位整数上下限
mysql.float({ unsigned: true });
// Schema
type.number.atLeast(0).atMost(16_777_215); // 24 位无符号整数上下限
mysql.mediumint({ unsigned: true });
// Schema
type.keywords.number.integer.atLeast(0).atMost(16_777_215); // 24 位无符号整数上下限
pg.integer();
pg.serial();
mysql.int();
// Schema
type.keywords.number.integer.atLeast(-2_147_483_648).atMost(2_147_483_647); // 32 位整数上下限
mysql.int({ unsigned: true });
// Schema
type.keywords.number.integer.atLeast(0).atMost(4_294_967_295); // 32 位无符号整数上下限
pg.doublePrecision();
mysql.double();
mysql.real();
sqlite.real();
// Schema
type.number.atLeast(-140_737_488_355_328).atMost(140_737_488_355_327); // 48 位整数上下限
mysql.double({ unsigned: true });
// Schema
type.number.atLeast(0).atMost(281_474_976_710_655); // 48 位无符号整数上下限
pg.bigint({ mode: 'number' });
pg.bigserial({ mode: 'number' });
mysql.bigint({ mode: 'number' });
mysql.bigserial({ mode: 'number' });
sqlite.integer({ mode: 'number' });
// Schema
type.keywords.number.integer.atLeast(-9_007_199_254_740_991).atMost(9_007_199_254_740_991); // JavaScript 最小最大安全整数
mysql.serial();
// Schema
type.keywords.number.integer.atLeast(0).atMost(9_007_199_254_740_991); // JavaScript 最大安全整数
pg.bigint({ mode: 'bigint' });
pg.bigserial({ mode: 'bigint' });
mysql.bigint({ mode: 'bigint' });
sqlite.blob({ mode: 'bigint' });
// Schema
type.bigint.narrow(
(value, ctx) => value < -9_223_372_036_854_775_808n ? ctx.mustBe('greater than') : value > 9_223_372_036_854_775_807n ? ctx.mustBe('less than') : true
); // 64 位整数上下限
mysql.bigint({ mode: 'bigint', unsigned: true });
// Schema
type.bigint.narrow(
(value, ctx) => value < 0n ? ctx.mustBe('greater than') : value > 18_446_744_073_709_551_615n ? ctx.mustBe('less than') : true
); // 64 位无符号整数上下限
mysql.year();
// Schema
type.keywords.number.integer.atLeast(1_901).atMost(2_155);
pg.geometry({ type: 'point', mode: 'tuple' });
pg.point({ mode: 'tuple' });
// Schema
type([type.number, type.number]);
pg.geometry({ type: 'point', mode: 'xy' });
pg.point({ mode: 'xy' });
// Schema
type({ x: type.number, y: type.number });
pg.halfvec({ dimensions: ... });
pg.vector({ dimensions: ... });
// Schema
type.number.array().exactlyLength(dimensions);
pg.line({ mode: 'abc' });
// Schema
type({ a: type.number, b: type.number, c: type.number });
pg.line({ mode: 'tuple' });
// Schema
type([type.number, type.number, type.number]);
pg.json();
pg.jsonb();
mysql.json();
sqlite.blob({ mode: 'json' });
sqlite.text({ mode: 'json' });
// Schema
type('string | number | boolean | null').or(type('unknown.any[] | Record<string, unknown.any>'));
sqlite.blob({ mode: 'buffer' });
// Schema
type.instanceOf(Buffer);
pg.dataType().array(...);
// Schema
baseDataTypeSchema.array().exactlyLength(size);