zinx 0.0.5 → 0.0.6
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- data/lib/sphinx/sphinx.rb +5 -5
- data/lib/sphinx/sphinx/client.rb +1125 -1125
- data/lib/sphinx/sphinx/request.rb +50 -50
- data/lib/sphinx/sphinx/response.rb +68 -68
- data/lib/zinx.rb +271 -271
- data/test/test.rb +11 -11
- metadata +4 -4
|
@@ -1,50 +1,50 @@
|
|
|
1
|
-
module Sphinx
|
|
2
|
-
# Pack ints, floats, strings, and arrays to internal representation
|
|
3
|
-
# needed by Sphinx search engine.
|
|
4
|
-
class Request
|
|
5
|
-
# Initialize new request.
|
|
6
|
-
def initialize
|
|
7
|
-
@request = ''
|
|
8
|
-
end
|
|
9
|
-
|
|
10
|
-
# Put int(s) to request.
|
|
11
|
-
def put_int(*ints)
|
|
12
|
-
ints.each { |i| @request << [i].pack('N') }
|
|
13
|
-
end
|
|
14
|
-
|
|
15
|
-
# Put 64-bit int(s) to request.
|
|
16
|
-
def put_int64(*ints)
|
|
17
|
-
ints.each { |i| @request << [i].pack('q').reverse }#[i >> 32, i & ((1 << 32) - 1)].pack('NN') }
|
|
18
|
-
end
|
|
19
|
-
|
|
20
|
-
# Put string(s) to request (first length, then the string itself).
|
|
21
|
-
def put_string(*strings)
|
|
22
|
-
strings.each { |s| @request << [s.length].pack('N') + s }
|
|
23
|
-
end
|
|
24
|
-
|
|
25
|
-
# Put float(s) to request.
|
|
26
|
-
def put_float(*floats)
|
|
27
|
-
floats.each do |f|
|
|
28
|
-
t1 = [f].pack('f') # machine order
|
|
29
|
-
t2 = t1.unpack('L*').first # int in machine order
|
|
30
|
-
@request << [t2].pack('N')
|
|
31
|
-
end
|
|
32
|
-
end
|
|
33
|
-
|
|
34
|
-
# Put array of ints to request (first length, then the array itself)
|
|
35
|
-
def put_int_array(arr)
|
|
36
|
-
put_int arr.length, *arr
|
|
37
|
-
end
|
|
38
|
-
|
|
39
|
-
# Put array of 64-bit ints to request (first length, then the array itself)
|
|
40
|
-
def put_int64_array(arr)
|
|
41
|
-
put_int arr.length
|
|
42
|
-
put_int64(*arr)
|
|
43
|
-
end
|
|
44
|
-
|
|
45
|
-
# Returns the entire message
|
|
46
|
-
def to_s
|
|
47
|
-
@request
|
|
48
|
-
end
|
|
49
|
-
end
|
|
50
|
-
end
|
|
1
|
+
module Sphinx
|
|
2
|
+
# Pack ints, floats, strings, and arrays to internal representation
|
|
3
|
+
# needed by Sphinx search engine.
|
|
4
|
+
class Request
|
|
5
|
+
# Initialize new request.
|
|
6
|
+
def initialize
|
|
7
|
+
@request = ''
|
|
8
|
+
end
|
|
9
|
+
|
|
10
|
+
# Put int(s) to request.
|
|
11
|
+
def put_int(*ints)
|
|
12
|
+
ints.each { |i| @request << [i].pack('N') }
|
|
13
|
+
end
|
|
14
|
+
|
|
15
|
+
# Put 64-bit int(s) to request.
|
|
16
|
+
def put_int64(*ints)
|
|
17
|
+
ints.each { |i| @request << [i].pack('q').reverse }#[i >> 32, i & ((1 << 32) - 1)].pack('NN') }
|
|
18
|
+
end
|
|
19
|
+
|
|
20
|
+
# Put string(s) to request (first length, then the string itself).
|
|
21
|
+
def put_string(*strings)
|
|
22
|
+
strings.each { |s| @request << [s.length].pack('N') + s }
|
|
23
|
+
end
|
|
24
|
+
|
|
25
|
+
# Put float(s) to request.
|
|
26
|
+
def put_float(*floats)
|
|
27
|
+
floats.each do |f|
|
|
28
|
+
t1 = [f].pack('f') # machine order
|
|
29
|
+
t2 = t1.unpack('L*').first # int in machine order
|
|
30
|
+
@request << [t2].pack('N')
|
|
31
|
+
end
|
|
32
|
+
end
|
|
33
|
+
|
|
34
|
+
# Put array of ints to request (first length, then the array itself)
|
|
35
|
+
def put_int_array(arr)
|
|
36
|
+
put_int arr.length, *arr
|
|
37
|
+
end
|
|
38
|
+
|
|
39
|
+
# Put array of 64-bit ints to request (first length, then the array itself)
|
|
40
|
+
def put_int64_array(arr)
|
|
41
|
+
put_int arr.length
|
|
42
|
+
put_int64(*arr)
|
|
43
|
+
end
|
|
44
|
+
|
|
45
|
+
# Returns the entire message
|
|
46
|
+
def to_s
|
|
47
|
+
@request
|
|
48
|
+
end
|
|
49
|
+
end
|
|
50
|
+
end
|
|
@@ -1,69 +1,69 @@
|
|
|
1
|
-
module Sphinx
|
|
2
|
-
# Unpack internal Sphinx representation of ints, floats, strings, and arrays.
|
|
3
|
-
# needed by Sphinx search engine.
|
|
4
|
-
class Response
|
|
5
|
-
# Initialize new request.
|
|
6
|
-
def initialize(response)
|
|
7
|
-
@response = response
|
|
8
|
-
@position = 0
|
|
9
|
-
@size = response.length
|
|
10
|
-
end
|
|
11
|
-
|
|
12
|
-
# Gets current stream position.
|
|
13
|
-
def position
|
|
14
|
-
@position
|
|
15
|
-
end
|
|
16
|
-
|
|
17
|
-
# Gets response size.
|
|
18
|
-
def size
|
|
19
|
-
@size
|
|
20
|
-
end
|
|
21
|
-
|
|
22
|
-
# Returns <tt>true</tt> when response stream is out.
|
|
23
|
-
def eof?
|
|
24
|
-
@position >= @size
|
|
25
|
-
end
|
|
26
|
-
|
|
27
|
-
# Get int from stream.
|
|
28
|
-
def get_int
|
|
29
|
-
raise EOFError if @position + 4 > @size
|
|
30
|
-
value = @response[@position, 4].unpack('N*').first
|
|
31
|
-
@position += 4
|
|
32
|
-
return value
|
|
33
|
-
end
|
|
34
|
-
|
|
35
|
-
# Get 64-bit int from stream.
|
|
36
|
-
def get_int64
|
|
37
|
-
raise EOFError if @position + 8 > @size
|
|
38
|
-
hi, lo = @response[@position, 8].unpack('N*N*')
|
|
39
|
-
@position += 8
|
|
40
|
-
return (hi << 32) + lo
|
|
41
|
-
end
|
|
42
|
-
|
|
43
|
-
# Get array of <tt>count</tt> ints from stream.
|
|
44
|
-
def get_ints(count)
|
|
45
|
-
length = 4 * count
|
|
46
|
-
raise EOFError if @position + length > @size
|
|
47
|
-
values = @response[@position, length].unpack('N*' * count)
|
|
48
|
-
@position += length
|
|
49
|
-
return values
|
|
50
|
-
end
|
|
51
|
-
|
|
52
|
-
# Get string from stream.
|
|
53
|
-
def get_string
|
|
54
|
-
length = get_int
|
|
55
|
-
raise EOFError if @position + length > @size
|
|
56
|
-
value = length > 0 ? @response[@position, length] : ''
|
|
57
|
-
@position += length
|
|
58
|
-
return value
|
|
59
|
-
end
|
|
60
|
-
|
|
61
|
-
# Get float from stream.
|
|
62
|
-
def get_float
|
|
63
|
-
raise EOFError if @position + 4 > @size
|
|
64
|
-
uval = @response[@position, 4].unpack('N*').first;
|
|
65
|
-
@position += 4
|
|
66
|
-
return ([uval].pack('L')).unpack('f*').first
|
|
67
|
-
end
|
|
68
|
-
end
|
|
1
|
+
module Sphinx
|
|
2
|
+
# Unpack internal Sphinx representation of ints, floats, strings, and arrays.
|
|
3
|
+
# needed by Sphinx search engine.
|
|
4
|
+
class Response
|
|
5
|
+
# Initialize new request.
|
|
6
|
+
def initialize(response)
|
|
7
|
+
@response = response
|
|
8
|
+
@position = 0
|
|
9
|
+
@size = response.length
|
|
10
|
+
end
|
|
11
|
+
|
|
12
|
+
# Gets current stream position.
|
|
13
|
+
def position
|
|
14
|
+
@position
|
|
15
|
+
end
|
|
16
|
+
|
|
17
|
+
# Gets response size.
|
|
18
|
+
def size
|
|
19
|
+
@size
|
|
20
|
+
end
|
|
21
|
+
|
|
22
|
+
# Returns <tt>true</tt> when response stream is out.
|
|
23
|
+
def eof?
|
|
24
|
+
@position >= @size
|
|
25
|
+
end
|
|
26
|
+
|
|
27
|
+
# Get int from stream.
|
|
28
|
+
def get_int
|
|
29
|
+
raise EOFError if @position + 4 > @size
|
|
30
|
+
value = @response[@position, 4].unpack('N*').first
|
|
31
|
+
@position += 4
|
|
32
|
+
return value
|
|
33
|
+
end
|
|
34
|
+
|
|
35
|
+
# Get 64-bit int from stream.
|
|
36
|
+
def get_int64
|
|
37
|
+
raise EOFError if @position + 8 > @size
|
|
38
|
+
hi, lo = @response[@position, 8].unpack('N*N*')
|
|
39
|
+
@position += 8
|
|
40
|
+
return (hi << 32) + lo
|
|
41
|
+
end
|
|
42
|
+
|
|
43
|
+
# Get array of <tt>count</tt> ints from stream.
|
|
44
|
+
def get_ints(count)
|
|
45
|
+
length = 4 * count
|
|
46
|
+
raise EOFError if @position + length > @size
|
|
47
|
+
values = @response[@position, length].unpack('N*' * count)
|
|
48
|
+
@position += length
|
|
49
|
+
return values
|
|
50
|
+
end
|
|
51
|
+
|
|
52
|
+
# Get string from stream.
|
|
53
|
+
def get_string
|
|
54
|
+
length = get_int
|
|
55
|
+
raise EOFError if @position + length > @size
|
|
56
|
+
value = length > 0 ? @response[@position, length] : ''
|
|
57
|
+
@position += length
|
|
58
|
+
return value
|
|
59
|
+
end
|
|
60
|
+
|
|
61
|
+
# Get float from stream.
|
|
62
|
+
def get_float
|
|
63
|
+
raise EOFError if @position + 4 > @size
|
|
64
|
+
uval = @response[@position, 4].unpack('N*').first;
|
|
65
|
+
@position += 4
|
|
66
|
+
return ([uval].pack('L')).unpack('f*').first
|
|
67
|
+
end
|
|
68
|
+
end
|
|
69
69
|
end
|
data/lib/zinx.rb
CHANGED
|
@@ -1,272 +1,272 @@
|
|
|
1
|
-
require File.dirname(__FILE__) + '/sphinx/sphinx'
|
|
2
|
-
|
|
3
|
-
module Zinx
|
|
4
|
-
|
|
5
|
-
class Client < Sphinx::Client
|
|
6
|
-
|
|
7
|
-
class << self
|
|
8
|
-
attr_reader :client, # Sphinx::Client instance
|
|
9
|
-
:query, # Query text specified by the user
|
|
10
|
-
:index_name, # Index name to run the query against
|
|
11
|
-
:multiple_queries, # Boolean if needs to run more then one query (add_query)
|
|
12
|
-
:field_weights, # Hash of field weights
|
|
13
|
-
:index_weights, # Hash of index weights
|
|
14
|
-
:results # Results from Sphinx
|
|
15
|
-
|
|
16
|
-
def filter(field, value, exclude = false)
|
|
17
|
-
@client.SetFilter(field, value.instance_of?(Array) ? value : [value], exclude)
|
|
18
|
-
end
|
|
19
|
-
|
|
20
|
-
def filter_range(field, min, max, exclude = false)
|
|
21
|
-
@client.SetFilterRange(field, min, max, exclude)
|
|
22
|
-
end
|
|
23
|
-
|
|
24
|
-
def filter_float_range(field, min, max, exclude = false)
|
|
25
|
-
@client.SetFilterFloatRange(field, min, max, exclude)
|
|
26
|
-
end
|
|
27
|
-
|
|
28
|
-
def sort(mode, value = '')
|
|
29
|
-
@client.SetSortMode(mode, value)
|
|
30
|
-
end
|
|
31
|
-
|
|
32
|
-
def group(mode, value)
|
|
33
|
-
@client.SetGroupBy(value, mode)
|
|
34
|
-
end
|
|
35
|
-
|
|
36
|
-
def group_distinct(field)
|
|
37
|
-
@client.SetGroupDistinct(field)
|
|
38
|
-
end
|
|
39
|
-
|
|
40
|
-
def select(value)
|
|
41
|
-
@client.SetSelect(value)
|
|
42
|
-
end
|
|
43
|
-
|
|
44
|
-
def last_error
|
|
45
|
-
@client.GetLastError
|
|
46
|
-
end
|
|
47
|
-
|
|
48
|
-
def last_warning
|
|
49
|
-
@client.GetLastWarning
|
|
50
|
-
end
|
|
51
|
-
|
|
52
|
-
def server(host, port)
|
|
53
|
-
@client.SetServer(host, port)
|
|
54
|
-
end
|
|
55
|
-
|
|
56
|
-
def limits(offset, limit, max = 0, cutoff = 0)
|
|
57
|
-
@client.SetLimits(offset, limit, max, cutoff)
|
|
58
|
-
end
|
|
59
|
-
|
|
60
|
-
def max_query_time(time)
|
|
61
|
-
@client.SetMaxQueryTime(time)
|
|
62
|
-
end
|
|
63
|
-
|
|
64
|
-
def match_mode(mode)
|
|
65
|
-
@client.SetMatchMode(mode)
|
|
66
|
-
end
|
|
67
|
-
|
|
68
|
-
def ranking_mode(mode)
|
|
69
|
-
@client.SetRankingMode(mode)
|
|
70
|
-
end
|
|
71
|
-
|
|
72
|
-
def field_weight(field, weight)
|
|
73
|
-
@field_weights[field] = weight
|
|
74
|
-
end
|
|
75
|
-
|
|
76
|
-
def field_weights(hash)
|
|
77
|
-
@field_weights = hash
|
|
78
|
-
end
|
|
79
|
-
|
|
80
|
-
def index_weight(index, weight)
|
|
81
|
-
@index_weights[index] = weight
|
|
82
|
-
end
|
|
83
|
-
|
|
84
|
-
def index_weights(hash)
|
|
85
|
-
@index_weights = hash
|
|
86
|
-
end
|
|
87
|
-
|
|
88
|
-
def id_range(min, max)
|
|
89
|
-
@client.SetIDRange(min, max)
|
|
90
|
-
end
|
|
91
|
-
|
|
92
|
-
def geo_anchor(attr_lat, attr_lng, lat, lng)
|
|
93
|
-
@client.Set(attr_lat, attr_lng, lat, lng)
|
|
94
|
-
end
|
|
95
|
-
|
|
96
|
-
def retries(count, delay = 0)
|
|
97
|
-
@client.SetRetries(count, delay)
|
|
98
|
-
end
|
|
99
|
-
|
|
100
|
-
def override(field, type, values)
|
|
101
|
-
@client.SetOverride(field, type, values)
|
|
102
|
-
end
|
|
103
|
-
|
|
104
|
-
def reset
|
|
105
|
-
@client.ResetFilters
|
|
106
|
-
@client.ResetGroupBy
|
|
107
|
-
@client.ResetOverrides
|
|
108
|
-
end
|
|
109
|
-
|
|
110
|
-
def reset_filters
|
|
111
|
-
@client.ResetFilters
|
|
112
|
-
end
|
|
113
|
-
|
|
114
|
-
def reset_groups
|
|
115
|
-
@client.ResetGroupBy
|
|
116
|
-
end
|
|
117
|
-
|
|
118
|
-
def reset_overrides
|
|
119
|
-
@client.ResetOverrides
|
|
120
|
-
end
|
|
121
|
-
|
|
122
|
-
def build_excerpts(docs, index, words, opts = {})
|
|
123
|
-
init
|
|
124
|
-
@client.BuildExcerpts(docs, index, words, opts)
|
|
125
|
-
end
|
|
126
|
-
|
|
127
|
-
def excerpts(docs, words, opts = {})
|
|
128
|
-
run if @results.empty?
|
|
129
|
-
@client.BuildExcerpts(docs, @index_name, words, opts)
|
|
130
|
-
end
|
|
131
|
-
|
|
132
|
-
def build_keywords(query, index, hits)
|
|
133
|
-
init
|
|
134
|
-
@client.BuildKeywords(query, index, hits)
|
|
135
|
-
end
|
|
136
|
-
|
|
137
|
-
def update(index, attrs, values, mva = false)
|
|
138
|
-
init
|
|
139
|
-
@client.UpdateAttributes(index, attrs, values, mva)
|
|
140
|
-
end
|
|
141
|
-
|
|
142
|
-
def results
|
|
143
|
-
run if @results.empty?
|
|
144
|
-
end
|
|
145
|
-
|
|
146
|
-
# syntax sugar for results[0].matches when using only one query
|
|
147
|
-
# you don't even have to call 'run' before using this
|
|
148
|
-
def matches
|
|
149
|
-
run if @results.empty?
|
|
150
|
-
!@multiple_queries && @results.count > 0 ? @results[0].matches : []
|
|
151
|
-
end
|
|
152
|
-
|
|
153
|
-
# must call run before accessing search results
|
|
154
|
-
def run
|
|
155
|
-
# set the weights
|
|
156
|
-
@client.SetFieldWeights(@field_weights) unless @field_weights.empty?
|
|
157
|
-
@client.SetIndexWeights(@index_weights) unless @index_weights.empty?
|
|
158
|
-
|
|
159
|
-
# run the query
|
|
160
|
-
if @multiple_queries
|
|
161
|
-
q = @client.RunQueries
|
|
162
|
-
q.each do |result|
|
|
163
|
-
@results << Result.new(result)
|
|
164
|
-
end
|
|
165
|
-
@results
|
|
166
|
-
else
|
|
167
|
-
q = @client.Query(@query, @index_name)
|
|
168
|
-
@results << Result.new(q)
|
|
169
|
-
@results.first
|
|
170
|
-
end
|
|
171
|
-
end
|
|
172
|
-
|
|
173
|
-
# add query for multiple queries
|
|
174
|
-
def add_query
|
|
175
|
-
@multiple_queries = true
|
|
176
|
-
@client.AddQuery(@query, @index_name)
|
|
177
|
-
end
|
|
178
|
-
|
|
179
|
-
# Entry point for searches
|
|
180
|
-
# Valid params are:
|
|
181
|
-
# :server => Sphinx server address (defaults to 'localhost')
|
|
182
|
-
# :port => Sphinx port number (defaults to 9312)
|
|
183
|
-
# :match_mode => Sphinx matching mode (defaults to Zinx::SPH_MATCH_EXTENDED)
|
|
184
|
-
# :index_name => Name of the index to search on
|
|
185
|
-
def search(query, params = {}, &block)
|
|
186
|
-
params[:query] = query
|
|
187
|
-
init(params)
|
|
188
|
-
yield self if block_given?
|
|
189
|
-
run
|
|
190
|
-
end
|
|
191
|
-
|
|
192
|
-
def init(params = {})
|
|
193
|
-
@client = Client.new
|
|
194
|
-
@client.SetServer(params[:server] || 'localhost', params[:port] || 9312)
|
|
195
|
-
@client.SetMatchMode(params[:match_mode] || Zinx::Client::SPH_MATCH_EXTENDED)
|
|
196
|
-
@query = params[:query]
|
|
197
|
-
@index_name = params[:index_name] || "*"
|
|
198
|
-
@multiple_queries = false
|
|
199
|
-
@results = []
|
|
200
|
-
@field_weights = {}
|
|
201
|
-
@index_weights = {}
|
|
202
|
-
end
|
|
203
|
-
end
|
|
204
|
-
end
|
|
205
|
-
|
|
206
|
-
class Match
|
|
207
|
-
def initialize(hash)
|
|
208
|
-
@match = hash
|
|
209
|
-
end
|
|
210
|
-
|
|
211
|
-
def each
|
|
212
|
-
@match.each do |m|
|
|
213
|
-
yield m
|
|
214
|
-
end
|
|
215
|
-
end
|
|
216
|
-
|
|
217
|
-
def method_missing(method, *args, &block)
|
|
218
|
-
if ['groupby', 'count', 'expr'].include?("#{method}")
|
|
219
|
-
@match["attrs"]["@#{method}"]
|
|
220
|
-
elsif ['id', 'weight', 'attrs'].include?("#{method}")
|
|
221
|
-
@match["#{method}"]
|
|
222
|
-
else
|
|
223
|
-
@match["attrs"]["#{method}"]
|
|
224
|
-
end
|
|
225
|
-
end
|
|
226
|
-
end
|
|
227
|
-
|
|
228
|
-
class Result
|
|
229
|
-
attr_reader :matches
|
|
230
|
-
|
|
231
|
-
def initialize(sphinx_hash)
|
|
232
|
-
@matches = []
|
|
233
|
-
@sphinx_hash = sphinx_hash
|
|
234
|
-
if @sphinx_hash.has_key?("matches")
|
|
235
|
-
@sphinx_hash["matches"].each do |match|
|
|
236
|
-
@matches << Match.new(match)
|
|
237
|
-
end
|
|
238
|
-
end
|
|
239
|
-
end
|
|
240
|
-
|
|
241
|
-
def method_missing(method, *args, &block)
|
|
242
|
-
@sphinx_hash["#{method}"]
|
|
243
|
-
end
|
|
244
|
-
end
|
|
245
|
-
|
|
246
|
-
module Search
|
|
247
|
-
class << self
|
|
248
|
-
attr_accessor :target
|
|
249
|
-
|
|
250
|
-
def delegate(*methods)
|
|
251
|
-
methods.each do |method|
|
|
252
|
-
define_method(method) do |*args, &block|
|
|
253
|
-
Search.target.send(method, *args, &block)
|
|
254
|
-
end
|
|
255
|
-
private method
|
|
256
|
-
end
|
|
257
|
-
end
|
|
258
|
-
end
|
|
259
|
-
|
|
260
|
-
self.target = Zinx::Client
|
|
261
|
-
|
|
262
|
-
delegate :filter, :filter_range, :filter_float_range, :sort, :group, :group_distinct,
|
|
263
|
-
:select, :last_error, :last_warning, :server, :limits, :max_query_time, :match_mode,
|
|
264
|
-
:ranking_mode, :field_weight, :field_weights, :index_weight, :index_weights,
|
|
265
|
-
:id_range, :geo_anchor, :retries, :override, :reset, :reset_filter, :reset_groups,
|
|
266
|
-
:reset_overrides, :build_excerpts, :excerpts, :build_keywords, :update, :results,
|
|
267
|
-
:matches, :run, :add_query, :search, :init
|
|
268
|
-
end
|
|
269
|
-
|
|
270
|
-
end
|
|
271
|
-
|
|
1
|
+
require File.dirname(__FILE__) + '/sphinx/sphinx'
|
|
2
|
+
|
|
3
|
+
module Zinx
|
|
4
|
+
|
|
5
|
+
class Client < Sphinx::Client
|
|
6
|
+
|
|
7
|
+
class << self
|
|
8
|
+
attr_reader :client, # Sphinx::Client instance
|
|
9
|
+
:query, # Query text specified by the user
|
|
10
|
+
:index_name, # Index name to run the query against
|
|
11
|
+
:multiple_queries, # Boolean if needs to run more then one query (add_query)
|
|
12
|
+
:field_weights, # Hash of field weights
|
|
13
|
+
:index_weights, # Hash of index weights
|
|
14
|
+
:results # Results from Sphinx
|
|
15
|
+
|
|
16
|
+
def filter(field, value, exclude = false)
|
|
17
|
+
@client.SetFilter(field, value.instance_of?(Array) ? value : [value], exclude)
|
|
18
|
+
end
|
|
19
|
+
|
|
20
|
+
def filter_range(field, min, max, exclude = false)
|
|
21
|
+
@client.SetFilterRange(field, min, max, exclude)
|
|
22
|
+
end
|
|
23
|
+
|
|
24
|
+
def filter_float_range(field, min, max, exclude = false)
|
|
25
|
+
@client.SetFilterFloatRange(field, min, max, exclude)
|
|
26
|
+
end
|
|
27
|
+
|
|
28
|
+
def sort(mode, value = '')
|
|
29
|
+
@client.SetSortMode(mode, value)
|
|
30
|
+
end
|
|
31
|
+
|
|
32
|
+
def group(mode, value)
|
|
33
|
+
@client.SetGroupBy(value, mode)
|
|
34
|
+
end
|
|
35
|
+
|
|
36
|
+
def group_distinct(field)
|
|
37
|
+
@client.SetGroupDistinct(field)
|
|
38
|
+
end
|
|
39
|
+
|
|
40
|
+
def select(value)
|
|
41
|
+
@client.SetSelect(value)
|
|
42
|
+
end
|
|
43
|
+
|
|
44
|
+
def last_error
|
|
45
|
+
@client.GetLastError
|
|
46
|
+
end
|
|
47
|
+
|
|
48
|
+
def last_warning
|
|
49
|
+
@client.GetLastWarning
|
|
50
|
+
end
|
|
51
|
+
|
|
52
|
+
def server(host, port)
|
|
53
|
+
@client.SetServer(host, port)
|
|
54
|
+
end
|
|
55
|
+
|
|
56
|
+
def limits(offset, limit, max = 0, cutoff = 0)
|
|
57
|
+
@client.SetLimits(offset, limit, max, cutoff)
|
|
58
|
+
end
|
|
59
|
+
|
|
60
|
+
def max_query_time(time)
|
|
61
|
+
@client.SetMaxQueryTime(time)
|
|
62
|
+
end
|
|
63
|
+
|
|
64
|
+
def match_mode(mode)
|
|
65
|
+
@client.SetMatchMode(mode)
|
|
66
|
+
end
|
|
67
|
+
|
|
68
|
+
def ranking_mode(mode)
|
|
69
|
+
@client.SetRankingMode(mode)
|
|
70
|
+
end
|
|
71
|
+
|
|
72
|
+
def field_weight(field, weight)
|
|
73
|
+
@field_weights[field] = weight
|
|
74
|
+
end
|
|
75
|
+
|
|
76
|
+
def field_weights(hash)
|
|
77
|
+
@field_weights = hash
|
|
78
|
+
end
|
|
79
|
+
|
|
80
|
+
def index_weight(index, weight)
|
|
81
|
+
@index_weights[index] = weight
|
|
82
|
+
end
|
|
83
|
+
|
|
84
|
+
def index_weights(hash)
|
|
85
|
+
@index_weights = hash
|
|
86
|
+
end
|
|
87
|
+
|
|
88
|
+
def id_range(min, max)
|
|
89
|
+
@client.SetIDRange(min, max)
|
|
90
|
+
end
|
|
91
|
+
|
|
92
|
+
def geo_anchor(attr_lat, attr_lng, lat, lng)
|
|
93
|
+
@client.Set(attr_lat, attr_lng, lat, lng)
|
|
94
|
+
end
|
|
95
|
+
|
|
96
|
+
def retries(count, delay = 0)
|
|
97
|
+
@client.SetRetries(count, delay)
|
|
98
|
+
end
|
|
99
|
+
|
|
100
|
+
def override(field, type, values)
|
|
101
|
+
@client.SetOverride(field, type, values)
|
|
102
|
+
end
|
|
103
|
+
|
|
104
|
+
def reset
|
|
105
|
+
@client.ResetFilters
|
|
106
|
+
@client.ResetGroupBy
|
|
107
|
+
@client.ResetOverrides
|
|
108
|
+
end
|
|
109
|
+
|
|
110
|
+
def reset_filters
|
|
111
|
+
@client.ResetFilters
|
|
112
|
+
end
|
|
113
|
+
|
|
114
|
+
def reset_groups
|
|
115
|
+
@client.ResetGroupBy
|
|
116
|
+
end
|
|
117
|
+
|
|
118
|
+
def reset_overrides
|
|
119
|
+
@client.ResetOverrides
|
|
120
|
+
end
|
|
121
|
+
|
|
122
|
+
def build_excerpts(docs, index, words, opts = {})
|
|
123
|
+
init
|
|
124
|
+
@client.BuildExcerpts(docs, index, words, opts)
|
|
125
|
+
end
|
|
126
|
+
|
|
127
|
+
def excerpts(docs, words, opts = {})
|
|
128
|
+
run if @results.empty?
|
|
129
|
+
@client.BuildExcerpts(docs, @index_name, words, opts)
|
|
130
|
+
end
|
|
131
|
+
|
|
132
|
+
def build_keywords(query, index, hits)
|
|
133
|
+
init
|
|
134
|
+
@client.BuildKeywords(query, index, hits)
|
|
135
|
+
end
|
|
136
|
+
|
|
137
|
+
def update(index, attrs, values, mva = false)
|
|
138
|
+
init
|
|
139
|
+
@client.UpdateAttributes(index, attrs, values, mva)
|
|
140
|
+
end
|
|
141
|
+
|
|
142
|
+
def results
|
|
143
|
+
run if @results.empty?
|
|
144
|
+
end
|
|
145
|
+
|
|
146
|
+
# syntax sugar for results[0].matches when using only one query
|
|
147
|
+
# you don't even have to call 'run' before using this
|
|
148
|
+
def matches
|
|
149
|
+
run if @results.empty?
|
|
150
|
+
!@multiple_queries && @results.count > 0 ? @results[0].matches : []
|
|
151
|
+
end
|
|
152
|
+
|
|
153
|
+
# must call run before accessing search results
|
|
154
|
+
def run
|
|
155
|
+
# set the weights
|
|
156
|
+
@client.SetFieldWeights(@field_weights) unless @field_weights.empty?
|
|
157
|
+
@client.SetIndexWeights(@index_weights) unless @index_weights.empty?
|
|
158
|
+
|
|
159
|
+
# run the query
|
|
160
|
+
if @multiple_queries
|
|
161
|
+
q = @client.RunQueries
|
|
162
|
+
q.each do |result|
|
|
163
|
+
@results << Result.new(result)
|
|
164
|
+
end
|
|
165
|
+
@results
|
|
166
|
+
else
|
|
167
|
+
q = @client.Query(@query, @index_name)
|
|
168
|
+
@results << Result.new(q)
|
|
169
|
+
@results.first
|
|
170
|
+
end
|
|
171
|
+
end
|
|
172
|
+
|
|
173
|
+
# add query for multiple queries
|
|
174
|
+
def add_query
|
|
175
|
+
@multiple_queries = true
|
|
176
|
+
@client.AddQuery(@query, @index_name)
|
|
177
|
+
end
|
|
178
|
+
|
|
179
|
+
# Entry point for searches
|
|
180
|
+
# Valid params are:
|
|
181
|
+
# :server => Sphinx server address (defaults to 'localhost')
|
|
182
|
+
# :port => Sphinx port number (defaults to 9312)
|
|
183
|
+
# :match_mode => Sphinx matching mode (defaults to Zinx::SPH_MATCH_EXTENDED)
|
|
184
|
+
# :index_name => Name of the index to search on
|
|
185
|
+
def search(query, params = {}, &block)
|
|
186
|
+
params[:query] = query
|
|
187
|
+
init(params)
|
|
188
|
+
yield self if block_given?
|
|
189
|
+
run
|
|
190
|
+
end
|
|
191
|
+
|
|
192
|
+
def init(params = {})
|
|
193
|
+
@client = Client.new
|
|
194
|
+
@client.SetServer(params[:server] || 'localhost', params[:port] || 9312)
|
|
195
|
+
@client.SetMatchMode(params[:match_mode] || Zinx::Client::SPH_MATCH_EXTENDED)
|
|
196
|
+
@query = params[:query]
|
|
197
|
+
@index_name = params[:index_name] || "*"
|
|
198
|
+
@multiple_queries = false
|
|
199
|
+
@results = []
|
|
200
|
+
@field_weights = {}
|
|
201
|
+
@index_weights = {}
|
|
202
|
+
end
|
|
203
|
+
end
|
|
204
|
+
end
|
|
205
|
+
|
|
206
|
+
class Match
|
|
207
|
+
def initialize(hash)
|
|
208
|
+
@match = hash
|
|
209
|
+
end
|
|
210
|
+
|
|
211
|
+
def each
|
|
212
|
+
@match.each do |m|
|
|
213
|
+
yield m
|
|
214
|
+
end
|
|
215
|
+
end
|
|
216
|
+
|
|
217
|
+
def method_missing(method, *args, &block)
|
|
218
|
+
if ['groupby', 'count', 'expr'].include?("#{method}")
|
|
219
|
+
@match["attrs"]["@#{method}"]
|
|
220
|
+
elsif ['id', 'weight', 'attrs'].include?("#{method}")
|
|
221
|
+
@match["#{method}"]
|
|
222
|
+
else
|
|
223
|
+
@match["attrs"]["#{method}"]
|
|
224
|
+
end
|
|
225
|
+
end
|
|
226
|
+
end
|
|
227
|
+
|
|
228
|
+
class Result
|
|
229
|
+
attr_reader :matches
|
|
230
|
+
|
|
231
|
+
def initialize(sphinx_hash)
|
|
232
|
+
@matches = []
|
|
233
|
+
@sphinx_hash = sphinx_hash
|
|
234
|
+
if @sphinx_hash.has_key?("matches")
|
|
235
|
+
@sphinx_hash["matches"].each do |match|
|
|
236
|
+
@matches << Match.new(match)
|
|
237
|
+
end
|
|
238
|
+
end
|
|
239
|
+
end
|
|
240
|
+
|
|
241
|
+
def method_missing(method, *args, &block)
|
|
242
|
+
@sphinx_hash["#{method}"]
|
|
243
|
+
end
|
|
244
|
+
end
|
|
245
|
+
|
|
246
|
+
module Search
|
|
247
|
+
class << self
|
|
248
|
+
attr_accessor :target
|
|
249
|
+
|
|
250
|
+
def delegate(*methods)
|
|
251
|
+
methods.each do |method|
|
|
252
|
+
define_method(method) do |*args, &block|
|
|
253
|
+
Search.target.send(method, *args, &block)
|
|
254
|
+
end
|
|
255
|
+
private method
|
|
256
|
+
end
|
|
257
|
+
end
|
|
258
|
+
end
|
|
259
|
+
|
|
260
|
+
self.target = Zinx::Client
|
|
261
|
+
|
|
262
|
+
delegate :filter, :filter_range, :filter_float_range, :sort, :group, :group_distinct,
|
|
263
|
+
:select, :last_error, :last_warning, :server, :limits, :max_query_time, :match_mode,
|
|
264
|
+
:ranking_mode, :field_weight, :field_weights, :index_weight, :index_weights,
|
|
265
|
+
:id_range, :geo_anchor, :retries, :override, :reset, :reset_filter, :reset_groups,
|
|
266
|
+
:reset_overrides, :build_excerpts, :excerpts, :build_keywords, :update, :results,
|
|
267
|
+
:matches, :run, :add_query, :search, :init
|
|
268
|
+
end
|
|
269
|
+
|
|
270
|
+
end
|
|
271
|
+
|
|
272
272
|
extend Zinx::Search
|