dont close until they close themself
This commit is contained in:
19
tcp.go
19
tcp.go
@@ -1,7 +1,6 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
|
||||||
"crypto/tls"
|
"crypto/tls"
|
||||||
"io"
|
"io"
|
||||||
"log"
|
"log"
|
||||||
@@ -44,27 +43,19 @@ func (tcp TCP) Listen() error {
|
|||||||
}
|
}
|
||||||
defer conn2.Close()
|
defer conn2.Close()
|
||||||
|
|
||||||
ctx, can := context.WithCancel(context.Background())
|
|
||||||
defer can()
|
|
||||||
|
|
||||||
errc := make(chan error)
|
errc := make(chan error)
|
||||||
go func() {
|
go func() {
|
||||||
_, err := io.Copy(conn, conn2)
|
_, err := io.Copy(conn, conn2)
|
||||||
select {
|
errc <- err
|
||||||
case errc <- err:
|
|
||||||
case <-ctx.Done():
|
|
||||||
close(errc)
|
|
||||||
}
|
|
||||||
}()
|
}()
|
||||||
go func() {
|
go func() {
|
||||||
_, err := io.Copy(conn2, conn)
|
_, err := io.Copy(conn2, conn)
|
||||||
select {
|
errc <- err
|
||||||
case errc <- err:
|
|
||||||
case <-ctx.Done():
|
|
||||||
close(errc)
|
|
||||||
}
|
|
||||||
}()
|
}()
|
||||||
|
for i := 0; i < 2; i++ {
|
||||||
<-errc
|
<-errc
|
||||||
|
}
|
||||||
|
close(errc)
|
||||||
}()
|
}()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user