calculate bytes when write to socket

reidmorrison#10
didww · Sep 26, 2017 · 2127d14 · 2127d14
1 parent 3679470
commit 2127d14
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/lib/net/tcp_client/tcp_client.rb b/lib/net/tcp_client/tcp_client.rb
@@ -612,8 +612,17 @@ def socket_write(data, timeout)
         socket.write(data)
       else
         deadline = Time.now.utc + timeout
+        length = data.bytesize
+        total_count = 0
         non_blocking(socket, deadline) do
-          socket.write_nonblock(data)
+          begin
+            count = socket.write_nonblock(data)
+          rescue Errno::EWOULDBLOCK
+            retry
+          end
+          total_count += count
+          return total_count if total_count >= length
+          data = data.byteslice(count..-1)
         end
       end
     rescue NonBlockingTimeout