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

fix(notification): add context handling for graceful shutdown in Live function

Browse source
This commit is contained in:
Jacky 2025-05-05 02:51:19 +00:00
parent fc507ccd3b
commit 918ae3a6b6
No known key found for this signature in database
GPG key ID: 215C21B10DF38B4D
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
api/notification/live.go
View file

@ -1,12 +1,14 @@
package notification package notification
import ( import (
"io"
"time"
"github.com/0xJacky/Nginx-UI/api" "github.com/0xJacky/Nginx-UI/api"
"github.com/0xJacky/Nginx-UI/internal/kernel"
"github.com/0xJacky/Nginx-UI/internal/notification" "github.com/0xJacky/Nginx-UI/internal/notification"
"github.com/0xJacky/Nginx-UI/model" "github.com/0xJacky/Nginx-UI/model"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
"io"
"time"
) )
func Live(c *gin.Context) { func Live(c *gin.Context) {
@ -38,6 +40,8 @@ func Live(c *gin.Context) {
case <-notify: case <-notify:
notification.RemoveClient(c) notification.RemoveClient(c)
return return
case <-kernel.Context.Done():
return
} }
} }
} }