Set basic timeout defaults in cmd/hdfs

Author: colinmarc

cmd/hdfs/main.go

@@ -3,7 +3,9 @@ package main
 import (
 	"errors"
 	"fmt"
+	"net"
 	"os"
+	"time"
 
 	"github.com/colinmarc/hdfs"
 	"github.com/pborman/getopt"
@@ -199,6 +201,16 @@ func getClient(namenode string) (*hdfs.Client, error) {
 		}
 	}
 
+	// Set some basic defaults.
+	dialFunc := (&net.Dialer{
+		Timeout:   5 * time.Second,
+		KeepAlive: 5 * time.Second,
+		DualStack: true,
+	}).DialContext
+
+	options.NamenodeDialFunc = dialFunc
+	options.DatanodeDialFunc = dialFunc
+
 	c, err := hdfs.NewClient(options)
 	if err != nil {
 		return nil, fmt.Errorf("Couldn't connect to namenode: %s", err)