lxc-exporter/lib/cborio.rb

90 lines
1.7 KiB
Ruby
Raw Normal View History

# vim: set noet sw=2 ts=2 sts=2:
require 'cbor'
require 'socket'
class CBORIO <IO
include Enumerable
extend Enumerable
def orig_io() @io end
def to_io() self end
def to_i() @io.to_i end
def fileno() @io.fileno end
def unpacker() @un end
def close() @io.close end
def close_write() @io.close_write end
def close_read() @io.close_read end
def write( obj) @io.write CBOR.pack( obj) end
def sync() @io.sync end
def sync=( v) @io.sync = v end
def inspect() "#<#{self.class.name} fd:#{@io.to_i} #{@mode}>" end
class <<self
def stream() Socket.pair( :UNIX, :STREAM, 0).map {|s| self.new s, :rw } end
def pipe
a, b = IO.pipe
[self.new( a, :r), self.new( b, :w)]
end
def popen mode = nil, &exe
s1, s2 =
case mode
when :rw, 'rw', nil then self.stream
when :r, 'r' then self.pipe
when :w, 'w' then self.pipe.reverse
end
pid =
fork do
s1.close
yield s2
exit 0
end
s2.close
[pid, s1]
end
def open io, mode = :r, &exe
s = self.new io, mode
if block_given?
begin yield s
ensure s.close
end
else s
end
end
def each fn, &exe
open fn, 'r' do |f|
f.each &exe
end
end
end
def initialize io, mode = :rw
case io
when IO then @io = io
when String then @io = File.open io, mode
when Pathname then @io = io.open mode
else raise ArgumentError, "This seems not to be a IO/String/Pathname: #{io.inspect}"
end
@mode, @un, @io.sync = mode, CBOR::Unpacker.new, true
end
def read
return @un.read rescue EOFError
loop do
@un.feed @io.readpartial( 1024)
return @un.read rescue EOFError
end
end
def each &exe
return to_enum __method__ unless block_given?
loop { yield read }
self
rescue EOFError
self
end
end