close connection to snmp properly

main.go

@@ -78,6 +78,15 @@ func main() {
 	stat := ok
 
 	defer func() {
+		if gosnmp.Default.Conn != nil {
+			err := gosnmp.Default.Conn.Close()
+
+			if err != nil {
+				fmt.Println(err)
+				os.Exit(unknown)
+			}
+		}
+
 		os.Exit(stat)
 	}()
 
@@ -136,8 +145,6 @@ func main() {
 			return
 		}
 
-		defer gosnmp.Default.Conn.Close()
-
 		status, err := dfFlags.diskSpaceUsage()
 
 		if err != nil {
@@ -145,8 +152,8 @@ func main() {
 			stat = int(status)
 			return
 		}
-
-		os.Exit(int(status))
+		stat = int(status)
+		return
 	case "version":
 		version.Parse(os.Args[2:])