From c83eaf09b9fa27c961c845c5eddc075384b244fc Mon Sep 17 00:00:00 2001
From: Jarred Sumner <709451+Jarred-Sumner@users.noreply.github.com>
Date: Wed, 15 Feb 2023 00:36:01 -0800
Subject: some cleanup

---
 src/bun.js/async_hooks.exports.js | 20 +++++++-------------
 1 file changed, 7 insertions(+), 13 deletions(-)

(limited to 'src/bun.js/async_hooks.exports.js')

diff --git a/src/bun.js/async_hooks.exports.js b/src/bun.js/async_hooks.exports.js
index 9dfff7ef1..c5bc6f620 100644
--- a/src/bun.js/async_hooks.exports.js
+++ b/src/bun.js/async_hooks.exports.js
@@ -3,18 +3,11 @@ var drainMicrotasks = () => {
   drainMicrotasks();
 };
 
-const warnOnce = fn => {
-  let warned = false;
-  return (...args) => {
-    if (!warned) {
-      warned = true;
-      fn(...args);
-    }
-  };
+var notImplemented = () => {
+  console.warn("[bun]: async_hooks has not been implemented yet :(");
+  notImplemented = () => {};
 };
 
-const notImplemented = warnOnce(() => console.warn("[bun]: async_hooks has not been implemented yet :("));
-
 class AsyncLocalStorage {
   #store;
   _enabled;
@@ -39,9 +32,9 @@ class AsyncLocalStorage {
 
   run(store, callback, ...args) {
     if (typeof callback !== "function") throw new TypeError("ERR_INVALID_CALLBACK");
-    const prev = this.#store;
+    var prev = this.#store;
     var result, err;
-    process.nextTick(() => {
+    process.nextTick(store => {
       this.enterWith(store);
       try {
         result = callback(...args);
@@ -49,8 +42,9 @@ class AsyncLocalStorage {
         err = e;
       } finally {
         this.#store = prev;
+        prev = undefined;
       }
-    });
+    }, store);
     drainMicrotasks();
     if (typeof err !== "undefined") {
       throw err;
-- 
cgit v1.2.3