add onError handler

This commit is contained in:
Robert Soriano
2022-05-18 09:09:11 -07:00
parent 83624e0ca8
commit 3eda32d33a
3 changed files with 13 additions and 40 deletions

View File

@@ -11,6 +11,7 @@ import { createURL } from 'ufo'
import type { CompatibilityEvent } from 'h3'
import { defineEventHandler, isMethod, useBody } from 'h3'
import type { TRPCResponse } from '@trpc/server/dist/declarations/src/rpc'
import type { OnErrorFunction } from '@trpc/server/dist/declarations/src/internals/OnErrorFunction'
type MaybePromise<T> = T | Promise<T>
@@ -28,17 +29,18 @@ export function createTRPCHandler<Router extends AnyRouter>({
router,
createContext,
responseMeta,
onError,
}: {
router: Router
createContext?: CreateContextFn<Router>
responseMeta?: ResponseMetaFn<Router>
onError?: OnErrorFunction<Router, CompatibilityEvent['req']>
}) {
const url = '/trpc'
return defineEventHandler(async (event) => {
const {
req,
res,
context,
} = event
const $url = createURL(req.url)
@@ -51,9 +53,15 @@ export function createTRPCHandler<Router extends AnyRouter>({
body: isMethod(event, 'GET') ? null : await useBody(event),
query: $url.searchParams,
},
path: $url.pathname.substring(url.length + 1),
path: context.params.path,
createContext: async () => createContext?.(event),
responseMeta,
onError: (o) => {
onError?.({
...o,
req,
})
},
})
const { status, headers, body } = httpResponse