Fix possible file descriptor leak

felixge · Aug 31, 2015 · 387b9d2 · 387b9d2
1 parent 59d8c3f
commit 387b9d2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/keepalive.go b/keepalive.go
@@ -78,9 +78,11 @@ func SetKeepAlive(c net.Conn, idleTime time.Duration, count int, interval time.D
 	}
 
 	var f *os.File
+
 	if f, err = conn.File(); err != nil {
 		return err
 	}
+	defer f.Close()
 
 	fd := int(f.Fd())
 
@@ -100,8 +102,6 @@ func SetKeepAlive(c net.Conn, idleTime time.Duration, count int, interval time.D
 		return err
 	}
 
-	f.Close()
-
 	return nil
 }