cirandas.net

 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

module SolrPlugin

  module ActsAsSearchable

    ACTS_AS_SEARCHABLE_ENABLED = true unless defined? ACTS_AS_SEARCHABLE_ENABLED

    def acts_as_searchable options = {}
      return if !ACTS_AS_SEARCHABLE_ENABLED

      extend FindByContents
      include InstanceMethods

      class_attribute :solr_extra_fields
      self.solr_extra_fields = []

      if options[:fields]
        options[:fields] << {schema_name: :string}
        fields = options[:fields]
      else
        options[:additional_fields] ||= []
        options[:additional_fields] << {schema_name: :string}
        fields = options[:additional_fields]
      end
      fields.concat self.solr_facets_fields if self.respond_to? :solr_facets_fields
      fields.concat self.solr_extra_fields

      acts_as_solr options

      # don't crash on index errors
      handle_asynchronously :solr_save
      # solr_destroy don't work with delayed_job, as AR won't be found
      #handle_asynchronously :solr_destroy
    end

    module InstanceMethods
      def schema_name
        self.class.schema_name
      end

      # replace solr_id from acts_as_solr_reloaded/lib/acts_as_solr/instance_methods.rb
      # to include schema_name
      def solr_id
        id = "#{self.class.name}:#{record_id(self)}"
        id.insert 0, "#{schema_name}:" unless schema_name.blank?
        id
      end
    end

    module FindByContents

      def schema_name
        (Noosfero::MultiTenancy.on? and ActiveRecord::Base.postgresql?) ? ActiveRecord::Base.connection.schema_search_path : ''
      end

      def find_by_contents query, pg_options = {}, options = {}
        scope = self

        pg_options[:page]     ||= 1
        pg_options[:per_page] ||= 20

        options[:page] = pg_options[:page].to_i
        per_page       = options[:per_page] = pg_options[:per_page].to_i

        if options[:extra_limit]
          options[:per_page] = options.delete :extra_limit
          scope = scope.limit per_page
        end

        options[:scores] ||= true
        options[:filter_queries] ||= []
        options[:filter_queries] << "schema_name:\"#{schema_name}\"" unless schema_name.blank?
        options[:query_fields] = options.delete :fields if options[:fields].present?
        options[:facets] ||= {}
        all_facets_enabled = options.delete :all_facets
        results = []
        facets = all_facets = {}

        solr_result = scope.find_by_solr query, options
        if all_facets_enabled
          options[:facets][:browse] = nil
          all_facets = find_by_solr(query, options.merge(per_page: 0, results_format: :none)).facets
        end

        facets = solr_result.facets if options.include? :facets and solr_result.present?
        # solr_results has pagination options
        results = solr_result

        {results: results, facets: facets, all_facets: all_facets}
      end
    end
  end
end