/
columns.rb
59 lines (51 loc) · 2.05 KB
/
columns.rb
1
2
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
class Cassandra
# A bunch of crap, mostly related to introspecting on column types
module Columns #:nodoc:
def is_super(column_family)
@is_super[column_family] ||= column_family_property(column_family, 'column_type') == "Super"
end
def column_name_class(column_family)
@column_name_class[column_family] ||= column_name_class_for_key(column_family, "comparator_type")
end
def sub_column_name_class(column_family)
@sub_column_name_class[column_family] ||= column_name_class_for_key(column_family, "subcomparator_type")
end
def column_family_property(column_family, key)
cfdef = schema.cf_defs.find {|cfdef| cfdef.name == column_family }
unless cfdef
raise AccessError, "Invalid column family \"#{column_family}\""
end
cfdef.send(key)
end
private
def _standard_insert_mutation(column_family, column_name, value, timestamp, ttl = nil)
CassandraThrift::Mutation.new(
:column_or_supercolumn => CassandraThrift::ColumnOrSuperColumn.new(
:column => CassandraThrift::Column.new(
:name => column_name_class(column_family).new(column_name).to_s,
:value => value,
:timestamp => timestamp,
:ttl => ttl
)
)
)
end
def _super_insert_mutation(column_family, super_column_name, sub_columns, timestamp, ttl = nil)
CassandraThrift::Mutation.new(:column_or_supercolumn =>
CassandraThrift::ColumnOrSuperColumn.new(
:super_column => CassandraThrift::SuperColumn.new(
:name => column_name_class(column_family).new(super_column_name).to_s,
:columns => sub_columns.collect { |sub_column_name, sub_column_value|
CassandraThrift::Column.new(
:name => sub_column_name_class(column_family).new(sub_column_name).to_s,
:value => sub_column_value.to_s,
:timestamp => timestamp,
:ttl => ttl
)
}
)
)
)
end
end
end