diff --git a/app/worker.go b/app/worker.go index fbc6226..5984e08 100644 --- a/app/worker.go +++ b/app/worker.go @@ -173,6 +173,8 @@ func xWorker(workerData Info, u url.URL) { defer leave.Stop() isLeave := false + var income int64 + income = 0 for { @@ -193,7 +195,7 @@ func xWorker(workerData Info, u url.URL) { _, message, err := c.ReadMessage() if err != nil { fmt.Println(err.Error(), workerData.room) - if workerData.Income > 1 && websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway) { + if income > 1 && websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway) { go reconnectRoom(workerData) } return @@ -291,6 +293,8 @@ func xWorker(workerData Info, u url.URL) { continue } + income += donate.Amount + if isLeave { leave.Reset(60 * 10 * time.Second) }