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
|
module ThinkingSphinx
class Collection < ::Array
attr_reader :total_entries, :total_pages, :current_page, :per_page
attr_accessor :results
# Compatibility with older versions of will_paginate
alias_method :page_count, :total_pages
def initialize(page, per_page, entries, total_entries)
@current_page, @per_page, @total_entries = page, per_page, total_entries
@total_pages = (entries / @per_page.to_f).ceil
end
def self.ids_from_results(results, page, limit, options)
collection = self.new(page, limit,
results[:total] || 0, results[:total_found] || 0
)
collection.results = results
collection.replace results[:matches].collect { |match|
match[:attributes]["sphinx_internal_id"]
}
return collection
end
def self.create_from_results(results, page, limit, options)
collection = self.new(page, limit,
results[:total] || 0, results[:total_found] || 0
)
collection.results = results
collection.replace instances_from_matches(results[:matches], options)
return collection
end
def self.instances_from_matches(matches, options = {})
if klass = options[:class]
instances_from_class klass, matches, options
else
instances_from_classes matches, options
end
end
def self.instances_from_class(klass, matches, options = {})
index_options = klass.sphinx_index_options
ids = matches.collect { |match| match[:attributes]["sphinx_internal_id"] }
instances = ids.length > 0 ? klass.find(
:all,
:conditions => {klass.primary_key.to_sym => ids},
:include => (options[:include] || index_options[:include]),
:select => (options[:select] || index_options[:select]),
:order => (options[:sql_order] || index_options[:sql_order])
) : []
# Raise an exception if we find records in Sphinx but not in the DB, so
# the search method can retry without them. See
# ThinkingSphinx::Search.retry_search_on_stale_index.
if options[:raise_on_stale] && instances.length < ids.length
stale_ids = ids - instances.map {|i| i.id }
raise StaleIdsException, stale_ids
end
# if the user has specified an SQL order, return the collection
# without rearranging it into the Sphinx order
return instances if options[:sql_order]
ids.collect { |obj_id|
instances.detect { |obj| obj.id == obj_id }
}
end
# Group results by class and call #find(:all) once for each group to reduce
# the number of #find's in multi-model searches.
#
def self.instances_from_classes(matches, options = {})
groups = matches.group_by { |match| match[:attributes]["class_crc"] }
groups.each do |crc, group|
group.replace(
instances_from_class(class_from_crc(crc), group, options)
)
end
matches.collect do |match|
groups.detect { |crc, group|
crc == match[:attributes]["class_crc"]
}[1].detect { |obj|
obj.id == match[:attributes]["sphinx_internal_id"]
}
end
end
def self.class_from_crc(crc)
@@models_by_crc ||= ThinkingSphinx.indexed_models.inject({}) do |hash, model|
hash[model.constantize.to_crc32] = model
model.constantize.subclasses.each { |subclass|
hash[subclass.to_crc32] = subclass.name
}
hash
end
@@models_by_crc[crc].constantize
end
def previous_page
current_page > 1 ? (current_page - 1) : nil
end
def next_page
current_page < total_pages ? (current_page + 1): nil
end
def offset
(current_page - 1) * @per_page
end
def method_missing(method, *args, &block)
super unless method.to_s[/^each_with_.*/]
each_with_attribute method.to_s.gsub(/^each_with_/, ''), &block
end
def each_with_groupby_and_count(&block)
results[:matches].each_with_index do |match, index|
yield self[index], match[:attributes]["@groupby"], match[:attributes]["@count"]
end
end
def each_with_attribute(attribute, &block)
results[:matches].each_with_index do |match, index|
yield self[index], (match[:attributes][attribute] || match[:attributes]["@#{attribute}"])
end
end
def each_with_weighting(&block)
results[:matches].each_with_index do |match, index|
yield self[index], match[:weight]
end
end
def inject_with_groupby_and_count(initial = nil, &block)
index = -1
results[:matches].inject(initial) do |memo, match|
index += 1
yield memo, self[index], match[:attributes]["@groupby"], match[:attributes]["@count"]
end
end
end
end
|
