| 52 | } |
| 53 | |
| 54 | Status InitialReservations::Init( |
| 55 | const TUniqueId& query_id, int64_t query_min_reservation) { |
| 56 | DCHECK_EQ(0, initial_reservations_.GetReservation()) << "Already inited"; |
| 57 | Status reservation_status; |
| 58 | if (!initial_reservations_.IncreaseReservation( |
| 59 | query_min_reservation, &reservation_status)) { |
| 60 | return Status(TErrorCode::MINIMUM_RESERVATION_UNAVAILABLE, |
| 61 | PrettyPrinter::Print(query_min_reservation, TUnit::BYTES), FLAGS_hostname, |
| 62 | FLAGS_krpc_port, PrintId(query_id), reservation_status.GetDetail()); |
| 63 | } |
| 64 | VLOG(2) << "Successfully claimed initial reservations (" |
| 65 | << PrettyPrinter::Print(query_min_reservation, TUnit::BYTES) << ") for" |
| 66 | << " query " << PrintId(query_id); |
| 67 | return Status::OK(); |
| 68 | } |
| 69 | |
| 70 | void InitialReservations::Claim(BufferPool::ClientHandle* dst, int64_t bytes) { |
| 71 | DCHECK_GE(bytes, 0); |
nothing calls this directly
no test coverage detected