cpu_relax.hpp

Go to the documentation of this file.

#pragma once
 
 
#include <chrono>
#include <cstddef>
 
#include <userver/tracing/scope_time.hpp>
 
USERVER_NAMESPACE_BEGIN
 
namespace utils {
 
class ScopeTimePause {
 public:
  explicit ScopeTimePause(tracing::ScopeTime* scope);
 
  void Pause();
 
  void Unpause();
 
 private:
  tracing::ScopeTime* scope_;
  std::string scope_name_;
};
 
class CpuRelax {
 public:
  explicit CpuRelax(std::size_t every, tracing::ScopeTime* scope);
 
  CpuRelax(const CpuRelax&) = delete;
  CpuRelax(CpuRelax&&) = delete;
 
  void Relax();
 
 private:
  ScopeTimePause pause_;
  const std::size_t every_iterations_;
  std::size_t iterations_{0};
};
 
class StreamingCpuRelax {
 public:
  explicit StreamingCpuRelax(std::uint64_t check_time_after_bytes,
                             tracing::ScopeTime* scope);
 
  void Relax(std::uint64_t bytes_processed);
 
  std::uint64_t GetBytesProcessed() const;
 
 private:
  ScopeTimePause pause_;
  std::uint64_t check_time_after_bytes_;
  std::uint64_t total_bytes_{0};
  std::uint64_t bytes_since_last_time_check_{0};
  std::chrono::steady_clock::time_point last_yield_time_;
};
 
}  // namespace utils
 
USERVER_NAMESPACE_END