Navigation Menu

Skip to content

Instantly share code, notes, and snippets.

@bporterfield
Created March 1, 2012 23:20
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save bporterfield/1953963 to your computer and use it in GitHub Desktop.
Save bporterfield/1953963 to your computer and use it in GitHub Desktop.
buffering.rb do_write that handles different encodings
##
# Writes +s+ to the buffer. When the buffer is full or #sync is true the
# buffer is flushed to the underlying socket.
def do_write(s)
@wbuffer = "" unless defined? @wbuffer
@wbuffer << s
@sync ||= false
if @sync or @wbuffer.bytesize > BLOCK_SIZE or idx = @wbuffer.rindex($/)
remain = idx ? idx + $/.size : @wbuffer.bytesize
nwritten = 0
while remain > 0
str = @wbuffer.byteslice(nwritten,remain)
begin
nwrote = syswrite(str)
rescue Errno::EAGAIN
retry
end
remain -= nwrote
nwritten += nwrote
end
@wbuffer = @wbuffer.byteslice(nwritten, @wbuffer.bytesize - nwritten)
end
end
public
##
# Writes +s+ to the stream. If the argument is not a string it will be
# converted using String#to_s. Returns the number of bytes written.
def write(s)
do_write(s)
s.bytesize
end
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment