Merge pull request #2547 from komeda-shinji/fix-compressable-data-lost

fix data lost on decompression
fluent · Aug 10, 2019 · 9e88103 · 9e88103
2 parents 33beb99 + 2976dde
commit 9e88103
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/lib/fluent/plugin/compressable.rb b/lib/fluent/plugin/compressable.rb
@@ -64,9 +64,11 @@ def string_decompress(compressed_data)
           unused = gz.unused
           gz.finish
 
-          break if unused.nil?
-          adjust = unused.length
-          io.pos -= adjust
+          unless unused.nil?
+            adjust = unused.length
+            io.pos -= adjust
+          end
+          break if io.eof?
         end
 
         out
@@ -80,9 +82,11 @@ def io_decompress(input, output)
           unused = gz.unused
           gz.finish
 
-          break if unused.nil?
-          adjust = unused.length
-          input.pos -= adjust
+          unless unused.nil?
+            adjust = unused.length
+            input.pos -= adjust
+          end
+          break if input.eof?
         end
 
         output