set timeout before write

phuslu · Oct 23, 2024 · 905a819 · 905a819
1 parent 01e0982
commit 905a819
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/client.go b/client.go
@@ -48,18 +48,18 @@ func (c *Client) exchange(ctx context.Context, req, resp *Message) error {
 		return err
 	}
 
-	_, err = conn.Write(req.Raw)
-	if err != nil {
-		return nil
-	}
-
 	if c.Timeout > 0 {
 		err = conn.SetDeadline(time.Now().Add(c.Timeout))
 		if err != nil {
 			return err
 		}
 	}
 
+	_, err = conn.Write(req.Raw)
+	if err != nil {
+		return nil
+	}
+
 	resp.Raw = resp.Raw[:cap(resp.Raw)]
 	n, err := conn.Read(resp.Raw)
 	if err != nil {