websocket.go

package protocol

import (
	"github.com/feel-easy/hole/utils"
	"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{}
	utils.UnmarshalJson(b, msg)
	return msg, nil
}

func (w WebsocketReadWriteCloser) Write(msg Packet) error {
	return w.conn.WriteMessage(websocket.TextMessage, utils.MarshalJson(msg))
}

func (w WebsocketReadWriteCloser) Close() error {
	return w.conn.Close()
}

func (w WebsocketReadWriteCloser) IP() string {
	return w.conn.RemoteAddr().String()
}