Buriburizaemon/nezha_domains
1
0
Fork 0
mirror of https://github.com/Buriburizaem0n/nezha_domains.git synced 2026-02-04 12:40:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

prevent writing response to websocket connections (#457)

Browse Source
This commit is contained in:
UUBulb
2024-10-25 21:45:05 +08:00
committed by GitHub
parent d086e98711
commit 380973a200
4 changed files with 33 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
cmd/dashboard/controller/terminal.go
View File

@@ -66,6 +66,7 @@ func createTerminal(c *gin.Context) (*model.CreateTerminalResponse, error) {
// @Description Terminal stream
// @Tags auth required
// @Param id path string true "Stream UUID"
// @Success 200 {object} model.CommonResponse[any]
// @Router /ws/terminal/{id} [get]
func terminalStream(c *gin.Context) (any, error) {
streamId := c.Param("id")
@@ -92,8 +93,8 @@ func terminalStream(c *gin.Context) (any, error) {
}()
if err = rpc.NezhaHandlerSingleton.UserConnected(streamId, conn); err != nil {
return nil, err
return nil, newWsError("%v", err)
}
return nil, rpc.NezhaHandlerSingleton.StartStream(streamId, time.Second*10)
return nil, newWsError("%v", rpc.NezhaHandlerSingleton.StartStream(streamId, time.Second*10))
}