32 lines
483 B
Go
32 lines
483 B
Go
package server
|
|
|
|
import (
|
|
"github.com/khlieng/name_pending/Godeps/_workspace/src/github.com/gorilla/websocket"
|
|
)
|
|
|
|
type WebSocket struct {
|
|
conn *websocket.Conn
|
|
|
|
Out chan []byte
|
|
}
|
|
|
|
func NewWebSocket(ws *websocket.Conn) *WebSocket {
|
|
return &WebSocket{
|
|
conn: ws,
|
|
Out: make(chan []byte, 32),
|
|
}
|
|
}
|
|
|
|
func (w *WebSocket) write() {
|
|
for {
|
|
err := w.conn.WriteMessage(websocket.TextMessage, <-w.Out)
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func (w *WebSocket) close() {
|
|
close(w.Out)
|
|
}
|