Sha256: b1c177ddbbed571fd0dbe2f93c41b60023f27d3a06b6849e23ebf1c4740d5484
Contents?: true
Size: 1.98 KB
Versions: 8
Compression:
Stored size: 1.98 KB
Contents
module MassiveRecord
module Adapters
module Thrift
class Scanner
attr_accessor :connection, :table_name, :column_family_names, :opened_scanner
attr_accessor :start_key, :offset_key, :created_at, :limit
attr_accessor :formatted_column_family_names, :column_family_names
def initialize(connection, table_name, column_family_names, opts = {})
@connection = connection
@table_name = table_name
@column_family_names = column_family_names.collect{|n| n.split(":").first}
@column_family_names = opts[:columns] unless opts[:columns].nil?
@formatted_column_family_names = @column_family_names.collect{|n| "#{n.split(":").first}:"}
@start_key = opts[:start_key].to_s
@offset_key = opts[:offset_key].to_s
@created_at = opts[:created_at].to_s
@limit = opts[:limit] || 10
end
def key
offset_key.empty? ? start_key : offset_key
end
def open
if created_at.empty?
self.opened_scanner = connection.scannerOpen(table_name, key, formatted_column_family_names)
else
self.opened_scanner = connection.scannerOpenTs(table_name, key, formatted_column_family_names, created_at)
end
end
def close
connection.scannerClose(opened_scanner)
end
def fetch_trows(opts = {})
connection.scannerGetList(opened_scanner, limit)
end
def fetch_rows(opts = {})
populate_rows(fetch_trows(opts))
end
def populate_rows(results)
results.collect do |result|
populate_row(result) unless result.row.match(/^#{start_key.gsub("|", "\\|")}/).nil?
end.select{|r| !r.nil?}
end
def populate_row(result)
Row.populate_from_trow_result(result, connection, table_name, column_family_names)
end
end
end
end
end
Version data entries
8 entries across 8 versions & 1 rubygems