| 152 | } |
| 153 | |
| 154 | ClientRequestState* QueryDriver::GetClientRequestState(const TUniqueId& query_id) { |
| 155 | lock_guard<SpinLock> l(client_request_state_lock_); |
| 156 | if (retried_client_request_state_ != nullptr |
| 157 | && retried_client_request_state_->query_id() == query_id) { |
| 158 | return retried_client_request_state_.get(); |
| 159 | } |
| 160 | DCHECK(client_request_state_ != nullptr); |
| 161 | // If cancel the retrying query the retried_client_request_state_ may not have been |
| 162 | // updated. In this case return nullptr |
| 163 | if (client_request_state_->query_id() != query_id) return nullptr; |
| 164 | return client_request_state_.get(); |
| 165 | } |
| 166 | |
| 167 | void QueryDriver::TryQueryRetry( |
| 168 | ClientRequestState* client_request_state, Status* error, bool* was_retried) { |
no test coverage detected