elasticband 0.1.9 → 0.1.10

checksums.yaml

@@ -1,15 +1,15 @@
 ---
 !binary "U0hBMQ==":
   metadata.gz: !binary |-
-    MGNmMTIxZjU0NmI4MjA2MGQxMmU3Y2UwZWYyZDlkMmU4MDJlMmFiYg==
+    OWFmNDU0YTUwMGZjYmQwOWU1ZTBkMWMyNWJmMzNhYmQ0NjM4NmMxMQ==
   data.tar.gz: !binary |-
-    ZGM2YmVjZTc4MmQzNjJhNmE4OTBiMzJkNWE3MjVmNzBjMjI5MTJiZg==
+    Yzg4YTQzM2M3ODM5Zjk3Nzg3MWYyY2JmOTFjY2U5MDAzMmI2OGVhOQ==
 SHA512:
   metadata.gz: !binary |-
-    ZDc2Yzg4M2VkYzIyZDI5ZDMyOTQ3OTNmNWI2YTYyMTcxOTg1MDZjM2Y4YTBk
-    OWU3Yzk5MzFlNGVmOWNjMDkzNWIzNDkwMzllNjAwZDRhYzNmYTUwMDE4NWEw
-    OTNlMjkzYTI4NWM0NWVjYmUyYjM0NGM2YWJkNzgwNmEwOTYxMDE=
+    N2NkOGUwZGUzMWIzNjBiMmZkZWRkOGQ1NTBhOGUwYWQ5ZDA3YTFmOTg5OGQ0
+    ODFjMjdiZWNiMWFiNGIyZjc1YWE0ZjhkYmNkY2UyZjVmYmRjYjRiYTQ0YWRm
+    YjBmODRmMmZjZGVlZTRjM2JhNmIwOTEyYmEyMzE0YjFmZjExNWE=
   data.tar.gz: !binary |-
-    MGM5OTc3Y2NkMmY5MWY3OGVhNTZlZTM3OTlhYjlhYjFiNDA0MmEyMmJlNzhh
-    ZDE1Njg5NzNmZmNjMWIzZTg1ZDMxMzM1NzFlZDFjYTg4OTdlYWQ5ZDgwZDc2
-    NDFkMTIxMzc0Yzc5ODE3YjcxMDJkMTNiNWIwNTk1YWQzNmMzMzQ=
+    YzUzMGVkM2FjY2IzMDIzZTM4M2VjMWQwODYxYmQzYTg2ODkzMWEzMzE1ODhm
+    NDc0ZmY1ODBiMWZmNzhiOWMzMjVhMmY4MjZjNTJjYzgyNDNiZjY2MjQwNWRh
+    MWMwZjY3NjU1YmJiY2RhYzljM2YyYzAzNTllYTc3ODk5OWYxOWY=

data/lib/elasticband/query.rb

@@ -73,8 +73,8 @@ module Elasticband
         query = parse_boost(
           query,
           options.slice(:boost_by, :boost_where, :boost_function),
-          options.slice(:score_mode),
-          options.slice(:boost_mode)
+          options[:score_mode],
+          options[:boost_mode]
         )
         query.to_h
       end

data/lib/elasticband/version.rb

@@ -1,3 +1,3 @@
 module Elasticband
-  VERSION = '0.1.9'
+  VERSION = '0.1.10'
 end

data/spec/query_spec.rb

@@ -67,6 +67,40 @@ RSpec.describe Elasticband::Query do
             }
           )
         end
+
+        context 'with `:boost_mode` option' do
+          let(:options) { { boost_by: :contents_count, boost_mode: :multiply } }
+
+          it 'returns a function score query with the boost_mode' do
+            is_expected.to eq(
+              function_score: {
+                query: { match: { _all: 'foo' } },
+                boost_mode: :multiply,
+                field_value_factor: {
+                  field: :contents_count,
+                  modifier: :ln2p
+                }
+              }
+            )
+          end
+        end
+
+        context 'with `:score_mode` option' do
+          let(:options) { { boost_by: :contents_count, score_mode: :multiply } }
+
+          it 'returns a function score query with the score_mode' do
+            is_expected.to eq(
+              function_score: {
+                query: { match: { _all: 'foo' } },
+                score_mode: :multiply,
+                field_value_factor: {
+                  field: :contents_count,
+                  modifier: :ln2p
+                }
+              }
+            )
+          end
+        end
       end
 
       context 'with `:boost_function` option' do
@@ -83,6 +117,38 @@ RSpec.describe Elasticband::Query do
               }
             )
           end
+
+          context 'with `:boost_mode` option' do
+            let(:options) { { boost_function: "_score * doc['users_count'].value", boost_mode: :multiply } }
+
+            it 'returns a function score query with the boost_mode' do
+              is_expected.to eq(
+                function_score: {
+                  query: { match: { _all: 'foo' } },
+                  boost_mode: :multiply,
+                  script_score: {
+                    script: "_score * doc['users_count'].value"
+                  }
+                }
+              )
+            end
+          end
+
+          context 'with `:score_mode` option' do
+            let(:options) { { boost_function: "_score * doc['users_count'].value", score_mode: :multiply } }
+
+            it 'returns a function score query with the score_mode' do
+              is_expected.to eq(
+                function_score: {
+                  query: { match: { _all: 'foo' } },
+                  score_mode: :multiply,
+                  script_score: {
+                    script: "_score * doc['users_count'].value"
+                  }
+                }
+              )
+            end
+          end
         end
 
         context 'with params' do
@@ -127,6 +193,44 @@ RSpec.describe Elasticband::Query do
               }
             )
           end
+
+          context 'with `:boost_mode` option' do
+            let(:options) { { boost_where: { status: :published }, boost_mode: :multiply } }
+
+            it 'returns a function score query with the boost_mode' do
+              is_expected.to eq(
+                function_score: {
+                  query: { match: { _all: 'foo' } },
+                  boost_mode: :multiply,
+                  functions: [
+                    {
+                      filter: { name: 'filter' },
+                      boost_factor: 1000
+                    }
+                  ]
+                }
+              )
+            end
+          end
+
+          context 'with `:score_mode` option' do
+            let(:options) { { boost_where: { status: :published }, score_mode: :multiply } }
+
+            it 'returns a function score query with the score_mode' do
+              is_expected.to eq(
+                function_score: {
+                  query: { match: { _all: 'foo' } },
+                  score_mode: :multiply,
+                  functions: [
+                    {
+                      filter: { name: 'filter' },
+                      boost_factor: 1000
+                    }
+                  ]
+                }
+              )
+            end
+          end
         end
       end
     end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: elasticband
 version: !ruby/object:Gem::Version
-  version: 0.1.9
+  version: 0.1.10
 platform: ruby
 authors:
 - Glauber Campinho