mirrors/nginx-ui
1
0
Fork 0
mirror of https://github.com/0xJacky/nginx-ui.git synced 2025-05-11 10:25:52 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

fix: implement context handling in analytic and processing functions for graceful shutdown

Browse source
This commit is contained in:
Jacky 2025-05-05 02:46:18 +00:00
parent 710b9f781e
commit fc507ccd3b
No known key found for this signature in database
GPG key ID: 215C21B10DF38B4D
4 changed files with 24 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
api/analytic/analytic.go
View file

@ -8,6 +8,7 @@ import (
"github.com/0xJacky/Nginx-UI/internal/analytic"
"github.com/0xJacky/Nginx-UI/internal/helper"
"github.com/0xJacky/Nginx-UI/internal/kernel"
"github.com/shirou/gopsutil/v4/cpu"
"github.com/shirou/gopsutil/v4/host"
"github.com/shirou/gopsutil/v4/load"
@ -91,7 +92,11 @@ func Analytic(c *gin.Context) {
break
}
time.Sleep(1 * time.Second)
select {
case <-kernel.Context.Done():
return
case <-time.After(1 * time.Second):
}
}
}

13
api/analytic/nodes.go
View file

@ -6,6 +6,7 @@ import (
"github.com/0xJacky/Nginx-UI/internal/analytic"
"github.com/0xJacky/Nginx-UI/internal/helper"
"github.com/0xJacky/Nginx-UI/internal/kernel"
"github.com/gin-gonic/gin"
"github.com/gorilla/websocket"
"github.com/uozi-tech/cosy/logger"
@ -36,7 +37,11 @@ func GetNodeStat(c *gin.Context) {
break
}
time.Sleep(10 * time.Second)
select {
case <-kernel.Context.Done():
return
case <-time.After(10 * time.Second):
}
}
}
@ -65,6 +70,10 @@ func GetNodesAnalytic(c *gin.Context) {
break
}
time.Sleep(10 * time.Second)
select {
case <-kernel.Context.Done():
return
case <-time.After(10 * time.Second):
}
}
}