65 lines
1.1 KiB
Go
65 lines
1.1 KiB
Go
package packet
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"io/ioutil"
|
|
|
|
"github.com/googollee/go-socket.io/engineio/frame"
|
|
)
|
|
|
|
type fakeConnReader struct {
|
|
frames []Frame
|
|
}
|
|
|
|
func NewFakeConnReader(frames []Frame) *fakeConnReader {
|
|
return &fakeConnReader{
|
|
frames: frames,
|
|
}
|
|
}
|
|
|
|
func (r *fakeConnReader) NextReader() (frame.Type, io.ReadCloser, error) {
|
|
if len(r.frames) == 0 {
|
|
return frame.String, nil, io.EOF
|
|
}
|
|
f := r.frames[0]
|
|
r.frames = r.frames[1:]
|
|
return f.FType, ioutil.NopCloser(bytes.NewReader(f.Data)), nil
|
|
}
|
|
|
|
type fakeOneFrameConst struct {
|
|
b byte
|
|
}
|
|
|
|
func (c *fakeOneFrameConst) Read(p []byte) (int, error) {
|
|
p[0] = c.b
|
|
return 1, nil
|
|
}
|
|
|
|
type fakeConstReader struct {
|
|
ft frame.Type
|
|
r *fakeOneFrameConst
|
|
}
|
|
|
|
func NewFakeConstReader() *fakeConstReader {
|
|
return &fakeConstReader{
|
|
ft: frame.String,
|
|
r: &fakeOneFrameConst{
|
|
b: MESSAGE.StringByte(),
|
|
},
|
|
}
|
|
}
|
|
|
|
func (r *fakeConstReader) NextReader() (frame.Type, io.ReadCloser, error) {
|
|
ft := r.ft
|
|
switch ft {
|
|
case frame.Binary:
|
|
r.ft = frame.String
|
|
r.r.b = MESSAGE.StringByte()
|
|
case frame.String:
|
|
r.ft = frame.Binary
|
|
r.r.b = MESSAGE.BinaryByte()
|
|
}
|
|
return ft, ioutil.NopCloser(r.r), nil
|
|
}
|