增加批量导入

Gemini-D · Jul 4, 2024 · 5a4b652 · 5a4b652
1 parent 5cfdb69
commit 5a4b652
Show file tree

Hide file tree

Showing 3 changed files with 53 additions and 2 deletions.
diff --git a/src/Builder.php b/src/Builder.php
@@ -131,6 +131,34 @@ public function toBody(): array
         ];
     }
 
+    public function bulk(
+        array $docs,
+        #[ArrayShape([
+            'refresh' => 'bool',
+            'retry_on_conflict' => 'int',
+        ])]
+        ?array $settings = null
+    ): bool {
+        $settings ??= $this->document->getConfig()->getUpdateSettings();
+
+        $body = [];
+        foreach ($docs as $doc) {
+            $body[] = [
+                'index' => [
+                    '_index' => $this->document->getIndex(),
+                    '_id' => $doc[$this->document->getKey()],
+                ],
+            ];
+
+            $body[] = $doc;
+        }
+
+        return $this->document->getClient()->bulk([
+            'body' => $body,
+            ...$settings,
+        ])->asBool();
+    }
+
     public function update(
         array $doc,
         mixed $id = null,

diff --git a/tests/Cases/BuilderTest.php b/tests/Cases/BuilderTest.php
@@ -106,10 +106,10 @@ public function testUpdateAndGet()
     public function testLTAndGT()
     {
         $res = Foo::query()->where('id', '>', 4)->get();
-        $this->assertSame(1, $res->count());
+        $this->assertSame(3, $res->count());
 
         $res = Foo::query()->where('id', '>=', 4)->get();
-        $this->assertSame(2, $res->count());
+        $this->assertSame(4, $res->count());
     }
 
     public function testTerms()
@@ -166,4 +166,20 @@ public function testOrWhereClosure()
         $this->assertSame(2, $res->first()['id']);
         $this->assertSame(5, $res->last()['id']);
     }
+
+    public function testBulk()
+    {
+        $res = Foo::query()->bulk([
+            ['id' => 6, 'name' => 'elastic', 'summary' => $id = uniqid()],
+            ['id' => 7, 'name' => 'elastic2', 'summary' => $id2 = uniqid()],
+        ]);
+
+        $this->assertTrue($res);
+
+        $first = Foo::query()->where('id', 6)->get()->first();
+        $this->assertSame($id, $first['summary']);
+
+        $first = Foo::query()->where('id', 7)->get()->first();
+        $this->assertSame($id2, $first['summary']);
+    }
 }
diff --git a/tests/put_documents.php b/tests/put_documents.php
@@ -58,3 +58,10 @@ public function getMapping(): array
 foreach ($docs as $doc) {
     Foo::query()->update($doc, $doc['id']);
 }
+
+$res = Foo::query()->bulk([
+    ['id' => 6, 'name' => 'elastic', 'summary' => uniqid()],
+    ['id' => 7, 'name' => 'elastic2', 'summary' => uniqid()],
+]);
+
+var_dump($res);