Merge pull request #7 from axiomhq/arne/fix-example

Use waitUntil, log.flush() in middleware, function
axiomhq · Jun 9, 2022 · 1162737 · 1162737 · vercel · Jun 9, 2022
2 parents bca8dfb + ed2534b
commit 1162737
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/examples/logger/pages/_middleware.js b/examples/logger/pages/_middleware.js
@@ -1,7 +1,8 @@
 import { NextResponse } from 'next/server'
 import { log } from 'next-axiom'
 
-export async function middleware(_req, _ev) {
+export async function middleware(_req, ev) {
   log.info("Hello from middleware", { 'bar': 'baz' });
+  ev.waitUntil(log.flush())
   return NextResponse.next()
 }
diff --git a/examples/logger/pages/api/hello.js b/examples/logger/pages/api/hello.js
@@ -3,5 +3,6 @@ import { log } from 'next-axiom'
 
 export default async function handler(req, res) {
   log.info('Hello from function', { url: req.url });
+  await log.flush()
   res.status(200).json({ name: 'John Doe' })
 }
diff --git a/src/logger.ts b/src/logger.ts
@@ -25,9 +25,7 @@ export const log = {
   info: (message: string, args: any = {}) => _log('info', message, args),
   warn: (message: string, args: any = {}) => _log('warn', message, args),
   error: (message: string, args: any = {}) => _log('error', message, args),
-  flush: async () => {
-    await sendLogs();
-  },
+  flush: sendLogs,
 };
 
 async function sendLogs() {