tcp_telemetry.h

Go to the documentation of this file.

// Copyright (c) Space-ng, inc. All rights reserved.
 
#pragma once
 
#include "sol3/cpp/thread_safe_value.h"
#include "sol3/net/msg/tcp_telemetry.h"
 
#include <cstddef>
 
namespace sol3::net {
 
using sol3::cpp::ThreadSafeValue;
 
// Tracks counters and state for TCP clients and servers.
class TcpTelemetry {
 public:
  void onBytesRead(size_t n) {
    telemetry_.mutate([n](auto& t) { t.bytes_read += n; });
  }
  void onBytesWritten(size_t n) {
    telemetry_.mutate([n](auto& t) {
      t.bytes_written += n;
      t.write_queue_length -= n;
    });
  }
  void onBytesDropped(size_t n) {
    telemetry_.mutate([n](auto& t) {
      t.bytes_dropped += n;
      t.write_queue_length -= n;
    });
  }
  void onBytesQueued(size_t n) {
    telemetry_.mutate([n](auto& t) { t.write_queue_length += n; });
  }
  void onKernelQueueSampled(int32_t n) {
    telemetry_.mutate([n](auto& t) { t.kernel_send_queue_length = n; });
  }
  void onFramesRead(size_t n) {
    telemetry_.mutate([n](auto& t) { t.frames_read += n; });
  }
  void onFramesWritten(size_t n) {
    telemetry_.mutate([n](auto& t) { t.frames_written += n; });
  }
  void onConnOpened() {
    telemetry_.mutate([](auto& t) {
      t.connections_opened++;
      t.active_connections++;
    });
  }
  void onConnClosed() {
    telemetry_.mutate([](auto& t) {
      t.connections_closed++;
      t.active_connections--;
      t.kernel_send_queue_length = -1;
    });
  }
 
  void onConnError() {
    telemetry_.mutate([](auto& t) { t.connection_errors++; });
  }
 
  msg::TcpTelemetryT snapshot() {
    return telemetry_.access([](auto& t) { return t; });
  }
 
 private:
  ThreadSafeValue<msg::TcpTelemetryT> telemetry_;
};
 
}  // namespace sol3::net