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

View File

@ -2,8 +2,6 @@
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
// +build ignore
package main
import (
@ -20,6 +18,18 @@ import (
// 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")
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() {
flag.Parse()
log.SetFlags(0)
@ -38,17 +48,8 @@ func main() {
done := make(chan struct{})
go func() {
defer close(done)
for {
_, message, err := c.ReadMessage()
if err != nil {
log.Println("read:", err)
return
}
log.Printf("recv: %s", message)
}
}()
// handle inbound messages on the channel
go readConn(done, c)
ticker := time.NewTicker(time.Second)
defer ticker.Stop()