Skip to content

Commit 9ce2f9f

Browse files
sandrokeilsandrokeil
committed
Add simdjson handler 🚀
1 parent e74dc7c commit 9ce2f9f

4 files changed

+230
-1
lines changed

‎src/Statement/SimdjsonHandler.php

+76
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,76 @@
1+
<?php
2+
/**
3+
* Sandro Keil (https://sandro-keil.de)
4+
*
5+
* @link http://github.com/sandrokeil/arangodb-php-client for the canonical source repository
6+
* @copyright Copyright (c) 2018-2020 Sandro Keil
7+
* @license http://github.com/sandrokeil/arangodb-php-client/blob/master/LICENSE.md New BSD License
8+
*/
9+
10+
declare(strict_types=1);
11+
12+
namespace ArangoDb\Statement;
13+
14+
use Psr\Http\Message\StreamInterface;
15+
16+
final class SimdjsonHandler implements StreamHandler
17+
{
18+
use SimdjsonStreamHandlerTrait;
19+
20+
/**
21+
* @var mixed
22+
*/
23+
private $data = [];
24+
25+
public function __construct(StreamInterface $stream)
26+
{
27+
$this->data[$this->fetches] = \simdjson_resource($stream->getContents());
28+
$this->length = count(\simdjson_key_value($this->data[$this->fetches], 'result', true));
29+
$this->batchSize = $this->length;
30+
}
31+
32+
/**
33+
* Return the current result row
34+
*
35+
* @return array
36+
*/
37+
public function current(): array
38+
{
39+
return \simdjson_key_value(
40+
$this->data[$this->fetches],
41+
"result\t" . ($this->position - ($this->batchSize * $this->fetches)),
42+
true
43+
);
44+
}
45+
46+
public function result(): array
47+
{
48+
return \simdjson_key_value(
49+
$this->data[$this->fetches],
50+
"result",
51+
true
52+
);
53+
}
54+
55+
public function raw(): array
56+
{
57+
return $this->data[$this->fetches];
58+
}
59+
60+
public function completeResult()
61+
{
62+
$completeResult = [[]];
63+
64+
foreach ($this->data as $result) {
65+
$completeResult[] = \simdjson_key_value($result, 'result', true);
66+
}
67+
return array_merge(...$completeResult);
68+
}
69+
70+
public function appendStream(StreamInterface $stream): void
71+
{
72+
$this->data[++$this->fetches] = \simdjson_resource($stream->getContents());
73+
$this->length += count(\simdjson_key_value($this->data[$this->fetches], 'result', true));
74+
}
75+
76+
}

‎src/Statement/SimdjsonStreamHandlerFactory.php

+22
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
1+
<?php
2+
/**
3+
* Sandro Keil (https://sandro-keil.de)
4+
*
5+
* @link http://github.com/sandrokeil/arangodb-php-client for the canonical source repository
6+
* @copyright Copyright (c) 2018-2020 Sandro Keil
7+
* @license http://github.com/sandrokeil/arangodb-php-client/blob/master/LICENSE.md New BSD License
8+
*/
9+
10+
declare(strict_types=1);
11+
12+
namespace ArangoDb\Statement;
13+
14+
use Psr\Http\Message\StreamInterface;
15+
16+
class SimdjsonStreamHandlerFactory implements StreamHandlerFactoryInterface
17+
{
18+
public function createStreamHandler(StreamInterface $body): StreamHandler
19+
{
20+
return new SimdjsonHandler($body);
21+
}
22+
}

‎src/Statement/SimdjsonStreamHandlerTrait.php

+129
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,129 @@
1+
<?php
2+
/**
3+
* Sandro Keil (https://sandro-keil.de)
4+
*
5+
* @link http://github.com/sandrokeil/arangodb-php-client for the canonical source repository
6+
* @copyright Copyright (c) 2018-2020 Sandro Keil
7+
* @license http://github.com/sandrokeil/arangodb-php-client/blob/master/LICENSE.md New BSD License
8+
*/
9+
10+
declare(strict_types=1);
11+
12+
namespace ArangoDb\Statement;
13+
14+
trait SimdjsonStreamHandlerTrait
15+
{
16+
/**
17+
* Current position in result set iteration (zero-based)
18+
*
19+
* @var int
20+
*/
21+
private $position = 0;
22+
23+
/**
24+
* Number of HTTP calls that were made to build the cursor result
25+
*
26+
* @var int
27+
*/
28+
private $fetches = 0;
29+
30+
/**
31+
* Total length of result set (in number of documents)
32+
*
33+
* @var int
34+
*/
35+
private $length;
36+
37+
/**
38+
* @var int
39+
*/
40+
private $batchSize;
41+
42+
public function cursorId(): ?string
43+
{
44+
return \simdjson_key_value($this->data[$this->fetches], 'id', true);
45+
}
46+
47+
public function hasMore(): bool
48+
{
49+
return \simdjson_key_value($this->data[$this->fetches], 'hasMore', true);
50+
}
51+
52+
public function resultCount(): ?int
53+
{
54+
return \simdjson_key_value($this->data[$this->fetches], 'count', true);
55+
}
56+
57+
/**
58+
* Get the total number of current loaded results.
59+
*
60+
* @return int Total number of laoded results
61+
*/
62+
public function count()
63+
{
64+
return $this->length;
65+
}
66+
67+
public function rewind(): void
68+
{
69+
$this->position = 0;
70+
}
71+
72+
public function key(): int
73+
{
74+
return $this->position;
75+
}
76+
77+
public function next(): void
78+
{
79+
$this->position++;
80+
}
81+
82+
/**
83+
* @return bool
84+
*/
85+
public function valid(): bool
86+
{
87+
return $this->position <= $this->length - 1;
88+
}
89+
90+
public function writesExecuted(): ?int
91+
{
92+
return \simdjson_key_value($this->data[$this->fetches], "extra\tstats\twritesExecuted", true);
93+
}
94+
95+
public function writesIgnored(): ?int
96+
{
97+
return \simdjson_key_value($this->data[$this->fetches], "extra\tstats\twritesIgnored", true);
98+
}
99+
100+
public function scannedFull(): ?int
101+
{
102+
return \simdjson_key_value($this->data[$this->fetches], "extra\tstats\tscannedFull", true);
103+
}
104+
105+
public function scannedIndex(): ?int
106+
{
107+
return \simdjson_key_value($this->data[$this->fetches], "extra\tstats\tscannedIndex", true);
108+
}
109+
110+
public function filtered(): ?int
111+
{
112+
return \simdjson_key_value($this->data[$this->fetches], "extra\tstats\tfiltered", true);
113+
}
114+
115+
public function fullCount(): ?int
116+
{
117+
return \simdjson_key_value($this->data[$this->fetches], "extra\tstats\tfullCount", true);
118+
}
119+
120+
public function warnings(): array
121+
{
122+
return \simdjson_key_value($this->data[$this->fetches], "extra\twarnings", true);
123+
}
124+
125+
public function isCached(): bool
126+
{
127+
return \simdjson_key_value($this->data[$this->fetches], "cached", true);
128+
}
129+
}

‎test/TestUtil.php

+3-1
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
use ArangoDb\Http\Client;
1515
use ArangoDb\Exception\ArangoDbException;
1616
use ArangoDb\Statement\ArrayStreamHandlerFactory;
17+
use ArangoDb\Statement\SimdjsonStreamHandlerFactory;
1718
use ArangoDb\Statement\StreamHandlerFactoryInterface;
1819
use ArangoDb\Type\Database;
1920
use ArangoDb\Http\ClientOptions;
@@ -84,7 +85,8 @@ public static function getStreamFactory(): StreamFactoryInterface
8485

8586
public static function getStreamHandlerFactory(): StreamHandlerFactoryInterface
8687
{
87-
return new ArrayStreamHandlerFactory();
88+
return new SimdjsonStreamHandlerFactory();
89+
// return new ArrayStreamHandlerFactory();
8890
}
8991

9092
public static function createDatabase(): void

0 commit comments

Comments
 (0)