3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
# File 'lib/priam/command/get.rb', line 3
def self.run(argv, input_stream=$stdin, output_stream=$stdout)
params = Priam::Core::Common.parse_opts(argv)
host = params[:host]
port = params[:port]
keyspace = params[:keyspace]
column_family = params[:column_family]
super_column = params[:super_column]
name = params[:value_column]
with_key_flag = params[:with_key_flag]
weight_second = params[:weight_second]
retry_max_count = params[:retry_max_count]
client = Cassandra.new(keyspace, "#{host}:#{port}")
Priam.logger.debug "Cluster Name: #{client.cluster_name}"
Priam.logger.debug "Key Space : #{keyspace}"
key_list = []
count = 0
while line = input_stream.gets do
key = line.chomp
key_list.push key
if key.nil? || key == '' then
output_stream.puts
next
end
retry_count = 0
begin
record = Priam.get_column(client, column_family, super_column, key, params)
if with_key_flag
output_stream.print "#{key}\t"
end
if name then
output_stream.puts "#{record[name]}"
else
output_stream.puts record.to_json
end
count += 1
if count % 10 == 0 then
Priam.logger.info " GET [#{key_list.join(',')}]"
key_list.clear
end
retry_count = 0
rescue Exception=>e
if retry_count < retry_max_count
Priam.logger.warn("(EXCEPTION)#{e.message}(#{e.class.name}): #{e.backtrace.map{|s| " #{s}"}.join("\n")}")
Priam.logger.warn("retry(#{retry_count})")
retry_count += 1
sleep(weight_second)
retry
else
raise e
end
end
end
if key_list != [] then
Priam.logger.info " GET [#{key_list.join(',')}]"
end
uri = "cassandra://#{host}:#{port}/#{keyspace}/#{column_family}"
uri += "/#{super_column}" if super_column
Priam.logger.info " got #{count} columns from #{uri}"
end
|