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
|
# frozen_string_literal: true
module Riddle
class Client
# This class takes care of the translation of ints, strings and arrays to
# the format required by the Sphinx service.
class Message
def initialize
@message = StringIO.new String.new(""), "w"
@message.set_encoding 'ASCII-8BIT'
@size_method = @message.respond_to?(:bytesize) ? :bytesize : :length
end
# Append raw data (only use if you know what you're doing)
def append(*args)
args.each { |arg| @message << arg }
end
# Append a string's length, then the string itself
def append_string(str)
string = Riddle.encode(str.dup, 'ASCII-8BIT')
@message << [string.send(@size_method)].pack('N') + string
end
# Append an integer
def append_int(int)
@message << [int.to_i].pack('N')
end
def append_64bit_int(int)
@message << [int.to_i >> 32, int.to_i & 0xFFFFFFFF].pack('NN')
end
# Append a float
def append_float(float)
@message << [float].pack('f').unpack('L*').pack("N")
end
def append_boolean(bool)
append_int(bool ? 1 : 0)
end
# Append multiple integers
def append_ints(*ints)
ints.each { |int| append_int(int) }
end
def append_64bit_ints(*ints)
ints.each { |int| append_64bit_int(int) }
end
# Append multiple floats
def append_floats(*floats)
floats.each { |float| append_float(float) }
end
# Append an array of strings - first appends the length of the array,
# then each item's length and value.
def append_array(array)
append_int(array.length)
array.each { |item| append_string(item) }
end
# Returns the entire message
def to_s
@message.string
end
end
end
end
|
