From 0b7eb1e9ddfce64725b13516bca9804b25dcbc81 Mon Sep 17 00:00:00 2001 From: James Barnett Date: Sun, 23 Dec 2018 21:11:21 +0000 Subject: Resolve target address before first client connection Add description to cmd flags --- slowpoke.go | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) (limited to 'slowpoke.go') diff --git a/slowpoke.go b/slowpoke.go index 98182f6..4a36306 100644 --- a/slowpoke.go +++ b/slowpoke.go @@ -10,7 +10,7 @@ import ( type Slowpoke struct { conn net.Conn - targetAddr string + targetAddr *net.TCPAddr latency time.Duration bufferSize int isClosed bool @@ -18,7 +18,7 @@ type Slowpoke struct { logger *logging.Logger } -func New(conn net.Conn, targetAddr string, latency time.Duration, bufferSize int, logger *logging.Logger) *Slowpoke { +func New(conn net.Conn, targetAddr *net.TCPAddr, latency time.Duration, bufferSize int, logger *logging.Logger) *Slowpoke { return &Slowpoke{ conn: conn, targetAddr: targetAddr, @@ -32,9 +32,8 @@ func New(conn net.Conn, targetAddr string, latency time.Duration, bufferSize int func (s *Slowpoke) StartTransfer() { defer s.conn.Close() - target, err := net.Dial("tcp", s.targetAddr) + target, err := net.DialTCP("tcp", nil, s.targetAddr) if err != nil { - // TODO validate target addr before this point s.logger.Errorf("Failed to connect to target address %s:\n%v", s.targetAddr, err) return } -- cgit v1.2.3