Added query timeout and canceled

Added query timeout and canceled

Author: ilyakharev

CHANGELOG.md

@@ -1,3 +1,5 @@
+* added query timeout and canceled params
+
 ## 1.11.0
 * added query stats
 * added ReadTokenFromFile

src/Session.php

@@ -4,6 +4,8 @@
 
 use Closure;
 use Exception;
+use Google\Protobuf\Duration;
+use Ydb\Operations\OperationParams;
 use Ydb\Table\Query;
 use Ydb\Table\QueryCachePolicy;
 // use Ydb\Table\StaleModeSettings;
@@ -366,6 +368,24 @@ public function query($yql, array $parameters = null, array $options = [])
         if(isset($options['collectStats'])){
             $query->collectStats($options['collectStats']);
         }
+        $operationParams = new OperationParams();
+        if(isset($options['operation_timeout_ms'])){
+            $seconds = intdiv( $options['operation_timeout_ms'], 1000); // get seconds
+            $nanos = fmod($options['operation_timeout_ms'], 1000) * 1000000; // get ns
+            $operationParams->setOperationTimeout(new Duration([
+                'seconds'   => $seconds,
+                'nanos'     => $nanos
+            ]));
+        }
+        if(isset($options['cancel_after_ms'])){
+            $seconds = intdiv( $options['cancel_after_ms'], 1000); // get seconds
+            $nanos = fmod($options['operation_timeout_ms'], 1000) * 1000000; // get ns
+            $operationParams->setCancelAfter(new Duration([
+                'seconds'   => $seconds,
+                'nanos'     => $nanos
+            ]));
+        }
+        $query->operationParams($operationParams);
 
         return $this->executeQuery($query);
     }

tests/CheckTimeoutAndDurationParamsTest.php

@@ -0,0 +1,49 @@
+<?php
+
+namespace YdbPlatform\Ydb\Test;
+
+use PHPUnit\Framework\TestCase;
+use YdbPlatform\Ydb\Auth\Implement\AnonymousAuthentication;
+use YdbPlatform\Ydb\Logger\SimpleStdLogger;
+use YdbPlatform\Ydb\Session;
+use YdbPlatform\Ydb\Ydb;
+
+class CheckTimeoutAndDurationParamsTest extends TestCase
+{
+    public function testTimeoutAndDurationParams(){
+        $config = [
+
+            // Database path
+            'database'    => '/local',
+
+            // Database endpoint
+            'endpoint'    => 'localhost:2136',
+
+            // Auto discovery (dedicated server only)
+            'discovery'   => false,
+
+            // IAM config
+            'iam_config'  => [
+                'insecure' => true,
+            ],
+            'credentials' => new AnonymousAuthentication()
+        ];
+
+        $ydb = new Ydb($config, new SimpleStdLogger(7));
+        $table = $ydb->table();
+
+        $this->expectException('YdbPlatform\Ydb\Exceptions\Ydb\TimeoutException');
+        $table->retrySession(function (Session $session){
+            $session->query('SELECT 1;', null, [
+                'operation_timeout_ms' => 1e-5
+            ]);
+        });
+
+        $this->expectException('YdbPlatform\Ydb\Exceptions\Ydb\CancelledException');
+        $table->retrySession(function (Session $session){
+            $session->query('SELECT 1;', null, [
+                'cancel_after_ms' => 1e-5
+            ]);
+        });
+    }
+}