From ed2534b2b02262664e128d3cabd38a2f32a880a0 Mon Sep 17 00:00:00 2001
From: Arne Bahlo <arne@axiom.co>
Date: Thu, 9 Jun 2022 17:18:50 +0200
Subject: [PATCH] Use waitUntil, log.flush() in middleware, function

---
 examples/logger/pages/_middleware.js | 3 ++-
 examples/logger/pages/api/hello.js   | 1 +
 src/logger.ts                        | 4 +---
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/examples/logger/pages/_middleware.js b/examples/logger/pages/_middleware.js
index 89a942d1..75e05496 100644
--- 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()
 }
\ No newline at end of file
diff --git a/examples/logger/pages/api/hello.js b/examples/logger/pages/api/hello.js
index 62145ab7..a57c9d81 100644
--- 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
index 2be4e848..f33fed34 100644
--- 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() {