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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
|
# encoding: binary
# frozen_string_literal: true
require "amq/protocol/type_constants"
require "amq/protocol/table_value_encoder"
require "amq/protocol/table_value_decoder"
module AMQ
module Protocol
class Table
#
# Behaviors
#
include TypeConstants
# Pack format string
PACK_UINT32_BE = 'N'.freeze
#
# API
#
class InvalidTableError < StandardError
def initialize(key, value)
super("Invalid table value on key #{key}: #{value.inspect} (#{value.class})")
end
end
def self.encode(table)
buffer = +''
table ||= {}
table.each do |key, value|
key = key.to_s # it can be a symbol as well
buffer << key.bytesize.chr << key
case value
when Hash
buffer << TYPE_HASH
buffer << self.encode(value)
else
buffer << TableValueEncoder.encode(value).force_encoding(buffer.encoding)
end
end
[buffer.bytesize].pack(PACK_UINT32_BE) << buffer
end
def self.decode(data)
table = {}
table_length = data.unpack1(PACK_UINT32_BE)
return table if table_length.zero?
offset = 4
while offset <= table_length
key, offset = decode_table_key(data, offset)
type, offset = TableValueDecoder.decode_value_type(data, offset)
table[key] = case type
when TYPE_STRING
v, offset = TableValueDecoder.decode_string(data, offset)
v
when TYPE_BYTE_ARRAY
# Ruby doesn't have a direct counterpart to
# ByteBuffer or byte[], so using a string feels
# more appropriate than an array of fixnums
v, offset = TableValueDecoder.decode_string(data, offset)
v
when TYPE_INTEGER
v, offset = TableValueDecoder.decode_integer(data, offset)
v
when TYPE_DECIMAL
v, offset = TableValueDecoder.decode_big_decimal(data, offset)
v
when TYPE_TIME
v, offset = TableValueDecoder.decode_time(data, offset)
v
when TYPE_HASH
v, offset = TableValueDecoder.decode_hash(data, offset)
v
when TYPE_BOOLEAN
v, offset = TableValueDecoder.decode_boolean(data, offset)
v
when TYPE_BYTE
v, offset = TableValueDecoder.decode_byte(data, offset)
v
when TYPE_SIGNED_16BIT
v, offset = TableValueDecoder.decode_short(data, offset)
v
when TYPE_SIGNED_64BIT
v, offset = TableValueDecoder.decode_long(data, offset)
v
when TYPE_32BIT_FLOAT
v, offset = TableValueDecoder.decode_32bit_float(data, offset)
v
when TYPE_64BIT_FLOAT
v, offset = TableValueDecoder.decode_64bit_float(data, offset)
v
when TYPE_VOID
nil
when TYPE_ARRAY
v, offset = TableValueDecoder.decode_array(data, offset)
v
else
raise ArgumentError, "Not a valid type: #{type.inspect}\nData: #{data.inspect}\nUnprocessed data: #{data[offset..-1].inspect}\nOffset: #{offset}\nTotal size: #{table_length}\nProcessed data: #{table.inspect}"
end
end
table
end
def self.length(data)
data.unpack1(PACK_UINT32_BE)
end
def self.hash_size(value)
acc = 0
value.each do |k, v|
acc += (1 + k.to_s.bytesize)
acc += TableValueEncoder.field_value_size(v)
end
acc
end
def self.decode_table_key(data, offset)
key_length = data.getbyte(offset)
offset += 1
key = data.byteslice(offset, key_length)
offset += key_length
[key, offset]
end
end # Table
end # Protocol
end # AMQ
|
