scalastic 0.6.0 → 0.7.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 7a323230b42e70b738a01c5ad1dcdba2ce284155
-  data.tar.gz: 2fe66cdd753c110c60daa3bbd17355d72a5eecd4
+  metadata.gz: 658c178987891b8d35d5e53453957d166be292e8
+  data.tar.gz: c01b6636034a41ea44c2c02216d53d6c7fb801e7
 SHA512:
-  metadata.gz: d9bbd87a04f3ab5ebd0012bd6e47bb28f1572e706fa12ea1abbca9f6289773bc47bfd2932907dae3ebd26215173fbca93b231b8239a7a9b9b22db0af8fb2e5b0
-  data.tar.gz: f042ba5eb5743a51684643c032fa6421a2b2c088cf438c2772e08c515147265f49b1f2c42cdf2ddca20d109641b21da2aea3656c01cadfd29ea0b3a523e0be03
+  metadata.gz: 85c8a15a5b37793884bd143493ceea7d75bfebdb9afa16f4aa67c19b8d500d8a53863190c7050bd20907b42b5ff8cd994718c765d40f00c3fe2b4c732f2f6440
+  data.tar.gz: 3d87aebccf6648854df7f4ac582fe183432cba07723847665313bc8456180ac87bdae5a3ae06aabec9f6b5cd404b201e7e210cb0a6643104110cecd96e5b817d

data/lib/scalastic/partition.rb

@@ -42,6 +42,11 @@ module Scalastic
       es_client.get(args)
     end
 
+    def mget(args)
+      args = args.merge(index: config.search_endpoint(id))
+      es_client.mget(args)
+    end
+
     def index(args)
       args[:body] ||= {}
       selector.apply_to(args[:body])
@@ -49,6 +54,13 @@ module Scalastic
       es_client.index(args)
     end
 
+    def create(args)
+      args[:body] ||= {}
+      selector.apply_to(args[:body])
+      args = args.merge(index: config.index_endpoint(id))
+      es_client.create(args)
+    end
+
     def delete(args = {})
       args = args.merge(index: config.search_endpoint(id))
       es_client.delete(args)

data/lib/scalastic/version.rb

@@ -1,3 +1,3 @@
 module Scalastic
-  VERSION = "0.6.0"
+  VERSION = "0.7.0"
 end

data/regression/regression_tests/document_create.rb

@@ -0,0 +1,32 @@
+module RegressionTests
+  module DocumentCreate
+    extend self
+
+    def cleanup
+      client = Elasticsearch::Client.new
+      client.indices.delete(index: 'document_create') if client.indices.exists?(index: 'document_create')
+    end
+
+    def run
+      client = Elasticsearch::Client.new
+      client.indices.create(index: 'document_create')
+      client.partitions.prepare_index(index: 'document_create')
+
+      partition = client.partitions.create(index: 'document_create', id: 1)
+      partition.create(type: 'test', id: 1, body: {subject: 'Test 1'})
+      partition.create(type: 'test', id: 2, body: {subject: 'Test 2'})
+
+      res = partition.create(type: 'test', id: 1, body: {subject: 'Test 1'}) rescue :failed
+      raise 'Indexing didn\'t fail' unless res == :failed
+      sleep(1.5)
+
+      hits = partition.search()['hits']['hits'].sort{|h1, h2| h1['_id'].to_i <=> h2['_id'].to_i}
+      expected_hits = [
+        {'_index' => 'document_create', '_type' => 'test', '_id' => '1', '_score' => 1.0, '_source' => {'subject' => 'Test 1', 'scalastic_partition_id' => 1}},
+        {'_index' => 'document_create', '_type' => 'test', '_id' => '2', '_score' => 1.0, '_source' => {'subject' => 'Test 2', 'scalastic_partition_id' => 1}},
+      ]
+
+      raise "Expected: #{expected_hits}, got: #{hits}" unless expected_hits == hits
+    end
+  end
+end

data/regression/regression_tests/mget.rb

@@ -0,0 +1,32 @@
+module RegressionTests
+  module Mget
+    extend self
+
+    def cleanup
+      client = Elasticsearch::Client.new
+      client.indices.delete(index: 'mget') if client.indices.exists?(index: 'mget')
+    end
+
+    def run
+      client = Elasticsearch::Client.new
+      client.indices.create index: 'mget'
+      client.partitions.prepare_index index: 'mget'
+
+      partition = client.partitions.create id: 1, index: 'mget'
+      partition.index id: 1, type: 'test', body: {subject: 'Test 1'}
+      partition.index id: 2, type: 'test', body: {subject: 'Test 2'}
+
+      expected_results = [
+        {'_index' => 'mget', '_type' => 'test', '_id' => '1', '_version' => 1, 'found' => true, '_source' => {'subject' => 'Test 1', 'scalastic_partition_id' => 1}},
+        {'_index' => 'mget', '_type' => 'test', '_id' => '2', '_version' => 1, 'found' => true, '_source' => {'subject' => 'Test 2', 'scalastic_partition_id' => 1}},
+        {'_index' => 'mget', '_type' => 'test', '_id' => '3', 'found' => false}
+      ]
+
+      results = partition.mget(type: 'test', body: {ids: [1, 2, 3]})['docs']
+      raise "Expected: #{expected_results}, got: #{results}" unless expected_results == results
+
+      results = partition.mget(body: {docs: [{_type: 'test', _id: 1}, {_type: 'test', _id: 2}, {_type: 'test', _id: 3}]})['docs']
+      raise "Expected: #{expected_results}, got: #{results}" unless expected_results == results
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: scalastic
 version: !ruby/object:Gem::Version
-  version: 0.6.0
+  version: 0.7.0
 platform: ruby
 authors:
 - Aliaksei Baturytski
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2016-06-12 00:00:00.000000000 Z
+date: 2016-06-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -110,11 +110,13 @@ files:
 - regression/regression_tests/custom_partition_prefix.rb
 - regression/regression_tests/delete_by_query.rb
 - regression/regression_tests/delete_partition.rb
+- regression/regression_tests/document_create.rb
 - regression/regression_tests/document_get.rb
 - regression/regression_tests/endpoints.rb
 - regression/regression_tests/extend_partition.rb
 - regression/regression_tests/index_destinations.rb
 - regression/regression_tests/list_partitions.rb
+- regression/regression_tests/mget.rb
 - regression/regression_tests/nested_selector.rb
 - regression/regression_tests/nested_selector_bulk.rb
 - regression/regression_tests/partition_operations.rb