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

enhance: handle websocket error

Browse source
This commit is contained in:
Jacky 2024-07-20 18:35:06 +08:00
parent d6b22c157c
commit ccb04c07d8
No known key found for this signature in database
GPG key ID: 215C21B10DF38B4D
8 changed files with 126 additions and 60 deletions
Download patch file Download diff file Expand all files Collapse all files

73
api/system/upgrade.go
View file

@ -41,6 +41,18 @@ func GetCurrentVersion(c *gin.Context) {
c.JSON(http.StatusOK, curVer)
}
const (
UpgradeStatusInfo = "info"
UpgradeStatusError = "error"
UpgradeStatusProgress = "progress"
)
type CoreUpgradeResp struct {
Status string `json:"status"`
Progress float64 `json:"progress"`
Message string `json:"message"`
}
func PerformCoreUpgrade(c *gin.Context) {
var upGrader = websocket.Upgrader{
CheckOrigin: func(r *http.Request) bool {
@ -67,49 +79,48 @@ func PerformCoreUpgrade(c *gin.Context) {
return
}
_ = ws.WriteJSON(gin.H{
"status": "info",
"message": "Initialing core upgrader",
_ = ws.WriteJSON(CoreUpgradeResp{
Status: UpgradeStatusInfo,
Message: "Initialing core upgrader",
})
u, err := upgrader.NewUpgrader(control.Channel)
if err != nil {
_ = ws.WriteJSON(gin.H{
"status": "error",
"message": "Initial core upgrader error",
_ = ws.WriteJSON(CoreUpgradeResp{
Status: UpgradeStatusError,
Message: "Initial core upgrader error",
})
_ = ws.WriteJSON(gin.H{
"status": "error",
"message": err.Error(),
_ = ws.WriteJSON(CoreUpgradeResp{
Status: UpgradeStatusError,
Message: err.Error(),
})
logger.Error(err)
return
}
_ = ws.WriteJSON(gin.H{
"status": "info",
"message": "Downloading latest release",
_ = ws.WriteJSON(CoreUpgradeResp{
Status: UpgradeStatusInfo,
Message: "Downloading latest release",
})
progressChan := make(chan float64)
go func() {
for progress := range progressChan {
_ = ws.WriteJSON(gin.H{
"status": "progress",
"progress": progress,
_ = ws.WriteJSON(CoreUpgradeResp{
Status: UpgradeStatusProgress,
Progress: progress,
})
}
}()
tarName, err := u.DownloadLatestRelease(progressChan)
if err != nil {
_ = ws.WriteJSON(gin.H{
"status": "error",
"message": "Download latest release error",
_ = ws.WriteJSON(CoreUpgradeResp{
Status: UpgradeStatusError,
Message: "Download latest release error",
})
_ = ws.WriteJSON(gin.H{
"status": "error",
"message": err.Error(),
_ = ws.WriteJSON(CoreUpgradeResp{
Status: UpgradeStatusError,
Message: err.Error(),
})
logger.Error(err)
return
@ -119,9 +130,9 @@ func PerformCoreUpgrade(c *gin.Context) {
_ = os.Remove(tarName)
_ = os.Remove(tarName + ".digest")
}()
_ = ws.WriteJSON(gin.H{
"status": "info",
"message": "Performing core upgrade",
_ = ws.WriteJSON(CoreUpgradeResp{
Status: UpgradeStatusInfo,
Message: "Performing core upgrade",
})
// dry run
if control.DryRun || settings.ServerSettings.Demo {
@ -132,13 +143,13 @@ func PerformCoreUpgrade(c *gin.Context) {
// bye, overseer will restart nginx-ui
err = u.PerformCoreUpgrade(u.ExPath, tarName)
if err != nil {
_ = ws.WriteJSON(gin.H{
"status": "error",
"message": "Perform core upgrade error",
_ = ws.WriteJSON(CoreUpgradeResp{
Status: UpgradeStatusError,
Message: "Perform core upgrade error",
})
_ = ws.WriteJSON(gin.H{
"status": "error",
"message": err.Error(),
_ = ws.WriteJSON(CoreUpgradeResp{
Status: UpgradeStatusError,
Message: err.Error(),
})
logger.Error(err)
return