78 lines
1.1 KiB
Go
78 lines
1.1 KiB
Go
package server
|
|
|
|
import (
|
|
"io"
|
|
"time"
|
|
|
|
"github.com/gorilla/websocket"
|
|
)
|
|
|
|
const (
|
|
PongWait = 10 * time.Second
|
|
PingWait = 5 * time.Second
|
|
)
|
|
|
|
type ProfileReq string
|
|
|
|
type ProfileRes string
|
|
|
|
type ReadWriteCloser struct {
|
|
ws *websocket.Conn
|
|
r io.Reader
|
|
w io.WriteCloser
|
|
}
|
|
|
|
func (rwc *ReadWriteCloser) Read(p []byte) (n int, err error) {
|
|
if rwc.r == nil {
|
|
var _ int
|
|
_, rwc.r, err = rwc.ws.NextReader()
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
}
|
|
for n = 0; n < len(p); {
|
|
var m int
|
|
m, err = rwc.r.Read(p[n:])
|
|
n += m
|
|
if err == io.EOF {
|
|
// done
|
|
rwc.r = nil
|
|
break
|
|
}
|
|
// ???
|
|
if err != nil {
|
|
break
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func (rwc *ReadWriteCloser) Write(p []byte) (n int, err error) {
|
|
if rwc.w == nil {
|
|
rwc.w, err = rwc.ws.NextWriter(websocket.TextMessage)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
}
|
|
for n = 0; n < len(p); {
|
|
var m int
|
|
m, err = rwc.w.Write(p)
|
|
n += m
|
|
if err != nil {
|
|
break
|
|
}
|
|
}
|
|
if err != nil || n == len(p) {
|
|
err = rwc.Close()
|
|
}
|
|
return
|
|
}
|
|
|
|
func (rwc *ReadWriteCloser) Close() (err error) {
|
|
if rwc.w != nil {
|
|
err = rwc.w.Close()
|
|
rwc.w = nil
|
|
}
|
|
return err
|
|
}
|