41 lines
648 B
Go
41 lines
648 B
Go
|
package packet
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
|
||
|
|
"github.com/googollee/go-socket.io/engineio/frame"
|
||
|
|
)
|
||
|
|
|
||
|
|
type fakeFrame struct {
|
||
|
|
w *fakeConnWriter
|
||
|
|
typ frame.Type
|
||
|
|
data *bytes.Buffer
|
||
|
|
}
|
||
|
|
|
||
|
|
func newFakeFrame(w *fakeConnWriter, fType frame.Type) *fakeFrame {
|
||
|
|
return &fakeFrame{
|
||
|
|
w: w,
|
||
|
|
typ: fType,
|
||
|
|
data: bytes.NewBuffer(nil),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (w *fakeFrame) Write(p []byte) (int, error) {
|
||
|
|
return w.data.Write(p)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (w *fakeFrame) Read(p []byte) (int, error) {
|
||
|
|
return w.data.Read(p)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (w *fakeFrame) Close() error {
|
||
|
|
if w.w == nil {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
w.w.Frames = append(w.w.Frames, Frame{
|
||
|
|
FType: w.typ,
|
||
|
|
Data: w.data.Bytes(),
|
||
|
|
})
|
||
|
|
return nil
|
||
|
|
}
|