1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-06-10 09:50:43 +00:00

close pipe when finish

This commit is contained in:
Darien Raymond 2018-07-02 15:40:32 +02:00
parent 6315217572
commit bc78e63877
No known key found for this signature in database
GPG Key ID: 7251FFA14BB18169

View File

@ -58,6 +58,8 @@ func (server *Server) handleConnection(conn net.Conn) {
pReader, pWriter := pipe.New(pipe.WithoutSizeLimit())
err := task.Run(task.Parallel(func() error {
defer pWriter.Close() // nolint: errcheck
for {
b := buf.New()
if err := b.AppendSupplier(buf.ReadFrom(conn)); err != nil {
@ -72,6 +74,8 @@ func (server *Server) handleConnection(conn net.Conn) {
}
}
}, func() error {
defer pReader.CloseError()
w := buf.NewWriter(conn)
for {
mb, err := pReader.ReadMultiBuffer()