astra/go-mastodon
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix WebSocket close

Browse source
This commit is contained in:
178inaba 2017-04-30 22:14:02 +09:00
parent 70b3dbf548
commit fc574ac209
2 changed files with 14 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
streaming_ws.go
View file

@ -66,6 +66,7 @@ func (c *WSClient) streamingWS(ctx context.Context, stream, tag string) (chan Ev
q := make(chan Event)
go func() {
defer close(q)
for {
err := c.handleWS(ctx, u.String(), q)
if err != nil {
@ -85,7 +86,12 @@ func (c *WSClient) handleWS(ctx context.Context, rawurl string, q chan Event) er
// End.
return err
}
defer conn.Close()
// Close the WebSocket when the context is canceled.
go func() {
<-ctx.Done()
conn.Close()
}()
for {
select {