diff options
author | Luke Shumaker <lukeshu@lukeshu.com> | 2018-04-18 14:28:19 -0400 |
---|---|---|
committer | Luke Shumaker <lukeshu@lukeshu.com> | 2018-04-18 14:28:19 -0400 |
commit | c78cb15705d6e4fb1c8a46a129391bb79e78d9a1 (patch) | |
tree | b37d364a1052a9b3e1da529ea9a0ccfd4d3c0a35 | |
parent | ce2c9ff07ea024b5c883139dd916cd31ca600a08 (diff) |
set a dial timeout
-rw-r--r-- | main.go | 8 |
1 files changed, 6 insertions, 2 deletions
@@ -97,14 +97,18 @@ func DoAddress(host, network, address string) Point { return NewPoint("public", tags, result) } +var dialer = net.Dialer{ + Timeout: 10 * time.Second, +} + func Hello(addr *net.TCPAddr) (name, version string, err error) { - conn, err := net.DialTCP(addr.Network(), nil, addr) + conn, err := dialer.Dial(addr.Network(), addr.String()) if err != nil { return "", "", err } defer conn.Close() conn.SetDeadline(time.Now().Add(10 * time.Second)) - conn.CloseWrite() + conn.(*net.TCPConn).CloseWrite() all, _ := ioutil.ReadAll(conn) line := strings.TrimRight(string(all), "\n") parts := strings.Split(line, " ") |