Class | Thrift::CompactProtocol |
In: |
lib/thrift/protocol/compact_protocol.rb
lib/thrift/protocol/compact_protocol.rb |
Parent: | BaseProtocol |
PROTOCOL_ID | = | [0x82].pack('c').unpack('c').first |
VERSION | = | 1 |
VERSION_MASK | = | 0x1f |
TYPE_MASK | = | 0xE0 |
TYPE_SHIFT_AMOUNT | = | 5 |
TSTOP | = | ["", Types::STOP, 0] |
SEVEN_BIT_MASK | = | 0x7F |
EVERYTHING_ELSE_MASK | = | ~SEVEN_BIT_MASK |
PROTOCOL_ID | = | [0x82].pack('c').unpack('c').first |
VERSION | = | 1 |
VERSION_MASK | = | 0x1f |
TYPE_MASK | = | 0xE0 |
TYPE_SHIFT_AMOUNT | = | 5 |
TSTOP | = | ["", Types::STOP, 0] |
SEVEN_BIT_MASK | = | 0x7F |
EVERYTHING_ELSE_MASK | = | ~SEVEN_BIT_MASK |
# File lib/thrift/protocol/compact_protocol.rb, line 96 96: def initialize(transport) 97: super(transport) 98: 99: @last_field = [0] 100: @boolean_value = nil 101: 102: # Pre-allocated read buffer for read_double(). 103: @rbuf = Bytes.empty_byte_buffer(8) 104: end
# File lib/thrift/protocol/compact_protocol.rb, line 96 96: def initialize(transport) 97: super(transport) 98: 99: @last_field = [0] 100: @boolean_value = nil 101: 102: # Pre-allocated read buffer for read_double(). 103: @rbuf = Bytes.empty_byte_buffer(8) 104: end
# File lib/thrift/protocol/compact_protocol.rb, line 298 298: def read_bool 299: unless @bool_value.nil? 300: bv = @bool_value 301: @bool_value = nil 302: bv 303: else 304: read_byte() == CompactTypes::BOOLEAN_TRUE 305: end 306: end
# File lib/thrift/protocol/compact_protocol.rb, line 298 298: def read_bool 299: unless @bool_value.nil? 300: bv = @bool_value 301: @bool_value = nil 302: bv 303: else 304: read_byte() == CompactTypes::BOOLEAN_TRUE 305: end 306: end
# File lib/thrift/protocol/compact_protocol.rb, line 308 308: def read_byte 309: val = trans.read_byte 310: if (val > 0x7f) 311: val = 0 - ((val - 1) ^ 0xff) 312: end 313: val 314: end
# File lib/thrift/protocol/compact_protocol.rb, line 308 308: def read_byte 309: val = trans.read_byte 310: if (val > 0x7f) 311: val = 0 - ((val - 1) ^ 0xff) 312: end 313: val 314: end
# File lib/thrift/protocol/compact_protocol.rb, line 328 328: def read_double 329: trans.read_into_buffer(@rbuf, 8) 330: val = @rbuf.reverse.unpack('G').first 331: val 332: end
# File lib/thrift/protocol/compact_protocol.rb, line 328 328: def read_double 329: trans.read_into_buffer(@rbuf, 8) 330: val = @rbuf.reverse.unpack('G').first 331: val 332: end
# File lib/thrift/protocol/compact_protocol.rb, line 246 246: def read_field_begin 247: type = read_byte() 248: 249: # if it's a stop, then we can return immediately, as the struct is over. 250: if (type & 0x0f) == Types::STOP 251: TSTOP 252: else 253: field_id = nil 254: 255: # mask off the 4 MSB of the type header. it could contain a field id delta. 256: modifier = (type & 0xf0) >> 4 257: if modifier == 0 258: # not a delta. look ahead for the zigzag varint field id. 259: @last_field.pop 260: field_id = read_i16() 261: else 262: # has a delta. add the delta to the last read field id. 263: field_id = @last_field.pop + modifier 264: end 265: 266: # if this happens to be a boolean field, the value is encoded in the type 267: if CompactTypes.is_bool_type?(type) 268: # save the boolean value in a special instance variable. 269: @bool_value = (type & 0x0f) == CompactTypes::BOOLEAN_TRUE 270: end 271: 272: # push the new field onto the field stack so we can keep the deltas going. 273: @last_field.push(field_id) 274: ["", CompactTypes.get_ttype(type & 0x0f), field_id] 275: end 276: end
# File lib/thrift/protocol/compact_protocol.rb, line 246 246: def read_field_begin 247: type = read_byte() 248: 249: # if it's a stop, then we can return immediately, as the struct is over. 250: if (type & 0x0f) == Types::STOP 251: TSTOP 252: else 253: field_id = nil 254: 255: # mask off the 4 MSB of the type header. it could contain a field id delta. 256: modifier = (type & 0xf0) >> 4 257: if modifier == 0 258: # not a delta. look ahead for the zigzag varint field id. 259: @last_field.pop 260: field_id = read_i16() 261: else 262: # has a delta. add the delta to the last read field id. 263: field_id = @last_field.pop + modifier 264: end 265: 266: # if this happens to be a boolean field, the value is encoded in the type 267: if CompactTypes.is_bool_type?(type) 268: # save the boolean value in a special instance variable. 269: @bool_value = (type & 0x0f) == CompactTypes::BOOLEAN_TRUE 270: end 271: 272: # push the new field onto the field stack so we can keep the deltas going. 273: @last_field.push(field_id) 274: ["", CompactTypes.get_ttype(type & 0x0f), field_id] 275: end 276: end
# File lib/thrift/protocol/compact_protocol.rb, line 316 316: def read_i16 317: zig_zag_to_int(read_varint32()) 318: end
# File lib/thrift/protocol/compact_protocol.rb, line 316 316: def read_i16 317: zig_zag_to_int(read_varint32()) 318: end
# File lib/thrift/protocol/compact_protocol.rb, line 320 320: def read_i32 321: zig_zag_to_int(read_varint32()) 322: end
# File lib/thrift/protocol/compact_protocol.rb, line 320 320: def read_i32 321: zig_zag_to_int(read_varint32()) 322: end
# File lib/thrift/protocol/compact_protocol.rb, line 324 324: def read_i64 325: zig_zag_to_long(read_varint64()) 326: end
# File lib/thrift/protocol/compact_protocol.rb, line 324 324: def read_i64 325: zig_zag_to_long(read_varint64()) 326: end
# File lib/thrift/protocol/compact_protocol.rb, line 284 284: def read_list_begin 285: size_and_type = read_byte() 286: size = (size_and_type >> 4) & 0x0f 287: if size == 15 288: size = read_varint32() 289: end 290: type = CompactTypes.get_ttype(size_and_type) 291: [type, size] 292: end
# File lib/thrift/protocol/compact_protocol.rb, line 284 284: def read_list_begin 285: size_and_type = read_byte() 286: size = (size_and_type >> 4) & 0x0f 287: if size == 15 288: size = read_varint32() 289: end 290: type = CompactTypes.get_ttype(size_and_type) 291: [type, size] 292: end
# File lib/thrift/protocol/compact_protocol.rb, line 278 278: def read_map_begin 279: size = read_varint32() 280: key_and_value_type = size == 0 ? 0 : read_byte() 281: [CompactTypes.get_ttype(key_and_value_type >> 4), CompactTypes.get_ttype(key_and_value_type & 0xf), size] 282: end
# File lib/thrift/protocol/compact_protocol.rb, line 278 278: def read_map_begin 279: size = read_varint32() 280: key_and_value_type = size == 0 ? 0 : read_byte() 281: [CompactTypes.get_ttype(key_and_value_type >> 4), CompactTypes.get_ttype(key_and_value_type & 0xf), size] 282: end
# File lib/thrift/protocol/compact_protocol.rb, line 218 218: def read_message_begin 219: protocol_id = read_byte() 220: if protocol_id != PROTOCOL_ID 221: raise ProtocolException.new("Expected protocol id #{PROTOCOL_ID} but got #{protocol_id}") 222: end 223: 224: version_and_type = read_byte() 225: version = version_and_type & VERSION_MASK 226: if (version != VERSION) 227: raise ProtocolException.new("Expected version #{VERSION} but got #{version}"); 228: end 229: 230: type = (version_and_type >> TYPE_SHIFT_AMOUNT) & 0x03 231: seqid = read_varint32() 232: messageName = read_string() 233: [messageName, type, seqid] 234: end
# File lib/thrift/protocol/compact_protocol.rb, line 218 218: def read_message_begin 219: protocol_id = read_byte() 220: if protocol_id != PROTOCOL_ID 221: raise ProtocolException.new("Expected protocol id #{PROTOCOL_ID} but got #{protocol_id}") 222: end 223: 224: version_and_type = read_byte() 225: version = version_and_type & VERSION_MASK 226: if (version != VERSION) 227: raise ProtocolException.new("Expected version #{VERSION} but got #{version}"); 228: end 229: 230: type = (version_and_type >> TYPE_SHIFT_AMOUNT) & 0x03 231: seqid = read_varint32() 232: messageName = read_string() 233: [messageName, type, seqid] 234: end
# File lib/thrift/protocol/compact_protocol.rb, line 294 294: def read_set_begin 295: read_list_begin 296: end
# File lib/thrift/protocol/compact_protocol.rb, line 294 294: def read_set_begin 295: read_list_begin 296: end
# File lib/thrift/protocol/compact_protocol.rb, line 334 334: def read_string 335: size = read_varint32() 336: buffer = trans.read_all(size) 337: Bytes.convert_to_string(buffer) 338: end
# File lib/thrift/protocol/compact_protocol.rb, line 334 334: def read_string 335: size = read_varint32() 336: buffer = trans.read_all(size) 337: Bytes.convert_to_string(buffer) 338: end
# File lib/thrift/protocol/compact_protocol.rb, line 236 236: def read_struct_begin 237: @last_field.push(0) 238: "" 239: end
# File lib/thrift/protocol/compact_protocol.rb, line 236 236: def read_struct_begin 237: @last_field.push(0) 238: "" 239: end
# File lib/thrift/protocol/compact_protocol.rb, line 241 241: def read_struct_end 242: @last_field.pop() 243: nil 244: end
# File lib/thrift/protocol/compact_protocol.rb, line 241 241: def read_struct_end 242: @last_field.pop() 243: nil 244: end
# File lib/thrift/protocol/compact_protocol.rb, line 180 180: def write_bool(bool) 181: type = bool ? CompactTypes::BOOLEAN_TRUE : CompactTypes::BOOLEAN_FALSE 182: unless @boolean_field.nil? 183: # we haven't written the field header yet 184: write_field_begin_internal(@boolean_field.first, @boolean_field.last, type) 185: @boolean_field = nil 186: else 187: # we're not part of a field, so just write the value. 188: write_byte(type) 189: end 190: end
# File lib/thrift/protocol/compact_protocol.rb, line 180 180: def write_bool(bool) 181: type = bool ? CompactTypes::BOOLEAN_TRUE : CompactTypes::BOOLEAN_FALSE 182: unless @boolean_field.nil? 183: # we haven't written the field header yet 184: write_field_begin_internal(@boolean_field.first, @boolean_field.last, type) 185: @boolean_field = nil 186: else 187: # we're not part of a field, so just write the value. 188: write_byte(type) 189: end 190: end
# File lib/thrift/protocol/compact_protocol.rb, line 192 192: def write_byte(byte) 193: @trans.write([byte].pack('c')) 194: end
# File lib/thrift/protocol/compact_protocol.rb, line 192 192: def write_byte(byte) 193: @trans.write([byte].pack('c')) 194: end
# File lib/thrift/protocol/compact_protocol.rb, line 208 208: def write_double(dub) 209: @trans.write([dub].pack("G").reverse) 210: end
# File lib/thrift/protocol/compact_protocol.rb, line 208 208: def write_double(dub) 209: @trans.write([dub].pack("G").reverse) 210: end
# File lib/thrift/protocol/compact_protocol.rb, line 124 124: def write_field_begin(name, type, id) 125: if type == Types::BOOL 126: # we want to possibly include the value, so we'll wait. 127: @boolean_field = [type, id] 128: else 129: write_field_begin_internal(type, id) 130: end 131: nil 132: end
# File lib/thrift/protocol/compact_protocol.rb, line 124 124: def write_field_begin(name, type, id) 125: if type == Types::BOOL 126: # we want to possibly include the value, so we'll wait. 127: @boolean_field = [type, id] 128: else 129: write_field_begin_internal(type, id) 130: end 131: nil 132: end
The workhorse of writeFieldBegin. It has the option of doing a ‘type override’ of the type header. This is used specifically in the boolean field case.
# File lib/thrift/protocol/compact_protocol.rb, line 139 139: def write_field_begin_internal(type, id, type_override=nil) 140: last_id = @last_field.pop 141: 142: # if there's a type override, use that. 143: typeToWrite = type_override || CompactTypes.get_compact_type(type) 144: 145: # check if we can use delta encoding for the field id 146: if id > last_id && id - last_id <= 15 147: # write them together 148: write_byte((id - last_id) << 4 | typeToWrite) 149: else 150: # write them separate 151: write_byte(typeToWrite) 152: write_i16(id) 153: end 154: 155: @last_field.push(id) 156: nil 157: end
The workhorse of writeFieldBegin. It has the option of doing a ‘type override’ of the type header. This is used specifically in the boolean field case.
# File lib/thrift/protocol/compact_protocol.rb, line 139 139: def write_field_begin_internal(type, id, type_override=nil) 140: last_id = @last_field.pop 141: 142: # if there's a type override, use that. 143: typeToWrite = type_override || CompactTypes.get_compact_type(type) 144: 145: # check if we can use delta encoding for the field id 146: if id > last_id && id - last_id <= 15 147: # write them together 148: write_byte((id - last_id) << 4 | typeToWrite) 149: else 150: # write them separate 151: write_byte(typeToWrite) 152: write_i16(id) 153: end 154: 155: @last_field.push(id) 156: nil 157: end
# File lib/thrift/protocol/compact_protocol.rb, line 159 159: def write_field_stop 160: write_byte(Types::STOP) 161: end
# File lib/thrift/protocol/compact_protocol.rb, line 159 159: def write_field_stop 160: write_byte(Types::STOP) 161: end
# File lib/thrift/protocol/compact_protocol.rb, line 196 196: def write_i16(i16) 197: write_varint32(int_to_zig_zag(i16)) 198: end
# File lib/thrift/protocol/compact_protocol.rb, line 196 196: def write_i16(i16) 197: write_varint32(int_to_zig_zag(i16)) 198: end
# File lib/thrift/protocol/compact_protocol.rb, line 200 200: def write_i32(i32) 201: write_varint32(int_to_zig_zag(i32)) 202: end
# File lib/thrift/protocol/compact_protocol.rb, line 200 200: def write_i32(i32) 201: write_varint32(int_to_zig_zag(i32)) 202: end
# File lib/thrift/protocol/compact_protocol.rb, line 204 204: def write_i64(i64) 205: write_varint64(long_to_zig_zag(i64)) 206: end
# File lib/thrift/protocol/compact_protocol.rb, line 204 204: def write_i64(i64) 205: write_varint64(long_to_zig_zag(i64)) 206: end
# File lib/thrift/protocol/compact_protocol.rb, line 172 172: def write_list_begin(etype, size) 173: write_collection_begin(etype, size) 174: end
# File lib/thrift/protocol/compact_protocol.rb, line 172 172: def write_list_begin(etype, size) 173: write_collection_begin(etype, size) 174: end
# File lib/thrift/protocol/compact_protocol.rb, line 163 163: def write_map_begin(ktype, vtype, size) 164: if (size == 0) 165: write_byte(0) 166: else 167: write_varint32(size) 168: write_byte(CompactTypes.get_compact_type(ktype) << 4 | CompactTypes.get_compact_type(vtype)) 169: end 170: end
# File lib/thrift/protocol/compact_protocol.rb, line 163 163: def write_map_begin(ktype, vtype, size) 164: if (size == 0) 165: write_byte(0) 166: else 167: write_varint32(size) 168: write_byte(CompactTypes.get_compact_type(ktype) << 4 | CompactTypes.get_compact_type(vtype)) 169: end 170: end
# File lib/thrift/protocol/compact_protocol.rb, line 106 106: def write_message_begin(name, type, seqid) 107: write_byte(PROTOCOL_ID) 108: write_byte((VERSION & VERSION_MASK) | ((type << TYPE_SHIFT_AMOUNT) & TYPE_MASK)) 109: write_varint32(seqid) 110: write_string(name) 111: nil 112: end
# File lib/thrift/protocol/compact_protocol.rb, line 106 106: def write_message_begin(name, type, seqid) 107: write_byte(PROTOCOL_ID) 108: write_byte((VERSION & VERSION_MASK) | ((type << TYPE_SHIFT_AMOUNT) & TYPE_MASK)) 109: write_varint32(seqid) 110: write_string(name) 111: nil 112: end
# File lib/thrift/protocol/compact_protocol.rb, line 176 176: def write_set_begin(etype, size) 177: write_collection_begin(etype, size); 178: end
# File lib/thrift/protocol/compact_protocol.rb, line 176 176: def write_set_begin(etype, size) 177: write_collection_begin(etype, size); 178: end
# File lib/thrift/protocol/compact_protocol.rb, line 212 212: def write_string(str) 213: str = Bytes.convert_to_utf8_byte_buffer(str) 214: write_varint32(str.length) 215: @trans.write(str) 216: end
# File lib/thrift/protocol/compact_protocol.rb, line 212 212: def write_string(str) 213: str = Bytes.convert_to_utf8_byte_buffer(str) 214: write_varint32(str.length) 215: @trans.write(str) 216: end
# File lib/thrift/protocol/compact_protocol.rb, line 114 114: def write_struct_begin(name) 115: @last_field.push(0) 116: nil 117: end
# File lib/thrift/protocol/compact_protocol.rb, line 114 114: def write_struct_begin(name) 115: @last_field.push(0) 116: nil 117: end
# File lib/thrift/protocol/compact_protocol.rb, line 119 119: def write_struct_end 120: @last_field.pop 121: nil 122: end