Add reduce function to QueryBuilder

Signed-off-by: davidarendsen <davidarendsen@hey.com>
This commit is contained in:
davidarendsen 2022-08-08 15:37:04 +00:00
commit bdc832aaab
3 changed files with 79 additions and 0 deletions

49
src/Function/Reduce.php Normal file
View 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;
}
}

View file

@ -6,6 +6,7 @@ use Arendsen\FluxQueryBuilder\Expression\KeyValue;
use Arendsen\FluxQueryBuilder\Function\Filter;
use Arendsen\FluxQueryBuilder\Function\From;
use Arendsen\FluxQueryBuilder\Function\Range;
use Arendsen\FluxQueryBuilder\Function\Reduce;
use Exception;
class QueryBuilder {
@ -13,10 +14,12 @@ class QueryBuilder {
const FLUX_PART_FROM = 'from';
const FLUX_PART_RANGE = 'range';
const FLUX_PART_FILTERS = 'filters';
const FLUX_PART_REDUCE = 'reduce';
const PARTS = [
self::FLUX_PART_FROM,
self::FLUX_PART_RANGE,
self::FLUX_PART_REDUCE,
self::FLUX_PART_FILTERS
];
@ -91,6 +94,15 @@ class QueryBuilder {
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)
{
$this->fluxQueryParts[$key] = $query;