refactor: minor

hi-ogawa · Apr 9, 2024 · dba45bc · dba45bc
1 parent ced97d9
commit dba45bc
Showing 1 changed file with 9 additions and 10 deletions.
diff --git a/examples/workerd/src/plugin.ts b/examples/workerd/src/plugin.ts
@@ -47,16 +47,15 @@ export function vitePluginWorkerd(pluginOptions: WorkerdPluginOptions): Plugin {
     },
 
     configureServer(server) {
-      return async () => {
-        const devEnv = server.environments["workerd"] as WorkerdDevEnvironment;
-        server.middlewares.use(
-          createMiddleware(
-            (ctx) => devEnv.api.dispatchFetch(pluginOptions.entry, ctx.request),
-            {
-              alwaysCallNext: false,
-            },
-          ),
-        );
+      const devEnv = server.environments["workerd"] as WorkerdDevEnvironment;
+      const nodeMiddleware = createMiddleware(
+        (ctx) => devEnv.api.dispatchFetch(pluginOptions.entry, ctx.request),
+        {
+          alwaysCallNext: false,
+        },
+      );
+      return () => {
+        server.middlewares.use(nodeMiddleware);
       };
     },
   };