Respect the response from block

If keepalive, #connect always returns integer returned by L1104 Signed-off-by: Yuta Iwama <[email protected]>
fluent · Jun 13, 2019 · 3cacaf5 · 3cacaf5
1 parent 12ffa0a
commit 3cacaf5
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lib/fluent/plugin/out_forward.rb b/lib/fluent/plugin/out_forward.rb
@@ -1081,8 +1081,9 @@ def connect(host = nil)
                end
 
         if block_given?
+          ret = nil
           begin
-            yield(sock)
+            ret = yield(sock)
           rescue
             @socket_cache.revoke if @keepalive
             raise
@@ -1091,6 +1092,8 @@ def connect(host = nil)
           ensure
             sock.close unless @keepalive
           end
+
+          ret
         else
           sock
         end