idk/sam-forwarder
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Figure out where I need to defer those closes.

Browse Source
This commit is contained in:
idk
2018-09-10 20:06:09 -04:00
parent a874659167
commit b53e0dee95
2 changed files with 12 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
forwarder.go
View File

@ -144,11 +144,10 @@ func (f *SAMForwarder) forward(conn *sam3.SAMConn) { //(conn net.Conn) {
var client net.Conn
if client, err = net.Dial("tcp", f.Target()); err != nil {
log.Fatalf("Dial failed: %v", err)
} else {
defer client.Close()
defer conn.Close()
}
go func() {
defer client.Close()
defer conn.Close()
if f.Type == "http" {
if requestbytes, request, err = f.HTTPRequestBytes(conn); err == nil {
log.Printf("Forwarding modified request: \n\t %s", string(requestbytes))
@ -161,6 +160,8 @@ func (f *SAMForwarder) forward(conn *sam3.SAMConn) { //(conn net.Conn) {
}
}()
go func() {
defer client.Close()
defer conn.Close()
if f.Type == "http" {
if responsebytes, err = f.HTTPResponseBytes(client, request); err == nil {
log.Printf("Forwarding modified response: \n\t%s", string(requestbytes))