mirror of
https://github.com/yusing/godoxy.git
synced 2026-04-23 00:38:33 +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) {
|
func (w *Writer) Write(p []byte) (int, error) {
|
||||||
return len(p), w.manager.WriteData(w.msgType, p, 10*time.Second)
|
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