dont close until they close themself
parent
7d82bdcdfe
commit
5fccea247e
21
tcp.go
21
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)
|
|
||||||
}
|
|
||||||
}()
|
}()
|
||||||
<-errc
|
for i := 0; i < 2; i++ {
|
||||||
|
<-errc
|
||||||
|
}
|
||||||
|
close(errc)
|
||||||
}()
|
}()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue