75 lines
1.5 KiB
JavaScript
75 lines
1.5 KiB
JavaScript
import Benchmark from "benchmark";
|
|
import { z } from "zod/v3";
|
|
const doubleSuite = new Benchmark.Suite("z.discriminatedUnion: double");
|
|
const manySuite = new Benchmark.Suite("z.discriminatedUnion: many");
|
|
const aSchema = z.object({
|
|
type: z.literal("a"),
|
|
});
|
|
const objA = {
|
|
type: "a",
|
|
};
|
|
const bSchema = z.object({
|
|
type: z.literal("b"),
|
|
});
|
|
const objB = {
|
|
type: "b",
|
|
};
|
|
const cSchema = z.object({
|
|
type: z.literal("c"),
|
|
});
|
|
const objC = {
|
|
type: "c",
|
|
};
|
|
const dSchema = z.object({
|
|
type: z.literal("d"),
|
|
});
|
|
const double = z.discriminatedUnion("type", [aSchema, bSchema]);
|
|
const many = z.discriminatedUnion("type", [aSchema, bSchema, cSchema, dSchema]);
|
|
doubleSuite
|
|
.add("valid: a", () => {
|
|
double.parse(objA);
|
|
})
|
|
.add("valid: b", () => {
|
|
double.parse(objB);
|
|
})
|
|
.add("invalid: null", () => {
|
|
try {
|
|
double.parse(null);
|
|
}
|
|
catch (_err) { }
|
|
})
|
|
.add("invalid: wrong shape", () => {
|
|
try {
|
|
double.parse(objC);
|
|
}
|
|
catch (_err) { }
|
|
})
|
|
.on("cycle", (e) => {
|
|
console.log(`${doubleSuite.name}: ${e.target}`);
|
|
});
|
|
manySuite
|
|
.add("valid: a", () => {
|
|
many.parse(objA);
|
|
})
|
|
.add("valid: c", () => {
|
|
many.parse(objC);
|
|
})
|
|
.add("invalid: null", () => {
|
|
try {
|
|
many.parse(null);
|
|
}
|
|
catch (_err) { }
|
|
})
|
|
.add("invalid: wrong shape", () => {
|
|
try {
|
|
many.parse({ type: "unknown" });
|
|
}
|
|
catch (_err) { }
|
|
})
|
|
.on("cycle", (e) => {
|
|
console.log(`${manySuite.name}: ${e.target}`);
|
|
});
|
|
export default {
|
|
suites: [doubleSuite, manySuite],
|
|
};
|