fluencelabs/tendermint
1
0
Fork 0
mirror of https://github.com/fluencelabs/tendermint synced 2025-07-02 22:21:44 +00:00
Code Issues Projects Releases Wiki Activity

Clean up event switch add helper function

Browse Source
This commit is contained in:
Ethan Frey
2017-02-24 19:59:40 +01:00
parent 175bb329e4
commit 6282fad518
4 changed files with 89 additions and 85 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
rpc/client/httpclient.go
View File

@ -173,6 +173,7 @@ type WSEvents struct {
endpoint string
ws *rpcclient.WSClient
quit chan bool
done chan bool
}
func newWSEvents(remote, endpoint string) *WSEvents {
@ -181,6 +182,7 @@ func newWSEvents(remote, endpoint string) *WSEvents {
endpoint: endpoint,
remote: remote,
quit: make(chan bool, 1),
done: make(chan bool, 1),
}
}
@ -211,7 +213,9 @@ func (w *WSEvents) Stop() bool {
if stop {
// send a message to quit to stop the eventListener
w.quit <- true
<-w.done
w.ws.Stop()
w.ws = nil
}
return stop
}
@ -249,7 +253,9 @@ func (w *WSEvents) eventListener() {
// FIXME: better logging/handling of errors??
fmt.Printf("ws err: %+v\n", err)
case <-w.quit:
// only way to finish this method
// send a message so we can wait for the routine to exit
// before cleaning up the w.ws stuff
w.done <- true
return
}
}