add exceptions for finished games

2024-07-20 15:04:19 -05:00 · 2024-07-20 15:04:19 -05:00 · be70ee7330
parent 5e471f5c2c
commit be70ee7330
2 changed files with 6 additions and 1 deletions
--- a/cmd/internal/statsapi/websocket.go
+++ b/cmd/internal/statsapi/websocket.go
@ -26,6 +26,11 @@ type Push struct {
 	UpdateId string
 }

+const (
+	GameFinalCode = 4200
+	GameUnavailableCode = 4400
+)
+
 type GamedayWebsocket struct {
 	baseURL url.URL
 	*websocket.Conn
--- a/cmd/subscribe.go
+++ b/cmd/subscribe.go
@ -88,7 +88,7 @@ func subscribe(cmd *cobra.Command, args []string) {

 		var p statsapi.Push
 		err = ws.ReadJSON(&p)
-		if websocket.IsUnexpectedCloseError(err) {
+		if websocket.IsUnexpectedCloseError(err, statsapi.GameFinalCode, statsapi.GameUnavailableCode) {
 			log.Println(err)
 			ws.Close()