From 61149fd36bb1959bc2ae4ecb71c68fbda3c35c7a Mon Sep 17 00:00:00 2001 From: Eric Wong Date: Sat, 24 Jan 2015 00:23:01 +0000 Subject: wbuf: lazily (re)create temporary file 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(-) (limited to 'lib') 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 et. al. +# Copyright (C) 2013-2015 all contributors # 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 @@ class Yahns::Wbuf # :nodoc: @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 @@ class Yahns::Wbuf # :nodoc: # 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 -- cgit v1.2.3-24-ge0c7