Add queryStats and contInfo

Author: Илья

src/QueryResult.php

@@ -3,13 +3,25 @@
 namespace YdbPlatform\Ydb;
 
 use DateTime;
+use Ydb\CostInfo;
+use Ydb\TableStats\QueryStats;
 
 class QueryResult
 {
     protected $columns = [];
     protected $rows = [];
     protected $truncated = false;
 
+    /**
+     * @var CostInfo|null
+     */
+    protected $costInfo = null;
+
+    /**
+     * @var QueryStats|null
+     */
+    protected $queryStats = null;
+
     public function __construct($result)
     {
         if (method_exists($result, 'getResultSets'))
@@ -42,6 +54,14 @@ public function __construct($result)
         {
             throw new Exception('Unknown result');
         }
+
+        if(isset($result->costInfo)){
+            $this->costInfo = $result->costInfo;
+        }
+        if (method_exists($result, 'getQueryStats'))
+        {
+            $this->queryStats = $result->getQueryStats();
+        }
     }
 
     public function isTruncated()
@@ -233,4 +253,13 @@ protected function fillRows($rows)
             $this->rows[] = $_row;
         }
     }
+
+    /**
+     * @return CostInfo|null
+     */
+    public function costInfo()
+    {
+        return $this->costInfo;
+    }
+
 }

src/Traits/RequestTrait.php

@@ -2,6 +2,7 @@
 
 namespace YdbPlatform\Ydb\Traits;
 
+use Ydb\Operations\OperationParams;
 use Ydb\StatusIds\StatusCode;
 
 use YdbPlatform\Ydb\Issue;
@@ -52,6 +53,13 @@ trait RequestTrait
      */
     protected function doRequest($service, $method, array $data = [])
     {
+
+        $data['operation_params'] = new OperationParams([
+            'report_cost_info' => 1
+        ]);
+        if($method==='ExecuteDataQuery')
+        $data['collect_stats'] = 10;
+
         $this->checkDiscovery();
 
         $this->meta['x-ydb-auth-ticket'] = [$this->credentials->token()];
@@ -234,7 +242,7 @@ protected function processResponse($service, $method, $response, $resultClass)
             }
 
             $this->resetLastRequest();
-
+            $result->costInfo = $response->getCostInfo();
             return $result;
         }
         $statusName = StatusCode::name($statusCode);