bughz/glider
1
0
Fork 0
mirror of https://github.com/nadoo/glider.git synced 2025-04-21 19:52:07 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Optimize tcp shutdown

Browse Source
This commit is contained in:
zhengyu 2019-08-19 19:04:05 +08:00
parent 8fd20daa95
commit e6f1c8928e
1 changed files with 52 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

83
common/conn/conn.go
View File

@ -35,6 +35,56 @@ func (c *Conn) Reader() *bufio.Reader {
return c.r
}
func Shutdown(dst, src net.Conn) {
switch s := src.(type) {
case *net.TCPConn:
s.CloseRead()
s.SetDeadline(time.Now().Add(time.Second * 60))
case *net.UnixConn:
s.CloseRead()
s.SetDeadline(time.Now().Add(time.Second * 60))
case *Conn:
{
switch s1 := dst.(type) {
case *net.TCPConn:
s1.CloseRead()
s1.SetDeadline(time.Now().Add(time.Second * 60))
case *net.UnixConn:
s1.CloseRead()
s1.SetDeadline(time.Now().Add(time.Second * 60))
default:
s1.SetDeadline(time.Now())
}
}
default:
s.SetDeadline(time.Now())
}
switch d := dst.(type) {
case *net.TCPConn:
d.CloseWrite()
d.SetDeadline(time.Now().Add(time.Second * 60))
case *net.UnixConn:
d.CloseWrite()
d.SetDeadline(time.Now().Add(time.Second * 60))
case *Conn:
{
switch d1 := dst.(type) {
case *net.TCPConn:
d1.CloseWrite()
d1.SetDeadline(time.Now().Add(time.Second * 60))
case *net.UnixConn:
d1.CloseWrite()
d1.SetDeadline(time.Now().Add(time.Second * 60))
default:
d1.SetDeadline(time.Now())
}
}
default:
d.SetDeadline(time.Now())
}
}
// Relay relays between left and right.
func Relay(left, right net.Conn) (int64, int64, error) {
type res struct {
@ -45,41 +95,12 @@ func Relay(left, right net.Conn) (int64, int64, error) {
go func() {
n, err := io.Copy(right, left)
switch src := left.(type) {
case *net.TCPConn:
src.CloseRead()
default:
src.SetDeadline(time.Now())
}
switch dst := right.(type) {
case *net.TCPConn:
dst.CloseWrite()
dst.SetDeadline(time.Now().Add(time.Second * 60))
default:
dst.SetDeadline(time.Now())
}
Shutdown(right, left)
ch <- res{n, err}
}()
n, err := io.Copy(left, right)
switch src := left.(type) {
case *net.TCPConn:
src.CloseWrite()
src.SetDeadline(time.Now().Add(time.Second * 60))
default:
src.SetDeadline(time.Now())
}
switch dst := right.(type) {
case *net.TCPConn:
dst.CloseRead()
default:
dst.SetDeadline(time.Now())
}
Shutdown(left, right)
rs := <-ch
if err == nil {