src/core/ext/filters/client_channel/lb_policy/grpclb/grpclb_client_stats.cc

   1 /*
   2  *
   3  * Copyright 2017 gRPC authors.
   4  *
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *     http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  *
  17  */
  18
  19 #include <grpc/support/port_platform.h>
  20
  21 #include "src/core/ext/filters/client_channel/lb_policy/grpclb/grpclb_client_stats.h"
  22
  23 #include <string.h>
  24
  25 #include <grpc/support/atm.h>
  26 #include <grpc/support/string_util.h>
  27
  28 namespace grpc_core {
  29
  30 void GrpcLbClientStats::AddCallStarted() {
  31   gpr_atm_full_fetch_add(&num_calls_started_, (gpr_atm)1);
  32 }
  33
  34 void GrpcLbClientStats::AddCallFinished(
  35     bool finished_with_client_failed_to_send, bool finished_known_received) {
  36   gpr_atm_full_fetch_add(&num_calls_finished_, (gpr_atm)1);
  37   if (finished_with_client_failed_to_send) {
  38     gpr_atm_full_fetch_add(&num_calls_finished_with_client_failed_to_send_,
  39                            (gpr_atm)1);
  40   }
  41   if (finished_known_received) {
  42     gpr_atm_full_fetch_add(&num_calls_finished_known_received_, (gpr_atm)1);
  43   }
  44 }
  45
  46 void GrpcLbClientStats::AddCallDroppedLocked(const char* token) {
  47   // Increment num_calls_started and num_calls_finished.
  48   gpr_atm_full_fetch_add(&num_calls_started_, (gpr_atm)1);
  49   gpr_atm_full_fetch_add(&num_calls_finished_, (gpr_atm)1);
  50   // Record the drop.
  51   if (drop_token_counts_ == nullptr) {
  52     drop_token_counts_.reset(New<DroppedCallCounts>());
  53   }
  54   for (size_t i = 0; i < drop_token_counts_->size(); ++i) {
  55     if (strcmp((*drop_token_counts_)[i].token.get(), token) == 0) {
  56       ++(*drop_token_counts_)[i].count;
  57       return;
  58     }
  59   }
  60   // Not found, so add a new entry.
  61   drop_token_counts_->emplace_back(UniquePtr<char>(gpr_strdup(token)), 1);
  62 }
  63
  64 namespace {
  65
  66 void AtomicGetAndResetCounter(int64_t* value, gpr_atm* counter) {
  67   *value = static_cast<int64_t>(gpr_atm_full_xchg(counter, (gpr_atm)0));
  68 }
  69
  70 }  // namespace
  71
  72 void GrpcLbClientStats::GetLocked(
  73     int64_t* num_calls_started, int64_t* num_calls_finished,
  74     int64_t* num_calls_finished_with_client_failed_to_send,
  75     int64_t* num_calls_finished_known_received,
  76     UniquePtr<DroppedCallCounts>* drop_token_counts) {
  77   AtomicGetAndResetCounter(num_calls_started, &num_calls_started_);
  78   AtomicGetAndResetCounter(num_calls_finished, &num_calls_finished_);
  79   AtomicGetAndResetCounter(num_calls_finished_with_client_failed_to_send,
  80                            &num_calls_finished_with_client_failed_to_send_);
  81   AtomicGetAndResetCounter(num_calls_finished_known_received,
  82                            &num_calls_finished_known_received_);
  83   *drop_token_counts = std::move(drop_token_counts_);
  84 }
  85
  86 }  // namespace grpc_core