Skip to content

Commit 06b7f45

Browse files
authored
do not close writer (#12)
1 parent fef4030 commit 06b7f45

1 file changed

Lines changed: 2 additions & 13 deletions

File tree

pkg/conntester/conntester.go

Lines changed: 2 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -134,19 +134,8 @@ func (c *ConnTester) send(ctx context.Context, wg *sync.WaitGroup) {
134134
}
135135
c.logger.Info(fmt.Sprintf("successful connection and data sent to %s", dstEndpoint))
136136

137-
if tcpConn, ok := conn.(*net.TCPConn); ok {
138-
if err := tcpConn.CloseWrite(); err != nil {
139-
c.logger.Info(fmt.Sprintf("error during CloseWrite to %s: %s", dstEndpoint, err))
140-
} else {
141-
c.logger.Info("TCP FIN packet sent, writing side closed")
142-
}
143-
144-
_, err = io.Copy(io.Discard, conn)
145-
if err != nil && err != io.EOF {
146-
c.logger.Info(fmt.Sprintf("stopped discarding data from %s: %s", dstEndpoint, err))
147-
} else {
148-
c.logger.Info("server response fully read and discarded")
149-
}
137+
if _, ok := conn.(*net.TCPConn); ok {
138+
_, _ = io.Copy(io.Discard, conn) // keep reading until either the server closes the connection or the timeout is reached
150139
}
151140

152141
if err := conn.Close(); err != nil {

0 commit comments

Comments
 (0)