jcarr
/
dnssecsocket
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

make a seperate function to handle reading messages from the channel 

Signed-off-by: Jeff Carr <jcarr@wit.com>
This commit is contained in:
Jeff Carr 2019-05-11 17:07:28 -07:00
parent 65f0d84f5b
commit 46ee8a456f
1 changed files with 14 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
example-gorilla/client.go
View File

@ -2,8 +2,6 @@
// Use of this source code is governed by a BSD-style // Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file. // license that can be found in the LICENSE file.
// +build ignore
package main package main
import ( import (
@ -20,6 +18,18 @@ import (
// var addr = flag.String("addr", "localhost:8080", "http service address") // var addr = flag.String("addr", "localhost:8080", "http service address")
var addr = flag.String("addr", "v000185.testing.com.customers.wprod.wit.com:9000", "http service address") var addr = flag.String("addr", "v000185.testing.com.customers.wprod.wit.com:9000", "http service address")
func readConn(done chan struct{}, conn *websocket.Conn) {
defer close(done)
for {
_, message, err := conn.ReadMessage()
if err != nil {
log.Println("read:", err)
return
}
log.Printf("recv: %s", message)
}
}
func main() { func main() {
flag.Parse() flag.Parse()
log.SetFlags(0) log.SetFlags(0)
@ -38,17 +48,8 @@ func main() {
done := make(chan struct{}) done := make(chan struct{})
go func() { // handle inbound messages on the channel
defer close(done) go readConn(done, c)
for {
_, message, err := c.ReadMessage()
if err != nil {
log.Println("read:", err)
return
}
log.Printf("recv: %s", message)
}
}()
ticker := time.NewTicker(time.Second) ticker := time.NewTicker(time.Second)
defer ticker.Stop() defer ticker.Stop()