Use readpartial for SSLSocket to avoid blocking read

fluent · May 10, 2017 · 751f843 · 751f843
1 parent a8e7119
commit 751f843
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/fluent/plugin/out_forward.rb b/lib/fluent/plugin/out_forward.rb
@@ -420,7 +420,7 @@ def on_heartbeat(sockaddr, msg)
     # return chunk id to be committed
     def read_ack_from_sock(sock, unpacker)
       begin
-        raw_data = sock.instance_of?(Fluent::PluginHelper::Socket::WrappedSocket::TLS) ? sock.read(@read_length) : sock.recv(@read_length)
+        raw_data = sock.instance_of?(Fluent::PluginHelper::Socket::WrappedSocket::TLS) ? sock.readpartial(@read_length) : sock.recv(@read_length)
       rescue Errno::ECONNRESET
         raw_data = ""
       end