mirror of
https://github.com/yusing/godoxy.git
synced 2026-03-30 05:41:50 +02:00
feat(websocket): implement Reader for reading binary data from the manager
- Removed Close method from Writer
This commit is contained in:
25
internal/net/gphttp/websocket/reader.go
Normal file
25
internal/net/gphttp/websocket/reader.go
Normal file
@@ -0,0 +1,25 @@
|
||||
package websocket
|
||||
|
||||
import (
|
||||
"io"
|
||||
"time"
|
||||
)
|
||||
|
||||
type Reader struct {
|
||||
manager *Manager
|
||||
}
|
||||
|
||||
func (m *Manager) NewReader() io.Reader {
|
||||
return &Reader{
|
||||
manager: m,
|
||||
}
|
||||
}
|
||||
|
||||
func (r *Reader) Read(p []byte) (int, error) {
|
||||
data, err := r.manager.ReadBinary(10 * time.Second)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
copy(p, data)
|
||||
return len(data), nil
|
||||
}
|
||||
@@ -20,7 +20,3 @@ func (cm *Manager) NewWriter(msgType int) io.Writer {
|
||||
func (w *Writer) Write(p []byte) (int, error) {
|
||||
return len(p), w.manager.WriteData(w.msgType, p, 10*time.Second)
|
||||
}
|
||||
|
||||
func (w *Writer) Close() error {
|
||||
return w.manager.conn.Close()
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user