mirror of
https://github.com/tiennm99/gomoku.git
synced 2026-06-02 22:11:40 +00:00
tiennm99
6e3670aec8
Import source files from ratel-online organization repos into monorepo structure for gomoku game development. Update README with project structure and credits for original authors.
37 lines
793 B
Go
37 lines
793 B
Go
package protocol
|
|
|
|
import (
|
|
"github.com/awesome-cap/im/core/util/json"
|
|
"github.com/gorilla/websocket"
|
|
)
|
|
|
|
type WebsocketReadWriteCloser struct {
|
|
conn *websocket.Conn
|
|
}
|
|
|
|
func NewWebsocketReadWriteCloser(conn *websocket.Conn) WebsocketReadWriteCloser {
|
|
return WebsocketReadWriteCloser{conn: conn}
|
|
}
|
|
|
|
func (w WebsocketReadWriteCloser) Read() (*Packet, error) {
|
|
_, b, err := w.conn.ReadMessage()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
msg := &Packet{}
|
|
json.Unmarshal(b, msg)
|
|
return msg, nil
|
|
}
|
|
|
|
func (w WebsocketReadWriteCloser) Write(msg Packet) error {
|
|
return w.conn.WriteMessage(websocket.BinaryMessage, json.Marshal(msg))
|
|
}
|
|
|
|
func (w WebsocketReadWriteCloser) Close() error {
|
|
return w.conn.Close()
|
|
}
|
|
|
|
func (w WebsocketReadWriteCloser) IP() string {
|
|
return w.conn.RemoteAddr().String()
|
|
}
|