|
2 | 2 |
|
3 | 3 | namespace YdbPlatform\Ydb\Traits; |
4 | 4 |
|
5 | | -use Ydb\FeatureFlag\Status; |
6 | | -use Ydb\Operations\OperationParams; |
7 | 5 | use Ydb\StatusIds\StatusCode; |
8 | 6 |
|
9 | | -use Ydb\Table\QueryStatsCollection\Mode; |
10 | | -use Ydb\TableStats\QueryStats; |
| 7 | +use YdbPlatform\Ydb\Enums\CollectQueryStatsMode; |
11 | 8 | use YdbPlatform\Ydb\Issue; |
12 | 9 | use YdbPlatform\Ydb\Exception; |
13 | 10 | use YdbPlatform\Ydb\QueryResult; |
@@ -45,9 +42,8 @@ trait RequestTrait |
45 | 42 | */ |
46 | 43 | protected $lastDiscovery = 0; |
47 | 44 |
|
48 | | - protected $collectStats = Mode::STATS_COLLECTION_UNSPECIFIED; |
| 45 | + protected $collectStats = CollectQueryStatsMode::STATS_COLLECTION_UNSPECIFIED; |
49 | 46 |
|
50 | | - protected $reportCostInfo = Status::STATUS_UNSPECIFIED; |
51 | 47 |
|
52 | 48 | /** |
53 | 49 | * Make a request to the service with the given method. |
@@ -92,10 +88,6 @@ protected function doRequest($service, $method, array $data = []) |
92 | 88 | $resultClass = '\\Ydb\\' . $service . '\\' . $method . 'Result'; |
93 | 89 | } |
94 | 90 |
|
95 | | - $data['operation_params'] = new OperationParams([ |
96 | | - 'report_cost_info' => $this->reportCostInfo |
97 | | - ]); |
98 | | - |
99 | 91 | $request = new $requestClass($data); |
100 | 92 |
|
101 | 93 | if (method_exists($request, 'setCollectStats')){ |
@@ -251,7 +243,6 @@ protected function processResponse($service, $method, $response, $resultClass) |
251 | 243 | } |
252 | 244 |
|
253 | 245 | $this->resetLastRequest(); |
254 | | - $result->costInfo = $response->getCostInfo(); |
255 | 246 | return $result; |
256 | 247 | } |
257 | 248 | $statusName = StatusCode::name($statusCode); |
@@ -393,19 +384,11 @@ protected function checkDiscovery(){ |
393 | 384 | ]; |
394 | 385 |
|
395 | 386 | /** |
396 | | - * @param int $collectStats \Ydb\Table\QueryStatsCollection\Mode |
| 387 | + * @param int $collectStats YdbPlatform\Ydb\Enums\CollectQueryStatsMode |
397 | 388 | */ |
398 | 389 | public function setCollectStats(int $collectStats): void |
399 | 390 | { |
400 | 391 | $this->collectStats = $collectStats; |
401 | 392 | } |
402 | 393 |
|
403 | | - /** |
404 | | - * @param int $reportCostInfo Ydb.FeatureFlag.Status |
405 | | - */ |
406 | | - public function setReportCostInfo(int $reportCostInfo): void |
407 | | - { |
408 | | - $this->reportCostInfo = $reportCostInfo; |
409 | | - } |
410 | | - |
411 | 394 | } |
0 commit comments