From c33000045a1e08e299a905abe7eda938bcba3fe8 Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Thu, 26 Sep 2024 17:23:24 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E6=B6=88=E6=81=AF=E8=BF=9B=E8=A1=8Cwg?=
 =?UTF-8?q?=E5=85=B3=E9=97=AD=E9=94=80=E6=AF=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 public/wiregasm/worker.js                     | 3 +++
 src/views/traceManage/tshark/hooks/usePCAP.ts | 6 +++---
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/public/wiregasm/worker.js b/public/wiregasm/worker.js
index 1d650b5a..7b5872af 100644
--- a/public/wiregasm/worker.js
+++ b/public/wiregasm/worker.js
@@ -63,6 +63,9 @@ function replacer(key, value) {
 this.onmessage = ev => {
   const data = ev.data;
   switch (data.type) {
+    case 'close':
+      wg.destroy();
+      break;
     case 'columns':
       const columns = wg.columns();
       if (Array.isArray(columns)) {
diff --git a/src/views/traceManage/tshark/hooks/usePCAP.ts b/src/views/traceManage/tshark/hooks/usePCAP.ts
index 0d40a51f..83e9b490 100644
--- a/src/views/traceManage/tshark/hooks/usePCAP.ts
+++ b/src/views/traceManage/tshark/hooks/usePCAP.ts
@@ -193,8 +193,8 @@ export function usePCAP() {
       stop_time: 0,
       elapsed_time: 0,
     };
-    state.finishedProcessing = false; 
-    
+    state.finishedProcessing = false;
+
     wk.send({ type: 'process', file: file });
   }
 
@@ -300,7 +300,7 @@ export function usePCAP() {
   });
 
   onBeforeUnmount(() => {
-    wk.close();
+    wk.send({ type: 'close' }) && wk.close();
   });
 
   return {