filifa
/
mlblive
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
}
func handleUnexpectedClose(ws *statsapi.GamedayWebsocket, feedResp []byte, gamePk string) error {
ws.Close()
newWs, _, err := newWebsocket(gamePk)
func handleUnexpectedClose(gamePk string) (*statsapi.GamedayWebsocket, []byte, error) {
ws, _, err := newWebsocket(gamePk)
if err != nil {
return err
return nil, nil, err
}
*ws = *newWs
feedResp, err = statsapi.RequestFeed(gamePk)
return err
feedResp, err := statsapi.RequestFeed(gamePk)
return ws, feedResp, err
}
func subscribe(cmd *cobra.Command, args []string) {
@ -103,7 +100,7 @@ func subscribe(cmd *cobra.Command, args []string) {
err = ws.ReadJSON(&p)
if websocket.IsUnexpectedCloseError(err, statsapi.GameFinalCode, statsapi.GameUnavailableCode) {
log.Println(err)
err = handleUnexpectedClose(ws, feedResp, pkStr)
ws, feedResp, err = handleUnexpectedClose(pkStr)
if err != nil {
log.Fatal(err)
}