mirror of https://github.com/v2ray/v2ray-core
buffered input in kcp client
parent
2b4104e491
commit
1e0cad50b1
|
@ -19,16 +19,26 @@ var (
|
||||||
)
|
)
|
||||||
|
|
||||||
func fetchInput(ctx context.Context, input io.Reader, reader PacketReader, conn *Connection) {
|
func fetchInput(ctx context.Context, input io.Reader, reader PacketReader, conn *Connection) {
|
||||||
payload := buf.New()
|
cache := make(chan *buf.Buffer, 1024)
|
||||||
defer payload.Release()
|
go func() {
|
||||||
|
for {
|
||||||
for {
|
payload := buf.New()
|
||||||
err := payload.Reset(buf.ReadFrom(input))
|
if err := payload.Reset(buf.ReadFrom(input)); err != nil {
|
||||||
if err != nil {
|
payload.Release()
|
||||||
payload.Release()
|
close(cache)
|
||||||
return
|
return
|
||||||
|
}
|
||||||
|
select {
|
||||||
|
case cache <- payload:
|
||||||
|
default:
|
||||||
|
payload.Release()
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
for payload := range cache {
|
||||||
segments := reader.Read(payload.Bytes())
|
segments := reader.Read(payload.Bytes())
|
||||||
|
payload.Release()
|
||||||
if len(segments) > 0 {
|
if len(segments) > 0 {
|
||||||
conn.Input(segments)
|
conn.Input(segments)
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue