Remaining features for 2.3.0 release.

Author: danielmewes

pb4php/ql2.proto

@@ -367,6 +367,8 @@ message Term {
                         // The arity of the function should be
                         // Sequence..., Function(sizeof...(Sequence)) -> Sequence
 
+        FOLD = 187; // Sequence, Datum, Function(2), {Function(3), Function(1)
+
         // Filter a sequence with either a function or a shortcut
         // object (see API docs for details).  The body of FILTER is
         // wrapped in an implicit `.default(false)`, and you can

rdb/DatumConverter.php

@@ -188,7 +188,7 @@ public function nativeToFunction($f)
             if (!isset($result)) {
                 // In case of null, assume that the user forgot to add a return.
                 // If null is the intended value, r\expr() should be wrapped around the return value.
-                throw new RqlDriverError("The function did not evaluate to a query (missing return?).");
+                throw new RqlDriverError("The function did not evaluate to a value (missing return?). If the function is intended to return `null,` please use `return r\expr(null);`.");
             } else {
                 $result = $this->nativeToDatum($result);
             }
@@ -197,7 +197,7 @@ public function nativeToFunction($f)
         return new RFunction($args, $result);
     }
 
-    public function nativeToDatumOrFunction($f)
+    public function nativeToDatumOrFunction($f, $wrapImplicit = true)
     {
         if (!(is_object($f) && is_subclass_of($f, '\r\Query'))) {
             try {
@@ -210,6 +210,10 @@ public function nativeToDatumOrFunction($f)
                 $f = $this->nativeToFunction($f);
             }
         }
-        return $this->wrapImplicitVar($f);
+        if ($wrapImplicit) {
+            return $this->wrapImplicitVar($f);
+        } else {
+            return $f;
+        }
     }
 }

rdb/ProtocolBuffer/TermTermType.php

@@ -54,6 +54,7 @@ class TermTermType
     const PB_FILL                = 167;
     const PB_FILTER              = 39;
     const PB_FLOOR               = 183;
+    const PB_FOLD                = 187;
     const PB_FOR_EACH            = 68;
     const PB_FRIDAY              = 111;
     const PB_FUNC                = 69;

rdb/Queries/Aggregations/Fold.php

@@ -0,0 +1,27 @@
+<?php
+
+namespace r\Queries\Aggregations;
+
+use r\ValuedQuery\ValuedQuery;
+use r\ProtocolBuffer\TermTermType;
+
+class Fold extends ValuedQuery
+{
+    public function __construct(ValuedQuery $sequence, $base, $fun, $opts = null)
+    {
+        $this->setPositionalArg(0, $sequence);
+        $this->setPositionalArg(1, $this->nativeToDatum($base));
+        $this->setPositionalArg(2, $this->nativeToFunction($fun));
+
+        if (isset($opts)) {
+            foreach ($opts as $opt => $val) {
+                $this->setOptionalArg($opt, $this->nativeToDatumOrFunction($val));
+            }
+        }
+    }
+
+    protected function getTermType()
+    {
+        return TermTermType::PB_FOLD;
+    }
+}

rdb/Queries/Control/Args.php

@@ -0,0 +1,19 @@
+<?php
+
+namespace r\Queries\Control;
+
+use r\ValuedQuery\ValuedQuery;
+use r\ProtocolBuffer\TermTermType;
+
+class Args extends ValuedQuery
+{
+    public function __construct($args)
+    {
+        $this->setPositionalArg(0, $this->nativeToDatum($args));
+    }
+
+    protected function getTermType()
+    {
+        return TermTermType::PB_ARGS;
+    }
+}

rdb/Queries/Control/Branch.php

@@ -10,8 +10,8 @@ class Branch extends ValuedQuery
 {
     public function __construct(Query $test, $trueBranch, $falseBranch)
     {
-        $trueBranch = $this->nativeToDatumOrFunction($trueBranch);
-        $falseBranch = $this->nativeToDatumOrFunction($falseBranch);
+        $trueBranch = $this->nativeToDatumOrFunction($trueBranch, false);
+        $falseBranch = $this->nativeToDatumOrFunction($falseBranch, false);
 
         $this->setPositionalArg(0, $test);
         $this->setPositionalArg(1, $trueBranch);

rdb/Queries/Transformations/Union.php

@@ -7,10 +7,15 @@
 
 class Union extends ValuedQuery
 {
-    public function __construct(ValuedQuery $sequence, ValuedQuery $otherSequence)
+    public function __construct(ValuedQuery $sequence, ValuedQuery $otherSequence, $opts = null)
     {
         $this->setPositionalArg(0, $sequence);
         $this->setPositionalArg(1, $otherSequence);
+        if (isset($opts)) {
+            foreach ($opts as $opt => $val) {
+                $this->setOptionalArg($opt, $this->nativeToDatum($val));
+            }
+        }
     }
 
     protected function getTermType()

rdb/Queries/Writing/Insert.php

@@ -28,7 +28,7 @@ public function __construct(Table $table, $document, $opts = null)
         $this->setPositionalArg(1, $document);
         if (isset($opts)) {
             foreach ($opts as $opt => $val) {
-                $this->setOptionalArg($opt, $this->nativeToDatum($val));
+                $this->setOptionalArg($opt, $this->nativeToDatumOrFunction($val));
             }
         }
     }

rdb/ValuedQuery/ValuedQuery.php

@@ -7,6 +7,7 @@
 use r\Queries\Aggregations\Contains;
 use r\Queries\Aggregations\Count;
 use r\Queries\Aggregations\Distinct;
+use r\Queries\Aggregations\Fold;
 use r\Queries\Aggregations\Group;
 use r\Queries\Aggregations\Max;
 use r\Queries\Aggregations\Min;
@@ -189,9 +190,9 @@ public function isEmpty()
     {
         return new IsEmpty($this);
     }
-    public function union(ValuedQuery $otherSequence)
+    public function union(ValuedQuery $otherSequence, $opts = null)
     {
-        return new Union($this, $otherSequence);
+        return new Union($this, $otherSequence, $opts);
     }
     public function sample($n)
     {
@@ -201,6 +202,10 @@ public function reduce($reductionFunction)
     {
         return new Reduce($this, $reductionFunction);
     }
+    public function fold($base, $fun, $opts = null)
+    {
+        return new Fold($this, $base, $fun, $opts);
+    }
     public function count($filter = null)
     {
         return new Count($this, $filter);

rdb/global.php

@@ -6,6 +6,7 @@
 use r\Exceptions\RqlDriverError;
 use r\Ordering\Asc;
 use r\Ordering\Desc;
+use r\Queries\Control\Args;
 use r\Queries\Control\Branch;
 use r\Queries\Control\Error;
 use r\Queries\Control\Http;
@@ -72,6 +73,7 @@
 use r\Queries\Tables\TableDrop;
 use r\Queries\Tables\TableList;
 use r\Queries\Transformations\MapMultiple;
+use r\Queries\Transformations\Union;
 use r\ValuedQuery\ImplicitVar;
 use r\Queries\Control\Js;
 use r\ValuedQuery\Json;
@@ -128,6 +130,11 @@ function rDo($args, $inExpr)
     return new RDo($args, $inExpr);
 }
 
+function args($args)
+{
+    return new Args($args);
+}
+
 function branch(Query $test, $trueBranch, $falseBranch)
 {
     return new Branch($test, $trueBranch, $falseBranch);
@@ -439,6 +446,11 @@ function mapMultiple($sequences, $mappingFunction)
     return new MapMultiple($sequences[0], array_slice($sequences, 1), $mappingFunction);
 }
 
+function union($sequence, $otherSequence, $opts = null)
+{
+    return new Union($sequence, $otherSequence, $opts);
+}
+
 function ceil($value)
 {
     return new Ceil($value);