send back a protobuf

Signed-off-by: Jeff Carr <jcarr@wit.com>
This commit is contained in:
Jeff Carr 2019-05-12 07:28:07 +00:00
parent 1bfaaa331d
commit 0b6d5fbac9
1 changed files with 13 additions and 2 deletions

View File

@ -19,7 +19,19 @@ var upgrader = websocket.Upgrader{
},
}
func sendProtobuf(conn *websocket.Conn) {
msg := pb.CreateSampleEvent()
msg.Name = "test echo over gorilla websocket"
data, _ := proto.Marshal(msg)
err := conn.WriteMessage(websocket.BinaryMessage, data)
if err != nil {
log.Println("sendProtobuf() write:", err)
}
}
func eventHandler(w http.ResponseWriter, r *http.Request) {
log.Println("echoHandler() START")
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Println("Upgrade:", err)
@ -27,8 +39,6 @@ func eventHandler(w http.ResponseWriter, r *http.Request) {
}
defer conn.Close()
log.Println("echoHandler() START")
for {
mytype, message, err := conn.ReadMessage()
if err != nil {
@ -49,6 +59,7 @@ func eventHandler(w http.ResponseWriter, r *http.Request) {
log.Printf("recv binary: %s", pdata)
if pdata.Type == pb.Event_MIGRATE {
log.Printf("GOT MIGRATE")
sendProtobuf(conn)
}
}
}