es_query_builder 1.0.3 → 2.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: dd7b765e4166da57cfc9715f0d4e5ab03c3108d4
-  data.tar.gz: 79c474f1f90498d471ab3f64a6b5cfe7504a894e
+  metadata.gz: 476bcc5f9cc581265e2a05f04d04762f43d022b3
+  data.tar.gz: 10434493e8ff90f36a1ac52cfdba78686d15d9d8
 SHA512:
-  metadata.gz: 0b14528af93dc4e9206f25b227cb72ba1bd6ebe97bb3526d302d68f81aa77cfef90f9301e97dc5cecc0bc51188fbcc219f9c8db9090dc6bd951795aafacfde44
-  data.tar.gz: 8845e9f956cdb0fed6ccc36ab9dae3906d2cb75faf2818f61ff8070014069ecac4abb13e5804f5692b75c4980ee3b6af662272472338ad1b984f3b951c090c55
+  metadata.gz: 3a826034f827ffa7444d9155ddab30ebac685bd3e0bcf756a49a189b90cc6a7c7896211d5025b4cf8a9c5f8d4e90d25b5f8c5daf3b431009c952ee39f2c78ba5
+  data.tar.gz: 5412d2c7016367ec5af2f74e4de4e70a9846eb46e4b06138e5b326e925602c3b1f237fd5d28d17e9c466d3b31c3c09fc3cf1186733d1ca0ef298be12cd36dc93

data/lib/aggregation_builders.rb

@@ -0,0 +1,59 @@
+class AggregationBuilders
+
+  def self.terms(*args)
+    return ::Aggregations::Buckets::TermsAggregationBuilder.new(*args)
+  end
+
+  def self.filter(*args)
+    return ::Aggregations::Buckets::FilterAggregationBuilder.new(*args)
+  end
+
+  def self.filters(*args)
+    return ::Aggregations::Buckets::FiltersAggregationBuilder.new(*args)
+  end
+
+  def self.geohash_grid(*args)
+    return ::Aggregations::Buckets::GeoGridAggregationBuilder.new(*args)
+  end
+
+  def self.top_hits(*args)
+    return ::Aggregations::Buckets::TopHitsAggregationBuilder.new(*args)
+  end
+
+  def self.nested(*args)
+    return ::Aggregations::Buckets::NestedAggregationBuilder.new(*args)
+  end
+
+  def self.reverse_nested(*args)
+    return ::Aggregations::Buckets::ReverseNestedAggregationBuilder.new(*args)
+  end
+
+  def self.date_histogram(*args)
+    return ::Aggregations::Buckets::DateHistogramAggregationBuilder.new(*args)
+  end
+
+  def self.histogram(*args)
+    return ::Aggregations::Buckets::HistogramAggregationBuilder.new(*args)
+  end
+
+  def self.date_range(*args)
+    return ::Aggregations::Buckets::DateRangeAggregationBuilder.new(*args)
+  end
+
+  def self.range(*args)
+    return ::Aggregations::Buckets::RangeAggregationBuilder.new(*args)
+  end
+
+  def self.max(*args)
+    return ::Aggregations::Metrics::MaxAggregationBuilder.new(*args)
+  end
+
+  def self.min(*args)
+    return ::Aggregations::Metrics::MinAggregationBuilder.new(*args)
+  end
+
+  def self.avg(*args)
+    return ::Aggregations::Metrics::AvgAggregationBuilder.new(*args)
+  end
+
+end

data/lib/attributes_reader.rb

@@ -0,0 +1,39 @@
+module AttributesReader
+
+  class Attributes
+
+    def initialize parent_object
+      @parent_object = parent_object
+    end
+
+    def method_missing(m, *args, &block)
+      if @parent_object.methods.include?("#{m}_expr".intern)
+        @parent_object.send("#{m}_expr", *args)
+      else
+        raise MissingAttributeError.new(m, @parent_object.class)
+      end
+    end
+
+  end
+
+  class MissingAttributeError < StandardError
+
+    def initialize attr_name, class_name
+      @attr = attr_name
+      @class = class_name
+      super(message)
+    end
+
+    def message
+      return "Missing Attribute #{@attr} for #{@class}"
+    end
+
+  end
+
+
+  def attributes
+    @attributes ||= Attributes.new(self)
+    return @attributes
+  end
+
+end

data/lib/es_query_builder.rb

@@ -1,8 +1,30 @@
-require 'elastic_search_query'
-require 'token_query_builder'
-require 'fetch_es_data'
-require 'indexer'
-require 'constants'
+# require 'elastic_search_query'
+# require 'token_query_builder'
+# require 'fetch_es_data'
+# require 'indexer'
+# require 'constants'
+require 'active_support/all'
+require 'elasticsearch'
+require_relative 'attributes_reader'
+Dir[File.join(__dir__, 'enums', '*.rb')].each { |file| require file }
+Dir[File.join(__dir__, 'common', '*.rb')].each { |file| require file }
+Dir[File.join(__dir__, 'function_scores', '*.rb')].each { |file| require file }
+Dir[File.join(__dir__, 'misc', '*.rb')].each { |file| require file }
+Dir[File.join(__dir__, 'queries', '*.rb')].each { |file| require file }
+Dir[File.join(__dir__, 'scripts', '*.rb')].each { |file| require file }
+Dir[File.join(__dir__, 'sort', '*.rb')].each { |file| require file }
+Dir[File.join(__dir__, 'misc', '*.rb')].each { |file| require file }
+Dir[File.join(__dir__, 'aggregations', 'helpers', '*.rb')].each { |file| require file }
+Dir[File.join(__dir__, 'aggregations', 'buckets', '*.rb')].each { |file| require file }
+Dir[File.join(__dir__, 'aggregations', 'metrics', '*.rb')].each { |file| require file }
+require_relative 'query_builders'
+require_relative 'function_score_builders'
+Dir[File.join(__dir__, 'aggregation_builders.rb')].each { |file| require file }
+require_relative 'sort_builders'
+require_relative 'search_source_builder'
+require_relative 'search_request'
+require_relative 'multi_search_request'
+require_relative 'rest_client'
 
 module EsQueryBuilder
 end

data/lib/function_score_builders.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+# Factory class for function scores.
+class FunctionScoreBuilders
+  # @return [FunctionScores::RandomScoreFunctionBuilder]
+  #   random_score_function_builder object
+  def self.random_score_function
+    FunctionScores::RandomScoreFunctionBuilder.new
+  end
+
+  # @params [Scripts::Script] script object for the function
+  # @return [FunctionScores::ScriptScoreFunctionBuilder]
+  #   script_score_function_builder object with the
+  #   provided script
+  def self.script_score_function(*args)
+    FunctionScores::ScriptScoreFunctionBuilder.new(*args)
+  end
+
+  # @return [FunctionScores::WeightedScoreFunctionBuilder]
+  #   weighted_score_function_builder object
+  def self.weighted_score_function
+    FunctionScores::WeightedScoreFunctionBuilder.new
+  end
+end

data/lib/multi_search_request.rb

@@ -0,0 +1,22 @@
+class MultiSearchRequest
+  include AttributesReader
+
+  def initialize
+    @requests = []
+  end
+
+  def body
+    return @requests.map{|request| {index: request.index, search: request.body}}
+  end
+
+# show requests
+  def requests_expr
+    return @requests
+  end
+# add requests
+  def add request
+    @requests.add(request)
+    return self
+  end
+
+end

data/lib/query_builders.rb

@@ -0,0 +1,105 @@
+# frozen_string_literal: true
+
+# Factory class for query classes
+class QueryBuilders
+  # @return [Queries::BoolQueryBuilder]
+  #   bool_query_builder object
+  def self.bool_query
+    Queries::BoolQueryBuilder.new
+  end
+
+  # @params [String] field_name
+  #   field on which match query to be performed
+  # @params [String] value value to be matched
+  # @return [Queries::MatchQueryBuilder]
+  #   match_query_builder object
+  def self.match_query(*args)
+    Queries::MatchQueryBuilder.new(*args)
+  end
+
+  # @params [String] field_name
+  #   field on which term query to be performed
+  # @params [String] value value to be matched
+  # @return [Queries::TermQueryBuilder]
+  #   term_query_builder object
+  def self.term_query(*args)
+    Queries::TermQueryBuilder.new(*args)
+  end
+
+  # @params [String] field_name
+  #   field on which terms query to be performed
+  # @params [String] values values to be matched
+  # @return [Queries::TermsQueryBuilder]
+  #   terms_query_builder object
+  def self.terms_query(*args)
+    Queries::TermsQueryBuilder.new(*args)
+  end
+
+  # @params [String] field_name
+  #   field on which range query to be performed
+  # @return [Queries::RangeQueryBuilder]
+  #   range_query_builder object
+  def self.range_query(*args)
+    Queries::RangeQueryBuilder.new(*args)
+  end
+
+  # @params [String] field_name
+  #   field on which exists query to be performed
+  # @return [Queries::ExistsQueryBuilder]
+  #   exists_query_builder object
+  def self.exists_query(*args)
+    Queries::ExistsQueryBuilder.new(*args)
+  end
+
+  # @params [Queries::QueryBuilder] inner_query
+  #   query to be wrapped by constant score query
+  # @return [Queries::ConstantScoreQueryBuilder]
+  #   constant_score_query_builder object
+  def self.constant_score_query(*args)
+    Queries::ConstantScoreQueryBuilder.new(*args)
+  end
+
+  # @return [Queries::DisMaxQueryBuilder]
+  #   dis_max_query_builder object
+  def self.dis_max_query
+    Queries::DisMaxQueryBuilder.new
+  end
+
+  # @params [Queries::QueryBuilder] query
+  #   query part in the function score query object
+  # @params [FunctionScores::ScoreFunctionBuilder] score_function
+  #   score_function for the function score query object
+  # @params [Queries::FunctionScoreQueryBuilder::FilterFunctionBuilder]
+  #   filter_functions filter functions for the function score query object
+  # @return [Queries::FunctionScorehQueryBuilder]
+  #   function_score_query_builder object
+  def self.function_score_query(*args)
+    Queries::FunctionScoreQueryBuilder.new(*args)
+  end
+
+  # @params [String] path
+  #   path to the nested field
+  # @params [Queries::QueryBuilder] inner_query
+  #   query on the nested field
+  # @params [Enums::ScoreModes] score_mode
+  #   score_mode for the matched documents
+  # @return [Queries::NestedQueryBuilder]
+  #   nested_query_builder object
+  def self.nested_query(*args)
+    Queries::NestedQueryBuilder.new(*args)
+  end
+
+  # @params [String] field_name
+  #   field on which exists query to be performed
+  # @return [Queries::GeoDistanceQueryBuilder]
+  #   geo_distance_query_builder object
+  def self.geo_distance_query(*args)
+    Queries::GeoDistanceQueryBuilder.new(*args)
+  end
+
+  # @return [Queries::MAtchAllQueryBuilder]
+  #   match_all_query_builder object
+  def self.match_all_query
+    Queries::MatchAllQueryBuilder.new
+  end
+end

data/lib/rest_client.rb

@@ -0,0 +1,53 @@
+class RestClient
+
+  def initialize host: '127.0.0.1', port: 9200
+    @client = Elasticsearch::Client.new(host: host, port: port)
+  end
+
+  def search search_request
+    return @client.search(index: search_request.attributes.index, body: search_request.body)
+  end
+
+  def msearch msearch_request
+    return @client.msearch(body: msearch_request.body)
+  end
+
+  def index *args
+    return @client.index(*args)
+  end
+
+  def bulk *args
+    return @client.bulk(*args)
+  end
+
+  def create *args
+    return @client.create(*args)
+  end
+
+  def delete *args
+    return @client.delete(*args)
+  end
+
+  def delete_by_query *args
+    args.first[:body] = SearchSourceBuilder.new.query(args.first[:body]).body
+    return @client.delete_by_query(*args)
+  end
+
+  def get *args
+    return @client.get(*args)
+  end
+
+  def mget *args
+    return @client.mget(*args)
+  end
+
+  def update *args
+    return @client.update(*args)
+  end
+
+  def update_by_query *args
+    args.first[:body] = SearchSourceBuilder.new.query(args.first[:body]).body
+    return @client.update_by_query(*args)
+  end
+
+end

data/lib/search_request.rb

@@ -0,0 +1,33 @@
+class SearchRequest
+  include AttributesReader
+
+  def initialize
+    @index = ''
+    @body = nil
+  end
+
+  def body
+    return @source.body
+  end
+
+# returns index
+  def index_expr
+    return @index
+  end
+# sets index
+  def index index
+    @index= index
+    return self
+  end
+
+# returns source
+  def source_expr
+    return @body
+  end
+# sets source
+  def source source
+    @source = source
+    return self
+  end 
+
+end

data/lib/search_source_builder.rb

@@ -0,0 +1,151 @@
+# A search source builder allowing to easily build search source.
+class SearchSourceBuilder
+  include AttributesReader
+
+  def initialize
+    @query = nil
+    @aggregations = []
+    @sort = []
+    @source = []
+    @size = nil
+    @explain = nil
+    @from = nil
+    @min_score = nil
+    @terminate_after = nil
+    @timeout = nil
+    @version = nil
+  end
+
+  def body
+    body = {}
+    body[:query] = @query.query
+    body[:aggregations] = @aggregations.map{|agg| agg.query}.reduce({}, :merge) if @aggregations.present?
+    body[:sort] = @sort.map{|sort_object| sort_object.query} if @sort.present?
+    body[:_source] = @source
+    body[:size] = @size
+    body[:explain] = @explain
+    body[:from] = @from
+    body[:min_score] = @min_score
+    body[:terminate_after] = @terminate_after
+    body[:timeout] = @timeout
+    body[:version] = @version
+    return body.compact
+  end
+
+# returns query
+  def query_expr
+    return @query
+  end
+# sets query
+  def query query
+    @query = query
+    return self
+  end
+
+# returns aggregation
+  def aggregations_expr
+    return @aggregations
+  end
+# sets aggregation
+  def aggregations agg_builder
+    if agg_builder.is_a?(Array)
+      @aggregations += agg_builder
+    else
+      @aggregations.append(agg_builder)
+    end
+    return self
+  end
+
+# returns sort
+  def sort_expr
+    return @sort
+  end
+# appends sort
+  def sort sort, reset: false
+    @sort = [] if reset
+    sort.is_a?(Array) ? @sort+= sort : @sort.append(sort)
+    return self
+  end
+
+# returns source_fields
+  def source_expr
+    return @source
+  end
+# sets source_fields
+  def source source_field, reset: false
+    @source = [] if reset
+    source_field.is_a?(Array) ? @source+= source_field : @source.append(source_field)
+    return self
+  end
+
+# returns size
+  def size_expr
+    return @size
+  end
+# sets size
+  def size size
+    @size = size
+    return self
+  end
+
+# returns explain
+  def explain_expr
+    return @explain
+  end
+# sets explain
+  def explain explain
+    @explain = explain
+    return self
+  end
+
+# returns from
+  def from_expr
+    return @from
+  end
+# sets from
+  def from from
+    @from = from
+    return self
+  end
+
+# returns min_score
+  def min_score_expr
+    return @min_score
+  end
+# sets min_score
+  def min_score min_score
+    @min_score = min_score
+    return self
+  end
+
+# returns terminate_after
+  def terminate_after_expr
+    return @terminate_after
+  end
+# sets terminate_after
+  def terminate_after terminate_after
+    @terminate_after = terminate_after
+    return self
+  end
+
+# returns timeout
+  def timeout_expr
+    return @timeout
+  end
+# sets timeout
+  def timeout timeout
+    @timeout = timeout
+    return self
+  end
+
+# returns version
+  def version_expr
+    return @version
+  end
+# sets version
+  def version version
+    @version = version
+    return self
+  end
+
+end

data/lib/sort_builders.rb

@@ -0,0 +1,23 @@
+class SortBuilders
+
+  def self.field_sort *args
+    return Sort::FieldSortBuilder.new(*args)
+  end
+
+  def self.geo_distance_sort *args
+    return Sort::GeoDistanceSortBuilder.new(*args)
+  end
+
+  def self.nested_sort *args
+    return Sort::NestedSortBuilder.new(*args)
+  end
+
+  def self.score_sort
+    return Sort::ScoreSortBuilder.new
+  end
+
+  def self.script_sort *args
+    return Sort::ScriptSortBuilder.new(*args)
+  end
+
+end

data/lib/test_script.rb

@@ -0,0 +1,29 @@
+Dir["/root/housing/es.query.builder/lib/*_helper.rb"].each {|file| require file }
+Dir["/root/housing/es.query.builder/lib/*_builder.rb"].each {|file| require file }
+Dir["/root/housing/es.query.builder/lib/*_builders.rb"].each {|file| require file }
+Dir["/root/housing/es.query.builder/lib/script.rb"].each {|file| require file }
+require 'active_support/all'
+
+script1 = Script.new("ctx._source.likes*modifier", {"modifier": 2}, :INLINE)
+t1 = AggregationBuilders.terms('term1').field('age').include(['anu1', 'anu2', 'anu3']).add_order('anu1').add_order('anu4').script(script1)
+puts t1.get_query
+t2 = AggregationBuilders.filter('filter1', {:bool => {:must => [{:f1 => 'v1'}]}}).sub_aggregation(t1)
+puts t2.get_query
+t3 = AggregationBuilders.geohash_grid('geo1').size(10).precision(5).sub_aggregation(t1)
+puts t3.get_query
+t4 = AggregationBuilders.top_hits('thits1').add_sort(:f1).add_sort(:f2, :asc).source(['a1', 'a2']).sub_aggregation(t1)
+puts t4.get_query
+t5 = AggregationBuilders.reverse_nested('rnested1').sub_aggregation(t1)
+puts t5.get_query
+t6 = AggregationBuilders.nested('nested1', 'room_types').sub_aggregation(t5)
+puts t6.get_query
+t7 = AggregationBuilders.date_histogram('dhist1').field('age').interval('M').add_order('anu1').add_order('anu4')
+puts t7.get_query
+t8 = AggregationBuilders.histogram('hist1').field('age').interval('M').add_order('anu1').add_order('anu4')
+puts t8.get_query
+t9 = AggregationBuilders.date_range('drange1').field('start_date').add_range("2016/02/01", "2017/02/01", 'older').add_range("2017/02/01", nil)
+puts t9.get_query
+t10 = AggregationBuilders.range('range1').field('age').add_range(10, 100, 'older').add_range(200, nil).keyed(true)
+puts t10.get_query
+t11 = AggregationBuilders.max('max1').field('age').missing(10)
+puts t11.get_query

metadata

@@ -1,14 +1,17 @@
 --- !ruby/object:Gem::Specification
 name: es_query_builder
 version: !ruby/object:Gem::Version
-  version: 1.0.3
+  version: 2.0.0
 platform: ruby
 authors:
 - Mohib Yousuf
+- Anurag Agrawal
+- Supantha Samanta
+- Neeraj Joshi
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-05-31 00:00:00.000000000 Z
+date: 2019-10-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -44,17 +47,59 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 3.5.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 3.5.0
 description: 
-email: mohib.yousuf@hotmail.com
+email:
+- mohib.yousuf@hotmail.com
+- anuragagrawal117@gmail.com
+- supantha.samanta@gmail.com
+- neerajjoshi00@gmail.com
 executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- lib/aggregation_builders.rb
+- lib/attributes_reader.rb
 - lib/constants.rb
 - lib/elastic_search_query.rb
 - lib/es_query_builder.rb
 - lib/fetch_es_data.rb
+- lib/function_score_builders.rb
 - lib/indexer.rb
+- lib/multi_search_request.rb
+- lib/query_builders.rb
+- lib/rest_client.rb
+- lib/search_request.rb
+- lib/search_source_builder.rb
+- lib/sort_builders.rb
+- lib/test_script.rb
 - lib/token_query_builder.rb
 homepage: https://github.com/elarahq/es.query.builder
 licenses:
@@ -76,7 +121,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.6.12
+rubygems_version: 2.6.11
 signing_key: 
 specification_version: 4
 summary: For Building Elastic Search Queries