| 1056 | } |
| 1057 | |
| 1058 | void AdmissionController::UpdateStatsOnAdmission( |
| 1059 | const ScheduleState& state, bool is_trivial, PerUserTracking& per_user_tracking) { |
| 1060 | for (const auto& entry : state.per_backend_schedule_states()) { |
| 1061 | const NetworkAddressPB& host_addr = entry.first; |
| 1062 | int64_t mem_to_admit = GetMemToAdmit(state, entry.second); |
| 1063 | UpdateHostStats(host_addr, mem_to_admit, 1, entry.second.exec_params->slots_to_use()); |
| 1064 | } |
| 1065 | PoolStats* pool_stats = GetPoolStats(state); |
| 1066 | pool_stats->AdmitQueryAndMemory(state, is_trivial, per_user_tracking); |
| 1067 | pools_for_updates_.insert(state.request_pool()); |
| 1068 | } |
| 1069 | |
| 1070 | void AdmissionController::UpdateHostStats(const NetworkAddressPB& host_addr, |
| 1071 | int64_t mem_to_admit, int num_queries_to_admit, int num_slots_to_admit) { |
nothing calls this directly
no test coverage detected