From 55ed67eaed6d0ebd3421f717f2f23d8daa52e971 Mon Sep 17 00:00:00 2001
From: zhengkunwang223 <31820853+zhengkunwang223@users.noreply.github.com>
Date: Mon, 10 Jul 2023 22:43:12 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E4=BC=98=E5=8C=96=E8=BF=9B=E7=A8=8B?=
 =?UTF-8?q?=E7=AE=A1=E7=90=86=E7=9A=84=E9=A1=B5=E9=9D=A2=E6=89=93=E5=BC=80?=
 =?UTF-8?q?=E9=80=9F=E5=BA=A6=20(#1595)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 backend/utils/websocket/process_data.go       | 62 ++++++++++++++++---
 .../src/views/host/process/process/index.vue  | 16 +++--
 2 files changed, 64 insertions(+), 14 deletions(-)

diff --git a/backend/utils/websocket/process_data.go b/backend/utils/websocket/process_data.go
index b71c5b98e..9f83db33f 100644
--- a/backend/utils/websocket/process_data.go
+++ b/backend/utils/websocket/process_data.go
@@ -8,7 +8,9 @@ import (
 	"github.com/shirou/gopsutil/v3/host"
 	"github.com/shirou/gopsutil/v3/net"
 	"github.com/shirou/gopsutil/v3/process"
+	"sort"
 	"strings"
+	"sync"
 	"time"
 )
 
@@ -82,6 +84,20 @@ type processConnect struct {
 	Name   string   `json:"name"`
 }
 
+type ProcessConnects []processConnect
+
+func (p ProcessConnects) Len() int {
+	return len(p)
+}
+
+func (p ProcessConnects) Less(i, j int) bool {
+	return p[i].PID < p[j].PID
+}
+
+func (p ProcessConnects) Swap(i, j int) {
+	p[i], p[j] = p[j], p[i]
+}
+
 type sshSession struct {
 	Username  string `json:"username"`
 	PID       int32  `json:"PID"`
@@ -163,20 +179,25 @@ func formatBytes(bytes uint64) string {
 }
 
 func getProcessData(processConfig PsProcessConfig) (res []byte, err error) {
-	var (
-		result    []PsProcessData
-		processes []*process.Process
-	)
+	var processes []*process.Process
 	processes, err = process.Processes()
 	if err != nil {
 		return
 	}
-	for _, proc := range processes {
+
+	var (
+		result      []PsProcessData
+		resultMutex sync.Mutex
+		wg          sync.WaitGroup
+		numWorkers  = 4
+	)
+
+	handleData := func(proc *process.Process) {
 		procData := PsProcessData{
 			PID: proc.Pid,
 		}
 		if processConfig.Pid > 0 && processConfig.Pid != proc.Pid {
-			continue
+			return
 		}
 		if procName, err := proc.Name(); err == nil {
 			procData.Name = procName
@@ -184,13 +205,13 @@ func getProcessData(processConfig PsProcessConfig) (res []byte, err error) {
 			procData.Name = "<UNKNOWN>"
 		}
 		if processConfig.Name != "" && !strings.Contains(procData.Name, processConfig.Name) {
-			continue
+			return
 		}
 		if username, err := proc.Username(); err == nil {
 			procData.Username = username
 		}
 		if processConfig.Username != "" && !strings.Contains(procData.Username, processConfig.Username) {
-			continue
+			return
 		}
 		procData.PPID, _ = proc.Ppid()
 		statusArray, _ := proc.Status()
@@ -251,8 +272,33 @@ func getProcessData(processConfig PsProcessConfig) (res []byte, err error) {
 		procData.OpenFiles, _ = proc.OpenFiles()
 		procData.Envs, _ = proc.Environ()
 
+		resultMutex.Lock()
 		result = append(result, procData)
+		resultMutex.Unlock()
 	}
+
+	chunkSize := (len(processes) + numWorkers - 1) / numWorkers
+	for i := 0; i < numWorkers; i++ {
+		wg.Add(1)
+		start := i * chunkSize
+		end := (i + 1) * chunkSize
+		if end > len(processes) {
+			end = len(processes)
+		}
+
+		go func(start, end int) {
+			defer wg.Done()
+			for j := start; j < end; j++ {
+				handleData(processes[j])
+			}
+		}(start, end)
+	}
+
+	wg.Wait()
+
+	sort.Slice(result, func(i, j int) bool {
+		return result[i].PID < result[j].PID
+	})
 	res, err = json.Marshal(result)
 	return
 }
diff --git a/frontend/src/views/host/process/process/index.vue b/frontend/src/views/host/process/process/index.vue
index 55ab05798..11176d2c4 100644
--- a/frontend/src/views/host/process/process/index.vue
+++ b/frontend/src/views/host/process/process/index.vue
@@ -169,6 +169,7 @@ const loading = ref(false);
 const tableRef = ref();
 const oldData = ref([]);
 const detailRef = ref();
+const isGetData = ref(true);
 
 const openDetail = (row: any) => {
     detailRef.value.acceptParams({ info: row });
@@ -213,10 +214,14 @@ const closeSocket = () => {
     }
 };
 
-const onOpenProcess = () => {};
+const onOpenProcess = () => {
+    loading.value = true;
+    isGetData.value = true;
+    processSocket.send(JSON.stringify(processSearch));
+};
 const onMessage = (message: any) => {
-    let result: any[] = JSON.parse(message.data);
-    oldData.value = result;
+    isGetData.value = false;
+    oldData.value = JSON.parse(message.data);
     data.value = filterByStatus();
     sortTable();
     loading.value = false;
@@ -253,8 +258,6 @@ const initProcess = () => {
     processSocket.onmessage = onMessage;
     processSocket.onerror = onerror;
     processSocket.onclose = onClose;
-    loading.value = true;
-    search();
     sendMsg();
 };
 
@@ -265,7 +268,8 @@ const sendMsg = () => {
 };
 
 const search = () => {
-    if (isWsOpen()) {
+    if (isWsOpen() && !isGetData.value) {
+        isGetData.value = true;
         if (typeof processSearch.pid === 'string') {
             processSearch.pid = undefined;
         }