Bigtable: Read modify write row model

Bigtable/src/DataClient.php

@@ -21,7 +21,9 @@
 use Google\ApiCore\Serializer;
 use Google\Cloud\Bigtable\Exception\BigtableDataOperationException;
 use Google\Cloud\Bigtable\Filter\FilterInterface;
+use Google\Cloud\Bigtable\ReadModifyWriteRowRules;
 use Google\Cloud\Bigtable\V2\BigtableClient as TableClient;
+use Google\Cloud\Bigtable\V2\Row;
 use Google\Cloud\Bigtable\V2\RowRange;
 use Google\Cloud\Bigtable\V2\RowSet;
 use Google\Cloud\Core\ArrayTrait;
@@ -322,4 +324,90 @@ public function readRow($rowKey, array $options = [])
             ->readAll()
             ->current();
     }
+
+    /**
+     * Modifies a row atomically on the server. The method reads the latest
+     * existing timestamp and value from the specified columns and writes a new
+     * entry based on pre-defined read/modify/write rules. The new value for the
+     * timestamp is the greater of the existing timestamp or the current server
+     * time. The method returns the new contents of all modified cells.
+     *
+     * Example:
+     * ```
+     * use Google\Cloud\Bigtable\ReadModifyWriteRowRules;
+     *
+     * $rules = (new ReadModifyWriteRowRules)->append('cf1', 'cq1', 'value12');
+     * $row = $dataClient->readModifyWriteRow('rk1', $rules);
+     *
+     * print_r($row);
+     * ```
+     *
+     * //Increments value
+     * ```
+     * use Google\Cloud\Bigtable\DataUtil;
+     * use Google\Cloud\Bigtable\ReadModifyWriteRowRules;
+     * use Google\Cloud\Bigtable\RowMutation;
+     *
+     * $rowMutation = new RowMutation('rk1');
+     * $rowMutation->upsert('cf1','cq1',DataUtil::intToByteString(2),5000);
+     *
+     * $dataClient->mutateRows([$rowMutation]);
+     *
+     * $rules = (new ReadModifyWriteRowRules)->increment('cf1', 'cq1', 3);
+     * $row = $dataClient->readModifyWriteRow('rk1', $rules);
+     *
+     * print_r($row);
+     * ```
+     *
+     * @param string $rowKey The row key to read.
+     * @param ReadModifyWriteRowRules $rules Rules to apply on row.
+     * @param array $options [optional] Configuration options.
+     * @return array Returns array containing all column family keyed by family name.
+     * @throws ApiException if the remote call fails or operation fails
+     */
+    public function readModifyWriteRow($rowKey, $rules, array $options = [])
+    {
+        if (!$rules instanceof ReadModifyWriteRowRules) {
+            throw new \InvalidArgumentException(
+                sprintf(
+                    'Expected rules to be of type \'%s\', instead got \'%s\'.',
+                    ReadModifyWriteRowRules::class,
+                    gettype($rules)
+                )
+            );
+        }
+        $readModifyWriteRowResponse = $this->bigtableClient->readModifyWriteRow(
+            $this->tableName,
+            $rowKey,
+            $rules->toProto(),
+            $options + $this->options
+        );
+        return $this->convertToArray($readModifyWriteRowResponse->getRow());
+    }
+
+    private function convertToArray(Row $row)
+    {
+        if ($row === null) {
+            return [];
+        }
+        $families = [];
+        foreach ($row->getFamilies() as $family) {
+            $qualifiers = [];
+            foreach ($family->getColumns() as $column) {
+                $values = [];
+                foreach ($column->getCells() as $cell) {
+                    $values[] = [
+                        'value' => $cell->getValue(),
+                        'timeStamp' => $cell->getTimestampMicros(),
+                        'labels' => ($cell->getLabels()->getIterator()->valid())
+                            ? implode(iterator_to_array($cell->getLabels()->getIterator()))
+                            : ''
+                    ];
+                }
+                $qualifiers[$column->getQualifier()] = $values;
+            }
+            $families[$family->getName()] = $qualifiers;
+        }
+        return $families;
+    }
 }

Bigtable/src/DataUtil.php

@@ -0,0 +1,59 @@
+<?php
+/**
+ * Copyright 2018, Google LLC All rights reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+namespace Google\Cloud\Bigtable;
+
+/**
+ * This class contains utility to convert integer to byte string and backward.
+ */
+class DataUtil
+{
+    /**
+     * Utility method to convert integer value in to 64 bit signed BigEndian
+     * representation.
+     *
+     * @param string|int $intValue Integer value to convert to.
+     * @return string Returns string of bytes representing 64 bit big signed BigEndian.
+     * @throws \InvalidArgumentException If value is not numberic.
+     */
+    public static function intToByteString($intValue)
+    {
+        if (!is_numeric($intValue)) {
+            throw new \InvalidArgumentException(
+                sprintf(
+                    'Expected argument to be of type int, instead got \'%s\'.',
+                    gettype($intValue)
+                )
+            );
+        }
+        $hex = base_convert($intValue, 10, 16);
+        $hex = str_pad($hex, 16, '0', STR_PAD_LEFT);
+        return hex2bin($hex);
+    }
+
+    /**
+     * Convertes string bytes of 64 bit signed BigEndian to int.
+     *
+     * @param string|array $bytes String of bytes to convert.
+     * @return int Integer value of the string bytes.
+     */
+    public static function byteStringToInt($bytes)
+    {
+        $hex = bin2hex($bytes);
+        return hexdec($hex);
+    }
+}

Bigtable/src/ReadModifyWriteRowRules.php

@@ -0,0 +1,91 @@
+<?php
+/**
+ * Copyright 2018, Google LLC All rights reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+namespace Google\Cloud\Bigtable;
+
+use Google\Cloud\Bigtable\V2\ReadModifyWriteRule;
+
+/**
+ * Represents collection of ReadModifyWriteRule to perform operation on Bigtable table.
+ * This is used in readModifyWriteRow operation on row in Bigtable table.
+ */
+class ReadModifyWriteRowRules
+{
+    /**
+     * @var array
+     */
+    private $options;
+
+    /**
+     * @var array ReadModifyWriteRule
+     */
+    private $rules = [];
+
+    public function __construct(array $options = [])
+    {
+        $this->options = $options;
+    }
+
+    /**
+     * Appends the value to the existing value of the cell. If targeted cell is unset,
+     * it will be treated as containing the empty string.
+     *
+     * @param string $familyName Family name of the row.
+     * @param string $qualifier Column qualifier of the row.
+     * @param string $value Value of the Column qualifier.
+     *
+     * @return ReadModifyWriteRowRules returns current ReadModifyWriteRowRules object.
+     */
+    public function append($familyName, $qualifier, $value)
+    {
+        $this->rules[] = (new ReadModifyWriteRule)
+            ->setFamilyName($familyName)
+            ->setColumnQualifier($qualifier)
+            ->setAppendValue($value);
+        return $this;
+    }
+
+    /**
+     * Adds `amount` to the existing value. If the targeted cell is unset, it will be treated
+     * as containing a zero. Otherwise, the targeted cell must containt an 8-byte value (interpreted
+     * as a 64-bit big-endian signed integer), or the entire request will fail.
+     *
+     * @param string $familyName Family name of the row.
+     * @param string $qualifier Column qualifier of the row.
+     * @param int $amount Amount to add to value of Column qualifier.
+     *
+     * @return ReadModifyWriteRowRules returns current ReadModifyWriteRowRules object.
+     */
+    public function increment($familyName, $qualifier, $amount)
+    {
+        $this->rules[] = (new ReadModifyWriteRule)
+            ->setFamilyName($familyName)
+            ->setColumnQualifier($qualifier)
+            ->setIncrementAmount($amount);
+        return $this;
+    }
+
+    /**
+     * Returns proto representation of ReadModifyWriteRule.
+     *
+     * @return array ReadModifyWriteRule Returns array of ReadModifyWriteRule rules.
+     */
+    public function toProto()
+    {
+        return $this->rules;
+    }
+}