elastic_query 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: b32205036e3213b2463a73ed622673113f5f58c6
-  data.tar.gz: 299718483f5eb901b6e15dd815c6bac3877a7d6d
+  metadata.gz: de207ca4fb2147781175ef0915edc4ad14bd9345
+  data.tar.gz: 4ee91de6a55bb5b109650b91b861159a64c26302
 SHA512:
-  metadata.gz: 7e614212fc89732d869cb1848ed83c5d0039f4a15bc697883feb79116a5dc2c4031e3359517dc0715e1704998de1eeed026ef811671b5cd77a6ecd46da8a1b57
-  data.tar.gz: 3b74350272a7cc5d0741f00fd6dec85262e74762cb325a3fb8e58af5854dc5846e48ee4c4c730ce9d79d85ccb86a9c30acadcea711d804d669d9d61607646303
+  metadata.gz: a4e7b46586192d4b32c192ec1531ec1c77f0547910907a82e5e2273414aa8605a741b8a90d48a8eea7abc0752575ce268126272afe615790ce4572186c89c47d
+  data.tar.gz: f9401f6a59b632ce777949a1441e00b0746f95a0a3430072bddbbf280006484fe835c6475d861a66c57cb915a29bd8e203a1055e2d21b8d1467aabc23cc46f30

data/lib/elastic_query/builder.rb

@@ -1,18 +1,18 @@
 module ElasticQuery
   class Builder
-    attr_reader :query
+    attr_accessor :query
 
     def initialize(query)
-      @query = Query.new(query)
+      self.query = Query.new(query)
     end
 
     def filter(filter_query)
-      @query = FilteredQuery.new(filter_query, query)
+      self.query = FilteredQuery.new(filter_query, query)
       self
     end
 
     def rescore(rescore_query, rescore_options = {})
-      @query = RescoreQuery.new(query, rescore_query, rescore_options)
+      self.query = RescoreQuery.new(query, rescore_query, rescore_options)
       self
     end
 
@@ -25,5 +25,10 @@ module ElasticQuery
     def filter_nested(object_name, query, options = {})
       filter(NestedQuery.new(object_name, query, options))
     end
+
+    def function_score(functions, options = {})
+      self.query = FunctionScore.new(query, functions, options)
+      self
+    end
   end
 end

data/lib/elastic_query/function_score.rb

@@ -0,0 +1,19 @@
+module ElasticQuery
+  class FunctionScore < Query
+    attr_reader :functions, :options
+
+    def initialize(query, functions, options = {})
+      @functions = functions
+      @options = options
+      super(query)
+    end
+
+    def to_hash
+      {
+        function_score: {
+          query: query.to_hash,
+        }.merge(functions.to_hash).merge(options)
+      }
+    end
+  end
+end

data/lib/elastic_query/query.rb

@@ -3,7 +3,7 @@ module ElasticQuery
   # @example
   #   Quey.new(match_all: {})
   class Query < BaseQuery
-    attr_reader :query
+    attr_accessor :query
 
     def initialize(query)
       if query.class < BaseQuery

data/lib/elastic_query/version.rb

@@ -1,3 +1,3 @@
 module ElasticQuery
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

data/lib/elastic_query.rb

@@ -6,6 +6,7 @@ end
 require "elastic_query/version"
 require "elastic_query/base_query"
 require "elastic_query/query"
+require "elastic_query/function_score"
 require "elastic_query/nested_query"
 require "elastic_query/query_collection"
 require "elastic_query/filtered_query"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: elastic_query
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Kristopher Bredemeier
@@ -74,6 +74,7 @@ files:
 - lib/elastic_query/base_query.rb
 - lib/elastic_query/builder.rb
 - lib/elastic_query/filtered_query.rb
+- lib/elastic_query/function_score.rb
 - lib/elastic_query/nested_query.rb
 - lib/elastic_query/query.rb
 - lib/elastic_query/query_collection.rb