hbase/bin/Formatter.rb

113 lines
2.9 KiB
Ruby

# Results formatter
module Formatter
class Formatter
# Base abstract class for results formatting.
def initialize(o, w = 80)
raise TypeError.new("Type %s of parameter %s is not IO" % [o.class, o]) \
unless o.instance_of? IO
@out = o
@maxWidth = w
@rowCount = 0
end
def header(args = [])
row(args) if args.length > 0
@rowCount = 0
end
def row(args = [])
if not args or args.length == 0
# Print out nothing
return
end
# TODO: Look at the type. Is it RowResult?
if args.length == 1
splits = split(@maxWidth, dump(args[0]))
for l in splits
output(@maxWidth, l)
puts
end
elsif args.length == 2
col1width = 8
col2width = 70
splits1 = split(col1width, dump(args[0]))
splits2 = split(col2width, dump(args[1]))
biggest = (splits2.length > splits1.length)? splits2.length: splits1.length
index = 0
while index < biggest
@out.print(" ")
output(col1width, splits1[index])
@out.print(" ")
output(col2width, splits2[index])
index += 1
puts
end
else
# Print a space to set off multi-column rows
print ' '
first = true
for e in args
@out.print " " unless first
first = false
@out.print e
end
puts
end
@rowCount += 1
end
def split(width, str)
result = []
index = 0
while index < str.length do
result << str.slice(index, index + width)
index += width
end
result
end
def dump(str)
# Remove double-quotes added by 'dump'.
return str.dump.slice(1, str.length)
end
def output(width, str)
# Make up a spec for printf
spec = "%%-%d.%ds" % [width, width]
@out.printf(spec % str)
end
def footer(startTime = nil)
if not startTime
return
end
# Only output elapsed time and row count if startTime passed
@out.puts("%d row(s) in %s seconds" % [@rowCount, Time.now - startTime])
end
end
class Console < Formatter
end
class XHTMLFormatter < Formatter
# http://www.germane-software.com/software/rexml/doc/classes/REXML/Document.html
# http://www.crummy.com/writing/RubyCookbook/test_results/75942.html
end
class JSON < Formatter
end
# Do a bit of testing.
if $0 == __FILE__
formatter = Console.new(STDOUT)
formatter.header(['a', 'b'])
formatter.row(['a', 'b'])
formatter.row(['xxxxxxxxx xxxxxxxxxxx xxxxxxxxxxx xxxxxxxxxxxx xxxxxxxxx xxxxxxxxxxxx xxxxxxxxxxxxxxx xxxxxxxxx xxxxxxxxxxxxxx'])
formatter.row(['yyyyyy yyyyyy yyyyy yyy', 'xxxxxxxxx xxxxxxxxxxx xxxxxxxxxxx xxxxxxxxxxxx xxxxxxxxx xxxxxxxxxxxx xxxxxxxxxxxxxxx xxxxxxxxx xxxxxxxxxxxxxx xxx xx x xx xxx xx xx xx x xx x x xxx x x xxx x x xx x x x x x x xx '])
formatter.footer()
end
end