Module: Thrift::Struct_Union
- Defined in:
- lib/thrift/struct_union.rb
Constant Summary collapse
- CONTAINER_TYPES =
[]
Instance Method Summary collapse
- #each_field ⇒ Object
- #field_info(field) ⇒ Object
- #inspect_collection(collection, field_info) ⇒ Object
- #inspect_field(value, field_info) ⇒ Object
- #is_container?(type) ⇒ Boolean
- #name_to_id(name) ⇒ Object
- #read_field(iprot, field = {}) ⇒ Object
- #write_container(oprot, value, field = {}) ⇒ Object
- #write_data(oprot, value, field) ⇒ Object
Instance Method Details
#each_field ⇒ Object
35 36 37 38 39 40 |
# File 'lib/thrift/struct_union.rb', line 35 def each_field struct_fields.keys.sort.each do |fid| data = struct_fields[fid] yield fid, data end end |
#field_info(field) ⇒ Object
118 119 120 121 122 123 124 |
# File 'lib/thrift/struct_union.rb', line 118 def field_info(field) { :type => field[:type], :class => field[:class], :key => field[:key], :value => field[:value], :element => field[:element] } end |
#inspect_collection(collection, field_info) ⇒ Object
151 152 153 154 155 156 157 |
# File 'lib/thrift/struct_union.rb', line 151 def inspect_collection(collection, field_info) buf = [] collection.each do |k| buf << inspect_field(k, field_info[:element]) end "[" + buf.join(", ") + "]" end |
#inspect_field(value, field_info) ⇒ Object
126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 |
# File 'lib/thrift/struct_union.rb', line 126 def inspect_field(value, field_info) if enum_class = field_info[:enum_class] "#{enum_class.const_get(:VALUE_MAP)[value]} (#{value})" elsif value.is_a? Hash if field_info[:type] == Types::MAP map_buf = [] value.each do |k, v| map_buf << inspect_field(k, field_info[:key]) + ": " + inspect_field(v, field_info[:value]) end "{" + map_buf.join(", ") + "}" else # old-style set inspect_collection(value.keys, field_info) end elsif value.is_a? Array inspect_collection(value, field_info) elsif value.is_a? Set inspect_collection(value, field_info) elsif value.is_a?(String) && field_info[:binary] value.unpack("H*").first else value.inspect end end |
#is_container?(type) ⇒ Boolean
114 115 116 |
# File 'lib/thrift/struct_union.rb', line 114 def is_container?(type) CONTAINER_TYPES[type] end |
#name_to_id(name) ⇒ Object
23 24 25 26 27 28 29 30 31 32 33 |
# File 'lib/thrift/struct_union.rb', line 23 def name_to_id(name) names_to_ids = self.class.instance_variable_get("@names_to_ids") unless names_to_ids names_to_ids = {} struct_fields.each do |fid, field_def| names_to_ids[field_def[:name]] = fid end self.class.instance_variable_set("@names_to_ids", names_to_ids) end names_to_ids[name] end |
#read_field(iprot, field = {}) ⇒ Object
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 71 72 73 74 |
# File 'lib/thrift/struct_union.rb', line 42 def read_field(iprot, field = {}) case field[:type] when Types::STRUCT value = field[:class].new value.read(iprot) when Types::MAP key_type, val_type, size = iprot.read_map_begin value = {} size.times do k = read_field(iprot, field_info(field[:key])) v = read_field(iprot, field_info(field[:value])) value[k] = v end iprot.read_map_end when Types::LIST e_type, size = iprot.read_list_begin value = Array.new(size) do |n| read_field(iprot, field_info(field[:element])) end iprot.read_list_end when Types::SET e_type, size = iprot.read_set_begin value = Set.new size.times do element = read_field(iprot, field_info(field[:element])) value << element end iprot.read_set_end else value = iprot.read_type(field[:type]) end value end |
#write_container(oprot, value, field = {}) ⇒ Object
84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 |
# File 'lib/thrift/struct_union.rb', line 84 def write_container(oprot, value, field = {}) case field[:type] when Types::MAP oprot.write_map_begin(field[:key][:type], field[:value][:type], value.size) value.each do |k, v| write_data(oprot, k, field[:key]) write_data(oprot, v, field[:value]) end oprot.write_map_end when Types::LIST oprot.write_list_begin(field[:element][:type], value.size) value.each do |elem| write_data(oprot, elem, field[:element]) end oprot.write_list_end when Types::SET oprot.write_set_begin(field[:element][:type], value.size) value.each do |v,| # the , is to preserve compatibility with the old Hash-style sets write_data(oprot, v, field[:element]) end oprot.write_set_end else raise "Not a container type: #{field[:type]}" end end |
#write_data(oprot, value, field) ⇒ Object
76 77 78 79 80 81 82 |
# File 'lib/thrift/struct_union.rb', line 76 def write_data(oprot, value, field) if is_container? field[:type] write_container(oprot, value, field) else oprot.write_type(field[:type], value) end end |