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
|
# Copyright (C) 2014-2017 MongoDB, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
module Mongo
module Protocol
module Serializers
# Class used to define a bitvector for a MongoDB wire protocol message.
#
# Defines serialization strategy upon initialization.
#
# @api private
class BitVector
# Initializes a BitVector with a layout
#
# @param layout [ Array<Symbol> ] the array of fields in the bit vector
def initialize(layout)
@masks = {}
layout.each_with_index do |field, index|
@masks[field] = 2**index if field
end
end
# Serializes vector by encoding each symbol according to its mask
#
# @param buffer [ String ] Buffer to receive the serialized vector
# @param value [ Array<Symbol> ] Array of flags to encode
#
# @return [ String ] Buffer that received the serialized vector
def serialize(buffer, value, validating_keys = BSON::Config.validating_keys?)
bits = 0
value.each { |flag| bits |= (@masks[flag] || 0) }
buffer.put_int32(bits)
end
# Deserializes vector by decoding the symbol according to its mask
#
# @param [ String ] buffer Buffer containing the vector to be deserialized.
#
# @return [ Array<Symbol> ] Flags contained in the vector
def deserialize(buffer)
vector = buffer.get_int32
flags = []
@masks.each do |flag, mask|
flags << flag if mask & vector != 0
end
flags
end
end
end
end
end
|
