drizzle-typebox

drizzle-typebox 是一个针对 Drizzle ORM 的插件,允许你从 Drizzle ORM 模式生成 Typebox 模式。

安装依赖

npm
yarn
pnpm
bun
npm i drizzle-typebox
IMPORTANT

本文件对于 drizzle-typebox@0.2.0 及以上版本的文档

你还必须安装 Drizzle ORM v0.36.0 或更高版本,以及 Typebox v0.34.8 或更高版本。

选择模式

定义从数据库查询的数据的结构 - 可用于验证 API 响应。

import { pgTable, text, integer } from 'drizzle-orm/pg-core';
import { createSelectSchema } from 'drizzle-typebox';
import { Value } from '@sinclair/typebox/value';

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 } = Value.Parse(userSelectSchema, rows[0]); // 错误:`age`在上述查询中未返回

const rows = await db.select().from(users).limit(1);
const parsed: { id: number; name: string; age: number } = Value.Parse(userSelectSchema, rows[0]); // 将成功解析

视图和枚举也被支持。

import { pgEnum } from 'drizzle-orm/pg-core';
import { createSelectSchema } from 'drizzle-typebox';
import { Value } from '@sinclair/typebox/value';

const roles = pgEnum('roles', ['admin', 'basic']);
const rolesSchema = createSelectSchema(roles);
const parsed: 'admin' | 'basic' = Value.Parse(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 } = Value.Parse(usersViewSchema, ...);

插入模式

定义要插入到数据库的数据的结构 - 可用于验证 API 请求。

import { pgTable, text, integer } from 'drizzle-orm/pg-core';
import { createInsertSchema } from 'drizzle-typebox';
import { Value } from '@sinclair/typebox/value';

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 } = Value.Parse(userInsertSchema, user); // 错误:`age`未定义

const user = { name: 'Jane', age: 30 };
const parsed: { name: string, age: number } = Value.Parse(userInsertSchema, user); // 将成功解析
await db.insert(users).values(parsed);

更新模式

定义要更新到数据库的数据的结构 - 可用于验证 API 请求。

import { pgTable, text, integer } from 'drizzle-orm/pg-core';
import { createUpdateSchema } from 'drizzle-typebox';
import { Value } from '@sinclair/typebox/value';

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 } = Value.Parse(userUpdateSchema, user); // 错误:`id`是一个生成列,无法更新

const user = { age: 35 };
const parsed: { name?: string | undefined, age?: number | undefined } = Value.Parse(userUpdateSchema, user); // 将成功解析
await db.update(users).set(parsed).where(eq(users.name, 'Jane'));

精炼

每个创建模式函数接受一个额外的可选参数,你可以用它来扩展、修改或完全覆盖字段的模式。定义一个回调函数将扩展或修改,而提供一个 Typebox 模式将覆盖它。

import { pgTable, text, integer, json } from 'drizzle-orm/pg-core';
import { createSelectSchema } from 'drizzle-typebox';
import { Type } from '@sinclair/typebox';
import { Value } from '@sinclair/typebox/value';

const users = pgTable('users', {
  id: integer().generatedAlwaysAsIdentity().primaryKey(),
  name: text().notNull(),
  bio: text(),
  preferences: json()
});

const userSelectSchema = createSelectSchema(users, {
  name: (schema) => Type.String({ ...schema, maxLength: 20 }), // 扩展模式
  bio: (schema) => Type.String({ ...schema, maxLength: 1000 }), // 在变为可选/可空之前扩展模式
  preferences: Type.Object({ theme: Type.String() }) // 覆盖字段,包括它的可空性
});

const parsed: {
  id: number;
  name: string,
  bio?: string | undefined;
  preferences: {
    theme: string;
  };
} = Value.Parse(userSelectSchema, ...);

工厂函数

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

用例:使用扩展的 Typebox 实例

import { pgTable, text, integer } from 'drizzle-orm/pg-core';
import { createSchemaFactory } from 'drizzle-typebox';
import { t } from 'elysia'; // 扩展 Typebox 实例

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

const { createInsertSchema } = createSchemaFactory({ typeboxInstance: t });

const userInsertSchema = createInsertSchema(users, {
  // 我们现在可以使用扩展的实例
  name: (schema) => t.Number({ ...schema }, { error: '`name` must be a string' })
});

数据类型参考

pg.boolean();

mysql.boolean();

sqlite.integer({ mode: 'boolean' });

// 模式
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' });

// 模式
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' });

// 模式
Type.String();
pg.bit({ dimensions: ... });

// 模式
t.RegExp(/^[01]+$/, { maxLength: dimensions });
pg.uuid();

// 模式
Type.String({ format: 'uuid' });
pg.char({ length: ... });

mysql.char({ length: ... });

// 模式
Type.String({ minLength: length, maxLength: length });
pg.varchar({ length: ... });

mysql.varchar({ length: ... });

sqlite.text({ mode: 'text', length: ... });

// 模式
Type.String({ maxLength: length });
mysql.tinytext();

// 模式
Type.String({ maxLength: 255 }); // 无符号 8 位整数限制
mysql.text();

// 模式
Type.String({ maxLength: 65_535 }); // 无符号 16 位整数限制
mysql.mediumtext();

// 模式
Type.String({ maxLength: 16_777_215 }); // 无符号 24 位整数限制
mysql.longtext();

// 模式
Type.String({ maxLength: 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: ... });

// 模式
Type.Enum(enum);
mysql.tinyint();

// 模式
Type.Integer({ minimum: -128, maximum: 127 }); // 8 位整数下限和上限
mysql.tinyint({ unsigned: true });

// 模式
Type.Integer({ minimum: 0, maximum: 255 }); // 无符号 8 位整数下限和上限
pg.smallint();
pg.smallserial();

mysql.smallint();

// 模式
Type.Integer({ minimum: -32_768, maximum: 32_767 }); // 16 位整数下限和上限
mysql.smallint({ unsigned: true });

// 模式
Type.Integer({ minimum: 0, maximum: 65_535 }); // 无符号 16 位整数下限和上限
pg.real();

mysql.float();

// 模式
Type.Number().min(-8_388_608).max(8_388_607); // 24 位整数下限和上限
mysql.mediumint();

// 模式
Type.Integer({ minimum: -8_388_608, maximum: 8_388_607 }); // 24 位整数下限和上限
mysql.float({ unsigned: true });

// 模式
Type.Number({ minimum: 0, maximum: 16_777_215 }); // 无符号 24 位整数下限和上限
mysql.mediumint({ unsigned: true });

// 模式
Type.Integer({ minimum: 0, maximum: 16_777_215 }); // 无符号 24 位整数下限和上限
pg.integer();
pg.serial();

mysql.int();

// 模式
Type.Integer({ minimum: -2_147_483_648, maximum: 2_147_483_647 }); // 32 位整数下限和上限
mysql.int({ unsigned: true });

// 模式
Type.Integer({ minimum: 0, maximum: 4_294_967_295 }); // 无符号 32 位整数下限和上限
pg.doublePrecision();

mysql.double();
mysql.real();

sqlite.real();

// 模式
Type.Number({ minimum: -140_737_488_355_328, maximum: 140_737_488_355_327 }); // 48 位整数下限和上限
mysql.double({ unsigned: true });

// 模式
Type.Numer({ minimum: 0, maximum: 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' });

// 模式
Type.Integer({ minimum: -9_007_199_254_740_991, maximum: 9_007_199_254_740_991 }); // Javascript 最小和最大安全整数
mysql.serial();

Type.Integer({ minimum: 0, maximum: 9_007_199_254_740_991 }); // Javascript 最大安全整数
pg.bigint({ mode: 'bigint' });
pg.bigserial({ mode: 'bigint' });

mysql.bigint({ mode: 'bigint' });

sqlite.blob({ mode: 'bigint' });

// 模式
Type.BigInt({ minimum: -9_223_372_036_854_775_808n, maximum: 9_223_372_036_854_775_807n }); // 64 位整数下限和上限
mysql.bigint({ mode: 'bigint', unsigned: true });

// 模式
Type.BigInt({ minimum: 0, maximum: 18_446_744_073_709_551_615n }); // 无符号 64 位整数下限和上限
mysql.year();

// 模式
Type.Integer({ minimum: 1_901, maximum: 2_155 });
pg.geometry({ type: 'point', mode: 'tuple' });
pg.point({ mode: 'tuple' });

// 模式
Type.Tuple([Type.Number(), Type.Number()]);
pg.geometry({ type: 'point', mode: 'xy' });
pg.point({ mode: 'xy' });

// 模式
Type.Object({ x: Type.Number(), y: Type.Number() });
pg.halfvec({ dimensions: ... });
pg.vector({ dimensions: ... });

// 模式
Type.Array(Type.Number(), { minItems: dimensions, maxItems: dimensions });
pg.line({ mode: 'abc' });

// 模式
Type.Object({ a: Type.Number(), b: Type.Number(), c: Type.Number() });
pg.line({ mode: 'tuple' });

// 模式
Type.Tuple([Type.Number(), Type.Number(), Type.Number()]);
pg.json();
pg.jsonb();

mysql.json();

sqlite.blob({ mode: 'json' });
sqlite.text({ mode: 'json' });

// 模式
Type.Recursive((self) => Type.Union([Type.Union([Type.String(), Type.Number(), Type.Boolean(), Type.Null()]), Type.Array(self), Type.Record(Type.String(), self)]));
sqlite.blob({ mode: 'buffer' });

// 模式
TypeRegistry.Set('Buffer', (_, value) => value instanceof Buffer); // drizzle-typebox 运行此方法以将其添加到 Typebox 的类型系统
{ [Kind]: 'Buffer', type: 'buffer' }; // Typebox 模式
pg.dataType().array(...);

// 模式
Type.Array(baseDataTypeSchema, { minItems: size, maxItems: size });