ScuroNeko/mtg
1
0
Fork 0
mirror of https://github.com/ScuroNeko/mtg.git synced 2026-08-31 17:14:02 +03:00
Code Issues Packages Projects Releases Wiki Activity

Guarantee of closing sockets after context is done

Browse Source
This commit is contained in:
9seconds
2018-07-28 15:36:40 +03:00
parent 243a89a68c
commit 1ee34fd502
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
proxy/proxy.go
+7
View File
@@ -47,6 +47,7 @@ func (p *Proxy) accept(conn net.Conn) {
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
defer func() { defer func() {
cancel()
conn.Close() // nolint: errcheck conn.Close() // nolint: errcheck
if err := recover(); err != nil { if err := recover(); err != nil {
@@ -74,6 +75,12 @@ func (p *Proxy) accept(conn net.Conn) {
} }
defer serverConn.(io.Closer).Close() // nolint: errcheck defer serverConn.(io.Closer).Close() // nolint: errcheck
go func() {
<-ctx.Done()
serverConn.(io.Closer).Close()
clientConn.(io.Closer).Close()
}()
wait := &sync.WaitGroup{} wait := &sync.WaitGroup{}
wait.Add(2) wait.Add(2)