nnet/internal/conn_ws.go

package internal

import (
	//	"fmt"

	"github.com/gorilla/websocket"
)

type WsConn struct {
	*websocket.Conn
}

func NewWsConn(c *websocket.Conn) *WsConn {
	return &WsConn{c}
}

func (self *WsConn) Read(p []byte) (int, error) {
	_, r, err := self.NextReader()
	if nil != err {
		return 0, err
	}
	return r.Read(p)
}

func (self *WsConn) Write(p []byte) (int, error) {
	w, err := self.NextWriter(websocket.BinaryMessage)
	if nil != err {
		return 0, err
	}
	n, err := w.Write(p)
	_ = w.Close()
	return n, err
}