acts_as_solr_reloaded 1.0.0

Sign up to get free protection for your applications and to get access to all the features.
Files changed (197) hide show
  1. data/LICENSE +22 -0
  2. data/README.markdown +64 -0
  3. data/README.rdoc +93 -0
  4. data/Rakefile +71 -0
  5. data/TESTING_THE_PLUGIN +25 -0
  6. data/VERSION +1 -0
  7. data/config/solr.yml +14 -0
  8. data/config/solr_environment.rb +35 -0
  9. data/generators/dynamic_attributes_migration/dynamic_attributes_migration_generator.rb +7 -0
  10. data/generators/dynamic_attributes_migration/templates/migration.rb +15 -0
  11. data/generators/local_migration/local_migration_generator.rb +7 -0
  12. data/generators/local_migration/templates/migration.rb +16 -0
  13. data/lib/acts_as_solr.rb +65 -0
  14. data/lib/acts_as_solr/acts_methods.rb +363 -0
  15. data/lib/acts_as_solr/class_methods.rb +240 -0
  16. data/lib/acts_as_solr/common_methods.rb +89 -0
  17. data/lib/acts_as_solr/deprecation.rb +61 -0
  18. data/lib/acts_as_solr/dynamic_attribute.rb +3 -0
  19. data/lib/acts_as_solr/instance_methods.rb +194 -0
  20. data/lib/acts_as_solr/lazy_document.rb +18 -0
  21. data/lib/acts_as_solr/local.rb +4 -0
  22. data/lib/acts_as_solr/parser_methods.rb +248 -0
  23. data/lib/acts_as_solr/search_results.rb +74 -0
  24. data/lib/acts_as_solr/solr_fixtures.rb +13 -0
  25. data/lib/acts_as_solr/tasks.rb +10 -0
  26. data/lib/acts_as_solr/tasks/database.rake +16 -0
  27. data/lib/acts_as_solr/tasks/solr.rake +142 -0
  28. data/lib/acts_as_solr/tasks/test.rake +5 -0
  29. data/lib/solr.rb +26 -0
  30. data/lib/solr/connection.rb +177 -0
  31. data/lib/solr/document.rb +75 -0
  32. data/lib/solr/exception.rb +13 -0
  33. data/lib/solr/field.rb +36 -0
  34. data/lib/solr/importer.rb +19 -0
  35. data/lib/solr/importer/array_mapper.rb +26 -0
  36. data/lib/solr/importer/delimited_file_source.rb +38 -0
  37. data/lib/solr/importer/hpricot_mapper.rb +27 -0
  38. data/lib/solr/importer/mapper.rb +51 -0
  39. data/lib/solr/importer/solr_source.rb +41 -0
  40. data/lib/solr/importer/xpath_mapper.rb +35 -0
  41. data/lib/solr/indexer.rb +52 -0
  42. data/lib/solr/request.rb +26 -0
  43. data/lib/solr/request/add_document.rb +58 -0
  44. data/lib/solr/request/base.rb +36 -0
  45. data/lib/solr/request/commit.rb +29 -0
  46. data/lib/solr/request/delete.rb +48 -0
  47. data/lib/solr/request/dismax.rb +46 -0
  48. data/lib/solr/request/index_info.rb +22 -0
  49. data/lib/solr/request/modify_document.rb +46 -0
  50. data/lib/solr/request/optimize.rb +19 -0
  51. data/lib/solr/request/ping.rb +36 -0
  52. data/lib/solr/request/select.rb +54 -0
  53. data/lib/solr/request/spellcheck.rb +30 -0
  54. data/lib/solr/request/standard.rb +406 -0
  55. data/lib/solr/request/update.rb +23 -0
  56. data/lib/solr/response.rb +27 -0
  57. data/lib/solr/response/add_document.rb +17 -0
  58. data/lib/solr/response/base.rb +42 -0
  59. data/lib/solr/response/commit.rb +15 -0
  60. data/lib/solr/response/delete.rb +13 -0
  61. data/lib/solr/response/dismax.rb +8 -0
  62. data/lib/solr/response/index_info.rb +26 -0
  63. data/lib/solr/response/modify_document.rb +17 -0
  64. data/lib/solr/response/optimize.rb +14 -0
  65. data/lib/solr/response/ping.rb +26 -0
  66. data/lib/solr/response/ruby.rb +42 -0
  67. data/lib/solr/response/select.rb +17 -0
  68. data/lib/solr/response/spellcheck.rb +20 -0
  69. data/lib/solr/response/standard.rb +65 -0
  70. data/lib/solr/response/xml.rb +39 -0
  71. data/lib/solr/solrtasks.rb +27 -0
  72. data/lib/solr/util.rb +32 -0
  73. data/lib/solr/xml.rb +44 -0
  74. data/solr/README.txt +36 -0
  75. data/solr/etc/jetty.xml +212 -0
  76. data/solr/etc/webdefault.xml +379 -0
  77. data/solr/exampledocs/books.csv +11 -0
  78. data/solr/exampledocs/hd.xml +46 -0
  79. data/solr/exampledocs/ipod_other.xml +50 -0
  80. data/solr/exampledocs/ipod_video.xml +35 -0
  81. data/solr/exampledocs/locales.xml +16 -0
  82. data/solr/exampledocs/mem.xml +58 -0
  83. data/solr/exampledocs/monitor.xml +31 -0
  84. data/solr/exampledocs/monitor2.xml +30 -0
  85. data/solr/exampledocs/mp500.xml +39 -0
  86. data/solr/exampledocs/post.jar +0 -0
  87. data/solr/exampledocs/post.sh +28 -0
  88. data/solr/exampledocs/sd500.xml +33 -0
  89. data/solr/exampledocs/solr.xml +38 -0
  90. data/solr/exampledocs/spellchecker.xml +58 -0
  91. data/solr/exampledocs/test_utf8.sh +83 -0
  92. data/solr/exampledocs/utf8-example.xml +42 -0
  93. data/solr/exampledocs/vidcard.xml +52 -0
  94. data/solr/lib/jetty-6.1.3.jar +0 -0
  95. data/solr/lib/jetty-util-6.1.3.jar +0 -0
  96. data/solr/lib/jsp-2.1/ant-1.6.5.jar +0 -0
  97. data/solr/lib/jsp-2.1/core-3.1.1.jar +0 -0
  98. data/solr/lib/jsp-2.1/jsp-2.1.jar +0 -0
  99. data/solr/lib/jsp-2.1/jsp-api-2.1.jar +0 -0
  100. data/solr/lib/servlet-api-2.5-6.1.3.jar +0 -0
  101. data/solr/multicore/README.txt +3 -0
  102. data/solr/multicore/core0/conf/schema.xml +41 -0
  103. data/solr/multicore/core0/conf/solrconfig.xml +40 -0
  104. data/solr/multicore/core1/conf/schema.xml +41 -0
  105. data/solr/multicore/core1/conf/solrconfig.xml +40 -0
  106. data/solr/multicore/exampledocs/ipod_other.xml +34 -0
  107. data/solr/multicore/exampledocs/ipod_video.xml +22 -0
  108. data/solr/multicore/solr.xml +35 -0
  109. data/solr/solr/README.txt +52 -0
  110. data/solr/solr/bin/abc +190 -0
  111. data/solr/solr/bin/abo +190 -0
  112. data/solr/solr/bin/backup +117 -0
  113. data/solr/solr/bin/backupcleaner +142 -0
  114. data/solr/solr/bin/commit +133 -0
  115. data/solr/solr/bin/optimize +134 -0
  116. data/solr/solr/bin/readercycle +129 -0
  117. data/solr/solr/bin/rsyncd-disable +77 -0
  118. data/solr/solr/bin/rsyncd-enable +76 -0
  119. data/solr/solr/bin/rsyncd-start +145 -0
  120. data/solr/solr/bin/rsyncd-stop +105 -0
  121. data/solr/solr/bin/scripts-util +99 -0
  122. data/solr/solr/bin/snapcleaner +154 -0
  123. data/solr/solr/bin/snapinstaller +198 -0
  124. data/solr/solr/bin/snappuller +269 -0
  125. data/solr/solr/bin/snappuller-disable +77 -0
  126. data/solr/solr/bin/snappuller-enable +77 -0
  127. data/solr/solr/bin/snapshooter +136 -0
  128. data/solr/solr/conf/admin-extra.html +31 -0
  129. data/solr/solr/conf/elevate.xml +36 -0
  130. data/solr/solr/conf/mapping-ISOLatin1Accent.txt +246 -0
  131. data/solr/solr/conf/protwords.txt +21 -0
  132. data/solr/solr/conf/schema.xml +132 -0
  133. data/solr/solr/conf/scripts.conf +24 -0
  134. data/solr/solr/conf/solrconfig.xml +906 -0
  135. data/solr/solr/conf/spellings.txt +2 -0
  136. data/solr/solr/conf/stopwords.txt +58 -0
  137. data/solr/solr/conf/synonyms.txt +31 -0
  138. data/solr/solr/conf/xslt/example.xsl +132 -0
  139. data/solr/solr/conf/xslt/example_atom.xsl +67 -0
  140. data/solr/solr/conf/xslt/example_rss.xsl +66 -0
  141. data/solr/solr/conf/xslt/luke.xsl +337 -0
  142. data/solr/solr/lib/localsolr.jar +0 -0
  143. data/solr/solr/lib/lucene-spatial-2.9-dev.jar +0 -0
  144. data/solr/start.jar +0 -0
  145. data/solr/webapps/solr.war +0 -0
  146. data/test/config/solr.yml +2 -0
  147. data/test/db/connections/mysql/connection.rb +11 -0
  148. data/test/db/connections/sqlite/connection.rb +8 -0
  149. data/test/db/migrate/001_create_books.rb +15 -0
  150. data/test/db/migrate/002_create_movies.rb +12 -0
  151. data/test/db/migrate/003_create_categories.rb +11 -0
  152. data/test/db/migrate/004_create_electronics.rb +16 -0
  153. data/test/db/migrate/005_create_authors.rb +12 -0
  154. data/test/db/migrate/006_create_postings.rb +9 -0
  155. data/test/db/migrate/007_create_posts.rb +13 -0
  156. data/test/db/migrate/008_create_gadgets.rb +11 -0
  157. data/test/db/migrate/009_create_dynamic_attributes.rb +15 -0
  158. data/test/db/migrate/010_create_advertises.rb +13 -0
  159. data/test/db/migrate/011_create_locals.rb +15 -0
  160. data/test/db/test.db +0 -0
  161. data/test/fixtures/advertises.yml +12 -0
  162. data/test/fixtures/authors.yml +9 -0
  163. data/test/fixtures/books.yml +13 -0
  164. data/test/fixtures/categories.yml +7 -0
  165. data/test/fixtures/db_definitions/mysql.sql +41 -0
  166. data/test/fixtures/dynamic_attributes.yml +11 -0
  167. data/test/fixtures/electronics.yml +49 -0
  168. data/test/fixtures/locals.yml +9 -0
  169. data/test/fixtures/movies.yml +9 -0
  170. data/test/fixtures/postings.yml +10 -0
  171. data/test/functional/acts_as_solr_test.rb +463 -0
  172. data/test/functional/association_indexing_test.rb +37 -0
  173. data/test/functional/faceted_search_test.rb +163 -0
  174. data/test/functional/multi_solr_search_test.rb +57 -0
  175. data/test/models/advertise.rb +6 -0
  176. data/test/models/author.rb +10 -0
  177. data/test/models/book.rb +10 -0
  178. data/test/models/category.rb +8 -0
  179. data/test/models/dynamic_attribute.rb +7 -0
  180. data/test/models/electronic.rb +25 -0
  181. data/test/models/gadget.rb +9 -0
  182. data/test/models/local.rb +7 -0
  183. data/test/models/movie.rb +17 -0
  184. data/test/models/novel.rb +2 -0
  185. data/test/models/post.rb +3 -0
  186. data/test/models/posting.rb +11 -0
  187. data/test/test_helper.rb +56 -0
  188. data/test/unit/acts_methods_shoulda.rb +95 -0
  189. data/test/unit/class_methods_shoulda.rb +85 -0
  190. data/test/unit/common_methods_shoulda.rb +111 -0
  191. data/test/unit/instance_methods_shoulda.rb +372 -0
  192. data/test/unit/lazy_document_shoulda.rb +34 -0
  193. data/test/unit/parser_instance.rb +19 -0
  194. data/test/unit/parser_methods_shoulda.rb +338 -0
  195. data/test/unit/solr_instance.rb +74 -0
  196. data/test/unit/test_helper.rb +24 -0
  197. metadata +290 -0
@@ -0,0 +1,58 @@
1
+ # The ASF licenses this file to You under the Apache License, Version 2.0
2
+ # (the "License"); you may not use this file except in compliance with
3
+ # the License. You may obtain a copy of the License at
4
+ #
5
+ # http://www.apache.org/licenses/LICENSE-2.0
6
+ #
7
+ # Unless required by applicable law or agreed to in writing, software
8
+ # distributed under the License is distributed on an "AS IS" BASIS,
9
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
10
+ # See the License for the specific language governing permissions and
11
+ # limitations under the License.
12
+
13
+ class Solr::Request::AddDocument < Solr::Request::Update
14
+
15
+ # create the request, optionally passing in a Solr::Document
16
+ #
17
+ # request = Solr::Request::AddDocument.new doc
18
+ #
19
+ # as a short cut you can pass in a Hash instead:
20
+ #
21
+ # request = Solr::Request::AddDocument.new :creator => 'Jorge Luis Borges'
22
+ #
23
+ # or an array, to add multiple documents at the same time:
24
+ #
25
+ # request = Solr::Request::AddDocument.new([doc1, doc2, doc3])
26
+
27
+ def initialize(doc={})
28
+ @docs = []
29
+ if doc.is_a?(Array)
30
+ doc.each { |d| add_doc(d) }
31
+ else
32
+ add_doc(doc)
33
+ end
34
+ end
35
+
36
+ # returns the request as a string suitable for posting
37
+
38
+ def to_s
39
+ e = Solr::XML::Element.new 'add'
40
+ for doc in @docs
41
+ e.add_element doc.to_xml
42
+ end
43
+ return e.to_s
44
+ end
45
+
46
+ private
47
+ def add_doc(doc)
48
+ case doc
49
+ when Hash
50
+ @docs << Solr::Document.new(doc)
51
+ when Solr::Document
52
+ @docs << doc
53
+ else
54
+ raise "must pass in Solr::Document or Hash"
55
+ end
56
+ end
57
+
58
+ end
@@ -0,0 +1,36 @@
1
+ # The ASF licenses this file to You under the Apache License, Version 2.0
2
+ # (the "License"); you may not use this file except in compliance with
3
+ # the License. You may obtain a copy of the License at
4
+ #
5
+ # http://www.apache.org/licenses/LICENSE-2.0
6
+ #
7
+ # Unless required by applicable law or agreed to in writing, software
8
+ # distributed under the License is distributed on an "AS IS" BASIS,
9
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
10
+ # See the License for the specific language governing permissions and
11
+ # limitations under the License.
12
+
13
+ class Solr::Request::Base
14
+
15
+
16
+ #TODO : Add base support for the debugQuery flag, and such that the response provides debug output easily
17
+
18
+ # returns either :xml or :ruby depending on what the
19
+ # response type is for a given request
20
+
21
+ def response_format
22
+ raise "unknown request type: #{self.class}"
23
+ end
24
+
25
+ def content_type
26
+ 'text/xml; charset=utf-8'
27
+ end
28
+
29
+ # returns the solr handler or url fragment that can
30
+ # respond to this type of request
31
+
32
+ def handler
33
+ raise "unknown request type: #{self.class}"
34
+ end
35
+
36
+ end
@@ -0,0 +1,29 @@
1
+ # The ASF licenses this file to You under the Apache License, Version 2.0
2
+ # (the "License"); you may not use this file except in compliance with
3
+ # the License. You may obtain a copy of the License at
4
+ #
5
+ # http://www.apache.org/licenses/LICENSE-2.0
6
+ #
7
+ # Unless required by applicable law or agreed to in writing, software
8
+ # distributed under the License is distributed on an "AS IS" BASIS,
9
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
10
+ # See the License for the specific language governing permissions and
11
+ # limitations under the License.
12
+
13
+ class Solr::Request::Commit < Solr::Request::Update
14
+
15
+ def initialize(options={})
16
+ @wait_searcher = options[:wait_searcher] || true
17
+ @wait_flush = options[:wait_flush] || true
18
+ end
19
+
20
+
21
+ def to_s
22
+ e = Solr::XML::Element.new('commit')
23
+ e.attributes['waitSearcher'] = @wait_searcher ? 'true' : 'false'
24
+ e.attributes['waitFlush'] = @wait_flush ? 'true' : 'false'
25
+
26
+ e.to_s
27
+ end
28
+
29
+ end
@@ -0,0 +1,48 @@
1
+ # The ASF licenses this file to You under the Apache License, Version 2.0
2
+ # (the "License"); you may not use this file except in compliance with
3
+ # the License. You may obtain a copy of the License at
4
+ #
5
+ # http://www.apache.org/licenses/LICENSE-2.0
6
+ #
7
+ # Unless required by applicable law or agreed to in writing, software
8
+ # distributed under the License is distributed on an "AS IS" BASIS,
9
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
10
+ # See the License for the specific language governing permissions and
11
+ # limitations under the License.
12
+
13
+ class Solr::Request::Delete < Solr::Request::Update
14
+
15
+ # A delete request can be for a specific document id
16
+ #
17
+ # request = Solr::Request::Delete.new(:id => 1234)
18
+ #
19
+ # or by query:
20
+ #
21
+ # request = Solr::Request::Delete.new(:query =>
22
+ #
23
+ def initialize(options)
24
+ unless options.kind_of?(Hash) and (options[:id] or options[:query])
25
+ raise Solr::Exception.new("must pass in :id or :query")
26
+ end
27
+ if options[:id] and options[:query]
28
+ raise Solr::Exception.new("can't pass in both :id and :query")
29
+ end
30
+ @document_id = options[:id]
31
+ @query = options[:query]
32
+ end
33
+
34
+ def to_s
35
+ delete_element = Solr::XML::Element.new('delete')
36
+ if @document_id
37
+ id_element = Solr::XML::Element.new('id')
38
+ id_element.text = @document_id
39
+ delete_element.add_element(id_element)
40
+ elsif @query
41
+ query = Solr::XML::Element.new('query')
42
+ query.text = @query
43
+ delete_element.add_element(query)
44
+ end
45
+ delete_element.to_s
46
+ end
47
+ end
48
+
@@ -0,0 +1,46 @@
1
+ # The ASF licenses this file to You under the Apache License, Version 2.0
2
+ # (the "License"); you may not use this file except in compliance with
3
+ # the License. You may obtain a copy of the License at
4
+ #
5
+ # http://www.apache.org/licenses/LICENSE-2.0
6
+ #
7
+ # Unless required by applicable law or agreed to in writing, software
8
+ # distributed under the License is distributed on an "AS IS" BASIS,
9
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
10
+ # See the License for the specific language governing permissions and
11
+ # limitations under the License.
12
+
13
+ class Solr::Request::Dismax < Solr::Request::Standard
14
+
15
+ VALID_PARAMS.replace(VALID_PARAMS + [:tie_breaker, :query_fields, :minimum_match, :phrase_fields, :phrase_slop,
16
+ :boost_query, :boost_functions])
17
+
18
+ def initialize(params)
19
+ @alternate_query = params.delete(:alternate_query)
20
+ @sort_values = params.delete(:sort)
21
+
22
+ super
23
+
24
+ @query_type = "dismax"
25
+ end
26
+
27
+ def to_hash
28
+ hash = super
29
+ hash[:tie] = @params[:tie_breaker]
30
+ hash[:mm] = @params[:minimum_match]
31
+ hash[:qf] = @params[:query_fields]
32
+ hash[:pf] = @params[:phrase_fields]
33
+ hash[:ps] = @params[:phrase_slop]
34
+ hash[:bq] = @params[:boost_query]
35
+ hash[:bf] = @params[:boost_functions]
36
+ hash["q.alt"] = @alternate_query
37
+ # FIXME: 2007-02-13 <coda.hale@gmail.com> -- This code is duplicated in
38
+ # Solr::Request::Standard. It should be refactored into a single location.
39
+ hash[:sort] = @sort_values.collect do |sort|
40
+ key = sort.keys[0]
41
+ "#{key.to_s} #{sort[key] == :descending ? 'desc' : 'asc'}"
42
+ end.join(',') if @sort_values
43
+ return hash
44
+ end
45
+
46
+ end
@@ -0,0 +1,22 @@
1
+ # The ASF licenses this file to You under the Apache License, Version 2.0
2
+ # (the "License"); you may not use this file except in compliance with
3
+ # the License. You may obtain a copy of the License at
4
+ #
5
+ # http://www.apache.org/licenses/LICENSE-2.0
6
+ #
7
+ # Unless required by applicable law or agreed to in writing, software
8
+ # distributed under the License is distributed on an "AS IS" BASIS,
9
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
10
+ # See the License for the specific language governing permissions and
11
+ # limitations under the License.
12
+
13
+ class Solr::Request::IndexInfo < Solr::Request::Select
14
+
15
+ def handler
16
+ 'admin/luke'
17
+ end
18
+
19
+ def to_hash
20
+ {:numTerms => 0}.merge(super.to_hash)
21
+ end
22
+ end
@@ -0,0 +1,46 @@
1
+ # The ASF licenses this file to You under the Apache License, Version 2.0
2
+ # (the "License"); you may not use this file except in compliance with
3
+ # the License. You may obtain a copy of the License at
4
+ #
5
+ # http://www.apache.org/licenses/LICENSE-2.0
6
+ #
7
+ # Unless required by applicable law or agreed to in writing, software
8
+ # distributed under the License is distributed on an "AS IS" BASIS,
9
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
10
+ # See the License for the specific language governing permissions and
11
+ # limitations under the License.
12
+
13
+ class Solr::Request::ModifyDocument < Solr::Request::Update
14
+
15
+ # Example: ModifyDocument.new(:id => 10, :overwrite => {:field_name => "new value"})
16
+ def initialize(update_data)
17
+ modes = []
18
+ @doc = {}
19
+ [:overwrite, :append, :distinct, :increment, :delete].each do |mode|
20
+ field_data = update_data[mode]
21
+ if field_data
22
+ field_data.each do |field_name, field_value|
23
+ modes << "#{field_name}:#{mode.to_s.upcase}"
24
+ @doc[field_name] = field_value if field_value # if value is nil, omit so it can be removed
25
+ end
26
+ update_data.delete mode
27
+ end
28
+ end
29
+ @mode = modes.join(",")
30
+
31
+ # only one key should be left over, the id
32
+ @doc[update_data.keys[0].to_s] = update_data.values[0]
33
+ end
34
+
35
+ # returns the request as a string suitable for posting
36
+ def to_s
37
+ e = Solr::XML::Element.new 'add'
38
+ e.add_element(Solr::Document.new(@doc).to_xml)
39
+ return e.to_s
40
+ end
41
+
42
+ def handler
43
+ "update?mode=#{@mode}"
44
+ end
45
+
46
+ end
@@ -0,0 +1,19 @@
1
+ # The ASF licenses this file to You under the Apache License, Version 2.0
2
+ # (the "License"); you may not use this file except in compliance with
3
+ # the License. You may obtain a copy of the License at
4
+ #
5
+ # http://www.apache.org/licenses/LICENSE-2.0
6
+ #
7
+ # Unless required by applicable law or agreed to in writing, software
8
+ # distributed under the License is distributed on an "AS IS" BASIS,
9
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
10
+ # See the License for the specific language governing permissions and
11
+ # limitations under the License.
12
+
13
+ class Solr::Request::Optimize < Solr::Request::Update
14
+
15
+ def to_s
16
+ Solr::XML::Element.new('optimize').to_s
17
+ end
18
+
19
+ end
@@ -0,0 +1,36 @@
1
+ # The ASF licenses this file to You under the Apache License, Version 2.0
2
+ # (the "License"); you may not use this file except in compliance with
3
+ # the License. You may obtain a copy of the License at
4
+ #
5
+ # http://www.apache.org/licenses/LICENSE-2.0
6
+ #
7
+ # Unless required by applicable law or agreed to in writing, software
8
+ # distributed under the License is distributed on an "AS IS" BASIS,
9
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
10
+ # See the License for the specific language governing permissions and
11
+ # limitations under the License.
12
+
13
+ # TODO: Consider something lazy like this?
14
+ # Solr::Request::Ping = Solr::Request.simple_request :format=>:xml, :handler=>'admin/ping'
15
+ # class Solr::Request
16
+ # def self.simple_request(options)
17
+ # Class.new do
18
+ # def response_format
19
+ # options[:format]
20
+ # end
21
+ # def handler
22
+ # options[:handler]
23
+ # end
24
+ # end
25
+ # end
26
+ # end
27
+
28
+ class Solr::Request::Ping < Solr::Request::Base
29
+ def response_format
30
+ :xml
31
+ end
32
+
33
+ def handler
34
+ 'admin/ping'
35
+ end
36
+ end
@@ -0,0 +1,54 @@
1
+ # The ASF licenses this file to You under the Apache License, Version 2.0
2
+ # (the "License"); you may not use this file except in compliance with
3
+ # the License. You may obtain a copy of the License at
4
+ #
5
+ # http://www.apache.org/licenses/LICENSE-2.0
6
+ #
7
+ # Unless required by applicable law or agreed to in writing, software
8
+ # distributed under the License is distributed on an "AS IS" BASIS,
9
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
10
+ # See the License for the specific language governing permissions and
11
+ # limitations under the License.
12
+
13
+ # "Abstract" base class, only useful with subclasses that add parameters
14
+ class Solr::Request::Select < Solr::Request::Base
15
+
16
+ attr_reader :query_type
17
+
18
+ def initialize(qt=nil, params={})
19
+ @query_type = qt
20
+ @select_params = params
21
+ end
22
+
23
+ def response_format
24
+ :ruby
25
+ end
26
+
27
+ def handler
28
+ 'select'
29
+ end
30
+
31
+ def content_type
32
+ 'application/x-www-form-urlencoded; charset=utf-8'
33
+ end
34
+
35
+ def to_hash
36
+ return {:qt => query_type, :wt => 'ruby'}.merge(@select_params)
37
+ end
38
+
39
+ def to_s
40
+ raw_params = self.to_hash
41
+
42
+ http_params = []
43
+ raw_params.each do |key,value|
44
+ if value.respond_to? :each
45
+ value.each { |v| http_params << "#{key}=#{ERB::Util::url_encode(v)}" unless v.nil?}
46
+ else
47
+ http_params << "#{key}=#{ERB::Util::url_encode(value)}" unless value.nil?
48
+ end
49
+ end
50
+
51
+ http_params.join("&")
52
+ end
53
+
54
+ end
@@ -0,0 +1,30 @@
1
+ # The ASF licenses this file to You under the Apache License, Version 2.0
2
+ # (the "License"); you may not use this file except in compliance with
3
+ # the License. You may obtain a copy of the License at
4
+ #
5
+ # http://www.apache.org/licenses/LICENSE-2.0
6
+ #
7
+ # Unless required by applicable law or agreed to in writing, software
8
+ # distributed under the License is distributed on an "AS IS" BASIS,
9
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
10
+ # See the License for the specific language governing permissions and
11
+ # limitations under the License.
12
+
13
+ class Solr::Request::Spellcheck < Solr::Request::Select
14
+
15
+ def initialize(params)
16
+ super('spellchecker')
17
+ @params = params
18
+ end
19
+
20
+ def to_hash
21
+ hash = super
22
+ hash[:q] = @params[:query]
23
+ hash[:suggestionCount] = @params[:suggestion_count]
24
+ hash[:accuracy] = @params[:accuracy]
25
+ hash[:onlyMorePopular] = @params[:only_more_popular]
26
+ hash[:cmd] = @params[:command]
27
+ return hash
28
+ end
29
+
30
+ end
@@ -0,0 +1,406 @@
1
+ # The ASF licenses this file to You under the Apache License, Version 2.0
2
+ # (the "License"); you may not use this file except in compliance with
3
+ # the License. You may obtain a copy of the License at
4
+ #
5
+ # http://www.apache.org/licenses/LICENSE-2.0
6
+ #
7
+ # Unless required by applicable law or agreed to in writing, software
8
+ # distributed under the License is distributed on an "AS IS" BASIS,
9
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
10
+ # See the License for the specific language governing permissions and
11
+ # limitations under the License.
12
+
13
+ class Solr::Request::Standard < Solr::Request::Select
14
+
15
+ VALID_PARAMS = [:query, :sort, :default_field, :operator, :start, :rows, :shards, :date_facets,
16
+ :filter_queries, :field_list, :debug_query, :explain_other, :facets, :highlighting, :mlt, :radius,
17
+ :latitude, :longitude]
18
+
19
+ def initialize(params)
20
+ super(params[:radius].nil? ? 'standard' : 'geo')
21
+
22
+ raise "Invalid parameters: #{(params.keys - VALID_PARAMS).join(',')}" unless
23
+ (params.keys - VALID_PARAMS).empty?
24
+
25
+ raise ":query parameter required" unless params[:query]
26
+
27
+ @params = params.dup
28
+
29
+ # Validate operator
30
+ if params[:operator]
31
+ raise "Only :and/:or operators allowed" unless
32
+ [:and, :or].include?(params[:operator])
33
+
34
+ @params[:operator] = params[:operator].to_s.upcase
35
+ end
36
+
37
+ # Validate start, rows can be transformed to ints
38
+ @params[:start] = params[:start].to_i if params[:start]
39
+ @params[:rows] = params[:rows].to_i if params[:rows]
40
+
41
+ @params[:field_list] ||= ["*","score"]
42
+
43
+ @params[:shards] ||= []
44
+ end
45
+
46
+ def to_hash
47
+ hash = {}
48
+
49
+ # standard request param processing
50
+ hash[:sort] = @params[:sort]
51
+ hash[:q] = @params[:query]
52
+ hash["q.op"] = @params[:operator]
53
+ hash[:df] = @params[:default_field]
54
+
55
+ # common parameter processing
56
+ hash[:start] = @params[:start]
57
+ hash[:rows] = @params[:rows]
58
+ hash[:fq] = @params[:filter_queries]
59
+ hash[:fl] = @params[:field_list].join(',')
60
+ hash[:debugQuery] = @params[:debug_query]
61
+ hash[:explainOther] = @params[:explain_other]
62
+ hash[:shards] = @params[:shards].join(',') unless @params[:shards].empty?
63
+
64
+ #if @params[:qt].eql? 'geo'
65
+ hash[:radius] = @params[:radius]
66
+ hash[:lat] = @params[:latitude]
67
+ hash[:long] = @params[:longitude]
68
+ #end
69
+
70
+ # facet parameter processing
71
+ if @params[:facets]
72
+ # TODO need validation of all that is under the :facets Hash too
73
+ hash[:facet] = true
74
+ hash["facet.field"] = []
75
+ hash["facet.query"] = @params[:facets][:queries]
76
+ hash["facet.sort"] = (@params[:facets][:sort] == :count) if @params[:facets][:sort]
77
+ hash["facet.limit"] = @params[:facets][:limit]
78
+ hash["facet.missing"] = @params[:facets][:missing]
79
+ hash["facet.mincount"] = @params[:facets][:mincount]
80
+ hash["facet.prefix"] = @params[:facets][:prefix]
81
+ hash["facet.offset"] = @params[:facets][:offset]
82
+ if @params[:facets][:fields] # facet fields are optional (could be facet.query only)
83
+ @params[:facets][:fields].each do |f|
84
+ if f.kind_of? Hash
85
+ key = f.keys[0]
86
+ value = f[key]
87
+ hash["facet.field"] << key
88
+ hash["f.#{key}.facet.sort"] = (value[:sort] == :count) if value[:sort]
89
+ hash["f.#{key}.facet.limit"] = value[:limit]
90
+ hash["f.#{key}.facet.missing"] = value[:missing]
91
+ hash["f.#{key}.facet.mincount"] = value[:mincount]
92
+ hash["f.#{key}.facet.prefix"] = value[:prefix]
93
+ hash["f.#{key}.facet.offset"] = value[:offset]
94
+ else
95
+ hash["facet.field"] << f
96
+ end
97
+ end
98
+ end
99
+
100
+ if @params[:date_facets]
101
+ hash["facet.date"] = []
102
+ if @params[:date_facets][:fields]
103
+ @params[:date_facets][:fields].each do |f|
104
+ if f.kind_of? Hash
105
+ key = f.keys[0]
106
+ hash["facet.date"] << key
107
+ f[key].each { |k, v|
108
+ hash["f.#{key}.facet.date.#{k}"] = v
109
+ }
110
+ else
111
+ hash["facet.date"] << f
112
+ end
113
+ end
114
+ end
115
+ hash["facet.date.start"] = @params[:date_facets][:start]
116
+ hash["facet.date.end"] = @params[:date_facets][:end]
117
+ hash["facet.date.gap"] = @params[:date_facets][:gap]
118
+ hash["facet.date.other"] = @params[:date_facets][:other]
119
+ hash["facet.date.hardend"] = @params[:date_facets][:hardend]
120
+ if @params[:date_facets][:filter]
121
+ if hash[:fq]
122
+ hash[:fq] << @params[:date_facets][:filter]
123
+ else
124
+ hash[:fq] = @params[:date_facets][:filter]
125
+ end
126
+ end
127
+ end
128
+ end
129
+
130
+ # highlighting parameter processing - http://wiki.apache.org/solr/HighlightingParameters
131
+ if @params[:highlighting]
132
+ hash[:hl] = true
133
+ hash["hl.fl"] = @params[:highlighting][:field_list].join(',') if @params[:highlighting][:field_list]
134
+
135
+ snippets = @params[:highlighting][:max_snippets]
136
+ if snippets
137
+ if snippets.kind_of? Hash
138
+ if snippets[:default]
139
+ hash["hl.snippets"] = snippets[:default]
140
+ end
141
+ if snippets[:fields]
142
+ snippets[:fields].each do |k,v|
143
+ hash["f.#{k}.hl.snippets"] = v
144
+ end
145
+ end
146
+ else
147
+ hash["hl.snippets"] = snippets
148
+ end
149
+ end
150
+
151
+ fragsize = @params[:highlighting][:fragment_size]
152
+ if fragsize
153
+ if fragsize.kind_of? Hash
154
+ if fragsize[:default]
155
+ hash["hl.fragsize"] = fragsize[:default]
156
+ end
157
+ if fragsize[:fields]
158
+ fragsize[:fields].each do |k,v|
159
+ hash["f.#{k}.hl.fragsize"] = v
160
+ end
161
+ end
162
+ else
163
+ hash["hl.fragsize"] = fragsize
164
+ end
165
+ end
166
+
167
+ rfm = @params[:highlighting][:require_field_match]
168
+ if nil != rfm
169
+ if rfm.kind_of? Hash
170
+ if nil != rfm[:default]
171
+ hash["hl.requireFieldMatch"] = rfm[:default]
172
+ end
173
+ if rfm[:fields]
174
+ rfm[:fields].each do |k,v|
175
+ hash["f.#{k}.hl.requireFieldMatch"] = v
176
+ end
177
+ end
178
+ else
179
+ hash["hl.requireFieldMatch"] = rfm
180
+ end
181
+ end
182
+
183
+ mac = @params[:highlighting][:max_analyzed_chars]
184
+ if mac
185
+ if mac.kind_of? Hash
186
+ if mac[:default]
187
+ hash["hl.maxAnalyzedChars"] = mac[:default]
188
+ end
189
+ if mac[:fields]
190
+ mac[:fields].each do |k,v|
191
+ hash["f.#{k}.hl.maxAnalyzedChars"] = v
192
+ end
193
+ end
194
+ else
195
+ hash["hl.maxAnalyzedChars"] = mac
196
+ end
197
+ end
198
+
199
+ prefix = @params[:highlighting][:prefix]
200
+ if prefix
201
+ if prefix.kind_of? Hash
202
+ if prefix[:default]
203
+ hash["hl.simple.pre"] = prefix[:default]
204
+ end
205
+ if prefix[:fields]
206
+ prefix[:fields].each do |k,v|
207
+ hash["f.#{k}.hl.simple.pre"] = v
208
+ end
209
+ end
210
+ else
211
+ hash["hl.simple.pre"] = prefix
212
+ end
213
+ end
214
+
215
+ suffix = @params[:highlighting][:suffix]
216
+ if suffix
217
+ if suffix.kind_of? Hash
218
+ if suffix[:default]
219
+ hash["hl.simple.post"] = suffix[:default]
220
+ end
221
+ if suffix[:fields]
222
+ suffix[:fields].each do |k,v|
223
+ hash["f.#{k}.hl.simple.post"] = v
224
+ end
225
+ end
226
+ else
227
+ hash["hl.simple.post"] = suffix
228
+ end
229
+ end
230
+
231
+ formatter = @params[:highlighting][:formatter]
232
+ if formatter
233
+ if formatter.kind_of? Hash
234
+ if formatter[:default]
235
+ hash["hl.formatter"] = formatter[:default]
236
+ end
237
+ if formatter[:fields]
238
+ formatter[:fields].each do |k,v|
239
+ hash["f.#{k}.hl.formatter"] = v
240
+ end
241
+ end
242
+ else
243
+ hash["hl.formatter"] = formatter
244
+ end
245
+ end
246
+
247
+ fragmenter = @params[:highlighting][:fragmenter]
248
+ if fragmenter
249
+ if fragmenter.kind_of? Hash
250
+ if fragmenter[:default]
251
+ hash["hl.fragmenter"] = fragmenter[:default]
252
+ end
253
+ if fragmenter[:fields]
254
+ fragmenter[:fields].each do |k,v|
255
+ hash["f.#{k}.hl.fragmenter"] = v
256
+ end
257
+ end
258
+ else
259
+ hash["hl.fragmenter"] = fragmenter
260
+ end
261
+ end
262
+
263
+ merge_contiguous = @params[:highlighting][:merge_contiguous]
264
+ if nil != merge_contiguous
265
+ if merge_contiguous.kind_of? Hash
266
+ if nil != merge_contiguous[:default]
267
+ hash["hl.mergeContiguous"] = merge_contiguous[:default]
268
+ end
269
+ if merge_contiguous[:fields]
270
+ merge_contiguous[:fields].each do |k,v|
271
+ hash["f.#{k}.hl.mergeContiguous"] = v
272
+ end
273
+ end
274
+ else
275
+ hash["hl.mergeContiguous"] = merge_contiguous
276
+ end
277
+ end
278
+
279
+ increment = @params[:highlighting][:increment]
280
+ if increment
281
+ if increment.kind_of? Hash
282
+ if increment[:default]
283
+ hash["hl.increment"] = increment[:default]
284
+ end
285
+ if increment[:fields]
286
+ increment[:fields].each do |k,v|
287
+ hash["f.#{k}.hl.increment"] = v
288
+ end
289
+ end
290
+ else
291
+ hash["hl.increment"] = increment
292
+ end
293
+ end
294
+
295
+ # support "old style"
296
+ alternate_fields = @params[:highlighting][:alternate_fields]
297
+ if alternate_fields
298
+ alternate_fields.each do |f,v|
299
+ hash["f.#{f}.hl.alternateField"] = v
300
+ end
301
+ end
302
+
303
+ alternate_field = @params[:highlighting][:alternate_field]
304
+ if alternate_field
305
+ if alternate_field.kind_of? Hash
306
+ if alternate_field[:default]
307
+ hash["hl.alternateField"] = alternate_field[:default]
308
+ end
309
+ if alternate_field[:fields]
310
+ alternate_field[:fields].each do |k,v|
311
+ hash["f.#{k}.hl.alternateField"] = v
312
+ end
313
+ end
314
+ else
315
+ hash["hl.alternateField"] = alternate_field
316
+ end
317
+ end
318
+
319
+ mafl = @params[:highlighting][:max_alternate_field_length]
320
+ if mafl
321
+ if mafl.kind_of? Hash
322
+ if mafl[:default]
323
+ hash["hl.maxAlternateFieldLength"] = mafl[:default]
324
+ end
325
+ if mafl[:fields]
326
+ mafl[:fields].each do |k,v|
327
+ hash["f.#{k}.hl.maxAlternateFieldLength"] = v
328
+ end
329
+ else
330
+ # support "old style"
331
+ mafl.each do |k,v|
332
+ hash["f.#{k}.hl.maxAlternateFieldLength"] = v
333
+ end
334
+ end
335
+ else
336
+ hash["hl.maxAlternateFieldLength"] = mafl
337
+ end
338
+ end
339
+
340
+ hash["hl.usePhraseHighlighter"] = @params[:highlighting][:use_phrase_highlighter]
341
+
342
+ regex = @params[:highlighting][:regex]
343
+ if regex
344
+ if regex[:slop]
345
+ if regex[:slop].kind_of? Hash
346
+ if regex[:slop][:default]
347
+ hash["hl.regex.slop"] = regex[:slop][:default]
348
+ end
349
+ if regex[:slop][:fields]
350
+ regex[:slop][:fields].each do |k,v|
351
+ hash["f.#{k}.hl.regex.slop"] = v
352
+ end
353
+ end
354
+ else
355
+ hash["hl.regex.slop"] = regex[:slop]
356
+ end
357
+ end
358
+ if regex[:pattern]
359
+ if regex[:pattern].kind_of? Hash
360
+ if regex[:pattern][:default]
361
+ hash["hl.regex.pattern"] = regex[:pattern][:default]
362
+ end
363
+ if regex[:pattern][:fields]
364
+ regex[:pattern][:fields].each do |k,v|
365
+ hash["f.#{k}.hl.regex.pattern"] = v
366
+ end
367
+ end
368
+ else
369
+ hash["hl.regex.pattern"] = regex[:pattern]
370
+ end
371
+ end
372
+ if regex[:max_analyzed_chars]
373
+ if regex[:max_analyzed_chars].kind_of? Hash
374
+ if regex[:max_analyzed_chars][:default]
375
+ hash["hl.regex.maxAnalyzedChars"] = regex[:max_analyzed_chars][:default]
376
+ end
377
+ if regex[:max_analyzed_chars][:fields]
378
+ regex[:max_analyzed_chars][:fields].each do |k,v|
379
+ hash["f.#{k}.hl.regex.maxAnalyzedChars"] = v
380
+ end
381
+ end
382
+ else
383
+ hash["hl.regex.maxAnalyzedChars"] = regex[:max_analyzed_chars]
384
+ end
385
+ end
386
+ end
387
+
388
+ end
389
+
390
+ if @params[:mlt]
391
+ hash[:mlt] = true
392
+ hash["mlt.count"] = @params[:mlt][:count]
393
+ hash["mlt.fl"] = @params[:mlt][:field_list].join(',')
394
+ hash["mlt.mintf"] = @params[:mlt][:min_term_freq]
395
+ hash["mlt.mindf"] = @params[:mlt][:min_doc_freq]
396
+ hash["mlt.minwl"] = @params[:mlt][:min_word_length]
397
+ hash["mlt.maxwl"] = @params[:mlt][:max_word_length]
398
+ hash["mlt.maxqt"] = @params[:mlt][:max_query_terms]
399
+ hash["mlt.maxntp"] = @params[:mlt][:max_tokens_parsed]
400
+ hash["mlt.boost"] = @params[:mlt][:boost]
401
+ end
402
+
403
+ hash.merge(super.to_hash)
404
+ end
405
+
406
+ end