filifa
/
mlbstats
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

change func signature

This commit is contained in:
filifa 2024-07-20 15:30:05 -05:00
parent 2fe2b615c9
commit 9238a8e03b
1 changed files with 6 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
cmd/subscribe.go
View File

@ -63,17 +63,14 @@ func newWebsocket(gamePk string) (*statsapi.GamedayWebsocket, <-chan error, erro
return &ws, ch, err return &ws, ch, err
} }
func handleUnexpectedClose(ws *statsapi.GamedayWebsocket, feedResp []byte, gamePk string) error { func handleUnexpectedClose(gamePk string) (*statsapi.GamedayWebsocket, []byte, error) {
ws.Close() ws, _, err := newWebsocket(gamePk)
newWs, _, err := newWebsocket(gamePk)
if err != nil { if err != nil {
return err return nil, nil, err
} }
*ws = *newWs
feedResp, err = statsapi.RequestFeed(gamePk) feedResp, err := statsapi.RequestFeed(gamePk)
return err return ws, feedResp, err
} }
func subscribe(cmd *cobra.Command, args []string) { func subscribe(cmd *cobra.Command, args []string) {
@ -103,7 +100,7 @@ func subscribe(cmd *cobra.Command, args []string) {
err = ws.ReadJSON(&p) err = ws.ReadJSON(&p)
if websocket.IsUnexpectedCloseError(err, statsapi.GameFinalCode, statsapi.GameUnavailableCode) { if websocket.IsUnexpectedCloseError(err, statsapi.GameFinalCode, statsapi.GameUnavailableCode) {
log.Println(err) log.Println(err)
err = handleUnexpectedClose(ws, feedResp, pkStr) ws, feedResp, err = handleUnexpectedClose(pkStr)
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
} }