function_profiler.hpp

#pragma once
 
#include <atomic>
#include <chrono>
#include <shared_mutex>
#include <string>
#include <unordered_map>
#include <vector>
 
#include <google/protobuf/util/json_util.h>
#include <spdlog/spdlog.h>
 
#include "api/performance.pb.h"
namespace synapse {

class FunctionProfiler {
 public:
  explicit FunctionProfiler(const std::string& name);

  void start();

  void stop();

  synapse::FunctionProfile get_profile();
 
 private:
  std::string name_;

  std::atomic<uint64_t> call_count_{0};

  std::atomic<uint64_t> min_duration_ns_{std::numeric_limits<uint64_t>::max()};

  std::atomic<uint64_t> max_duration_ns_{0};

  std::atomic<uint64_t> latest_duration_ns_{0};

  std::atomic<uint64_t> total_duration_ns_{0};

  static constexpr size_t kNumSamples = 100;

  std::mutex samples_mutex_;

  std::vector<uint64_t> time_samples_;

  std::chrono::steady_clock::time_point start_time_;
};

class FunctionProfilerManager {
 public:
  FunctionProfilerManager() = default;

  void add(const std::string& name);

  void remove(const std::string& name);

  void start(const std::string& name);

  void stop(const std::string& name);

  synapse::FunctionProfile get(const std::string& name);

  std::vector<synapse::FunctionProfile> get_all();

  void print_profile(const std::string& name);
 
 private:
  std::shared_mutex profilers_mutex_;

  std::unordered_map<std::string, std::unique_ptr<FunctionProfiler>> profilers_;
};
}  // namespace synapse