We may not need this temporary file if we've flushed everything out and entered bypass mode. --- lib/yahns/wbuf.rb | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-)
diff --git a/lib/yahns/wbuf.rb b/lib/yahns/wbuf.rb index 991557e..4eed2c5 100644 --- a/lib/yahns/wbuf.rb +++ b/lib/yahns/wbuf.rb @@ -1,5 +1,5 @@ # -*- encoding: binary -*- -# Copyright (C) 2013, Eric Wong <normalper...@yhbt.net> et. al. +# Copyright (C) 2013-2015 all contributors <yahns-public@yhbt.net> # License: GPLv3 or later (https://www.gnu.org/licenses/gpl-3.0.txt) require_relative 'wbuf_common' @@ -31,7 +31,7 @@ class Yahns::Wbuf # :nodoc: include Yahns::WbufCommon def initialize(body, persist, tmpdir) - @tmpio = Yahns::TmpIO.new(tmpdir) + @tmpio = nil @tmpdir = tmpdir @sf_offset = @sf_count = 0 @wbuf_persist = persist # whether or not we keep the connection alive @@ -50,6 +50,7 @@ def wbuf_write(client, buf) @bypass = false # ugh, continue to buffering to file end while @bypass + @tmpio ||= Yahns::TmpIO.new(@tmpdir) @sf_count += @tmpio.write(buf) case rv = client.trysendfile(@tmpio, @sf_offset, @sf_count) when Integer @@ -64,16 +65,15 @@ def wbuf_write(client, buf) # we're all caught up, try to prevent dirty data from getting flushed # to disk if we can help it. - @tmpio.close + @tmpio = @tmpio.close @sf_offset = 0 - @tmpio = Yahns::TmpIO.new(@tmpdir) @bypass = true nil end # called by last wbuf_flush def wbuf_close(client) - @tmpio = @tmpio.close + @tmpio = @tmpio.close if @tmpio wbuf_close_common(client) end end -- EW