Add reduce function to QueryBuilder
Signed-off-by: davidarendsen <davidarendsen@hey.com>
This commit is contained in:
parent
6014384afc
commit
bdc832aaab
3 changed files with 79 additions and 0 deletions
49
src/Function/Reduce.php
Normal file
49
src/Function/Reduce.php
Normal file
|
|
@ -0,0 +1,49 @@
|
||||||
|
<?php
|
||||||
|
|
||||||
|
namespace Arendsen\FluxQueryBuilder\Function;
|
||||||
|
|
||||||
|
class Reduce extends Base {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @var array $settings
|
||||||
|
*/
|
||||||
|
private $settings;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @var array $identity
|
||||||
|
*/
|
||||||
|
private $identity;
|
||||||
|
|
||||||
|
public function __construct(array $settings, array $identity)
|
||||||
|
{
|
||||||
|
$this->settings = $settings;
|
||||||
|
$this->identity = $identity;
|
||||||
|
}
|
||||||
|
|
||||||
|
public function __toString()
|
||||||
|
{
|
||||||
|
//reduce(fn: (r, accumulator) => ({sum: r._value + accumulator.sum}), identity: {sum: 0})
|
||||||
|
|
||||||
|
return '|> reduce(fn: (r, accumulator) => ({' . implode(', ', $this->formatSettings($this->settings)) . '}), ' .
|
||||||
|
'identity: {' . implode(', ', $this->format($this->identity)) . '}) ';
|
||||||
|
}
|
||||||
|
|
||||||
|
protected function formatSettings(array $settings)
|
||||||
|
{
|
||||||
|
array_walk($settings, function(&$value, $key) {
|
||||||
|
$value = $key . ': ' . $value;
|
||||||
|
});
|
||||||
|
|
||||||
|
return $settings;
|
||||||
|
}
|
||||||
|
|
||||||
|
protected function format(array $settings)
|
||||||
|
{
|
||||||
|
array_walk($settings, function(&$value, $key) {
|
||||||
|
$value = $key . ': ' . (is_string($value) ? '"' . $value . '"' : $value);
|
||||||
|
});
|
||||||
|
|
||||||
|
return $settings;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
@ -6,6 +6,7 @@ use Arendsen\FluxQueryBuilder\Expression\KeyValue;
|
||||||
use Arendsen\FluxQueryBuilder\Function\Filter;
|
use Arendsen\FluxQueryBuilder\Function\Filter;
|
||||||
use Arendsen\FluxQueryBuilder\Function\From;
|
use Arendsen\FluxQueryBuilder\Function\From;
|
||||||
use Arendsen\FluxQueryBuilder\Function\Range;
|
use Arendsen\FluxQueryBuilder\Function\Range;
|
||||||
|
use Arendsen\FluxQueryBuilder\Function\Reduce;
|
||||||
use Exception;
|
use Exception;
|
||||||
|
|
||||||
class QueryBuilder {
|
class QueryBuilder {
|
||||||
|
|
@ -13,10 +14,12 @@ class QueryBuilder {
|
||||||
const FLUX_PART_FROM = 'from';
|
const FLUX_PART_FROM = 'from';
|
||||||
const FLUX_PART_RANGE = 'range';
|
const FLUX_PART_RANGE = 'range';
|
||||||
const FLUX_PART_FILTERS = 'filters';
|
const FLUX_PART_FILTERS = 'filters';
|
||||||
|
const FLUX_PART_REDUCE = 'reduce';
|
||||||
|
|
||||||
const PARTS = [
|
const PARTS = [
|
||||||
self::FLUX_PART_FROM,
|
self::FLUX_PART_FROM,
|
||||||
self::FLUX_PART_RANGE,
|
self::FLUX_PART_RANGE,
|
||||||
|
self::FLUX_PART_REDUCE,
|
||||||
self::FLUX_PART_FILTERS
|
self::FLUX_PART_FILTERS
|
||||||
];
|
];
|
||||||
|
|
||||||
|
|
@ -91,6 +94,15 @@ class QueryBuilder {
|
||||||
return $this;
|
return $this;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public function addReduce(array $settings, array $identity): QueryBuilder
|
||||||
|
{
|
||||||
|
$this->addToQueryArray(
|
||||||
|
self::FLUX_PART_REDUCE,
|
||||||
|
new Reduce($settings, $identity)
|
||||||
|
);
|
||||||
|
return $this;
|
||||||
|
}
|
||||||
|
|
||||||
protected function addToQuery($key, $query)
|
protected function addToQuery($key, $query)
|
||||||
{
|
{
|
||||||
$this->fluxQueryParts[$key] = $query;
|
$this->fluxQueryParts[$key] = $query;
|
||||||
|
|
|
||||||
18
tests/Function/ReduceFunctionTest.php
Normal file
18
tests/Function/ReduceFunctionTest.php
Normal file
|
|
@ -0,0 +1,18 @@
|
||||||
|
<?php
|
||||||
|
declare(strict_types=1);
|
||||||
|
|
||||||
|
use Arendsen\FluxQueryBuilder\Function\Reduce;
|
||||||
|
use PHPUnit\Framework\TestCase;
|
||||||
|
|
||||||
|
final class ReduceFunctionTest extends TestCase {
|
||||||
|
|
||||||
|
public function testSimpleReduce()
|
||||||
|
{
|
||||||
|
$expression = new Reduce(['sum' => 'r._value + accumulator.sum'], ['sum' => 0]);
|
||||||
|
|
||||||
|
$query = '|> reduce(fn: (r, accumulator) => ({sum: r._value + accumulator.sum}), identity: {sum: 0}) ';
|
||||||
|
|
||||||
|
$this->assertEquals($query, $expression->__toString());
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
Loading…
Add table
editor.link_modal.header
Reference in a new issue