如何使用关联模型条件搜索Elasticsearch-rails?

时间:2015-12-18 13:30:44

标签: ruby-on-rails-3 elasticsearch elasticsearch-rails

我正在尝试搜索属于特定项目的Doc模型has_and_belongs_to_many项目。

模型/ doc.rb 需要'elasticsearch / model'

class Doc < ActiveRecord::Base
  include Elasticsearch::Model
  include Elasticsearch::Model::Callbacks

  settings index: {
    analysis: {
      tokenizer: {
        ngram_tokenizer: {
          type: "nGram",
          min_gram: "2",
          max_gram: "3",
          token_chars: [
            "letter",
            "digit",
            "punctuation"
          ]
        }
      },
      analyzer: {
        ngram_analyzer: {
          tokenizer: "ngram_tokenizer"
        }
      },
    },
  } do
      mappings do
        indexes :sourcedb, type: 'string', analyzer: 'ngram_analyzer'
        indexes :sourceid, type: 'string', analyzer: 'ngram_analyzer'
        indexes :body, type: 'string', analyzer: 'ngram_analyzer'
        indexes :docs_projects do
          indexes :doc_id
          indexes :project_id
          indexes :projects do
            indexes :id, index: :not_analyzed
          end
        end
      end
  end

  def as_indexed_json(options={})
    as_json(
      only: [:id, :sourcedb, :sourceid, :body],
      include: { projects: {only: :id} }  
    )
  end
end

搜索方法如下

  search_docs = docs.search(
    query: {
      bool:{
        must: [
          {match: {
              'projects.id' => project_id
            }
          }
        ]
      }
    },
    size: 5000,
  ).records.order('sourcedb ASC, sourceid ASC').paginate(page:params[:page], per_page: 10)

此搜索方法没有错误但没有返回任何内容。

  Doc Load (4.6ms)  SELECT "docs".* FROM "docs" INNER JOIN "docs_projects" ON "docs"."id" = "docs_projects"."doc_id" WHERE "docs_projects"."project_id" = 56
   (0.4ms)  SELECT COUNT(*) FROM "docs" WHERE 1=0
  Doc Load (0.3ms)  SELECT "docs".* FROM "docs" WHERE 1=0 ORDER BY sourcedb ASC, sourceid ASC LIMIT 10 OFFSET 0
  CACHE (0.0ms)  SELECT COUNT(*) FROM "docs" WHERE 1=0

我曾尝试搜索where([projects.id IN(?)],project_ids),但它无法搜索docs属于具有max size的项目。
如何通过与协会匹配进行搜索? 提前谢谢。

1 个答案:

答案 0 :(得分:0)

现在,我找到了解决方案。这就是我解决的问题。

doc.rb

  settings index: {
    analysis: {
      tokenizer: {
        ngram_tokenizer: {
          type: "nGram",
          min_gram: "2",
          max_gram: "3",
          token_chars: [
            "letter",
            "digit",
            "punctuation"
          ]
        }
      },
      analyzer: {
        ngram_analyzer: {
          tokenizer: "ngram_tokenizer"
        }
      },
    },
  } do
      mappings do
        indexes :sourcedb, type: 'string', analyzer: 'ngram_analyzer'
        indexes :sourceid, type: 'string', analyzer: 'ngram_analyzer'
        indexes :body, type: 'string', analyzer: 'ngram_analyzer'

        # indexes :docs_projects, type: 'nested' do
        indexes :docs_projects do
          indexes :doc_id
          indexes :project_id
        end

        indexes :projects do
          indexes :id, index: :not_analyzed
        end
      end
  end

搜索方法

  def self.search_docs(attributes = {})
    minimum_should_match = 0
    minimum_should_match += 1 if attributes[:sourcedb].present?
    minimum_should_match += 1 if attributes[:sourceid].present?
    minimum_should_match += 1 if attributes[:body].present?

    if attributes[:project_id].present?
      must_array = [
        {match: {
            'projects.id' => attributes[:project_id]
          }
        }
      ] 
    end

    docs = search(
      query: {
        bool:{
          must: must_array,
          should: [ 
            {match: {
              sourcedb: {
                query: attributes[:sourcedb],
                fuzziness: 0
              }
            }
            },
            {match: {
              sourceid: {
                query: attributes[:sourceid],
                fuzziness: 0
              }
            }
            },
            {match: {
              body: {
                query: attributes[:body],
                fuzziness: 'AUTO'
              }
            }
            },
          ],
          minimum_should_match: minimum_should_match
        }
      },
      size: SEARCH_SIZE,
    )
    return {
      total: docs.results.total,
      docs: docs.records.order('sourcedb ASC, sourceid ASC')
    }
  end