postgres_control.hpp

#pragma once
 
#include <chrono>
 
#include <userver/engine/deadline.hpp>
 
USERVER_NAMESPACE_BEGIN
 
namespace testsuite {
 
class PostgresControl {
 public:
  enum class ReadonlyMaster { kNotExpected, kExpected };
 
  PostgresControl() = default;
 
  PostgresControl(std::chrono::milliseconds execute_timeout,
                  std::chrono::milliseconds statement_timeout,
                  ReadonlyMaster readonly_master);
 
  [[nodiscard]] engine::Deadline MakeExecuteDeadline(
      std::chrono::milliseconds duration) const;
 
  [[nodiscard]] std::chrono::milliseconds MakeStatementTimeout(
      std::chrono::milliseconds duration) const;
 
  bool IsReadonlyMasterExpected() const;
 
 private:
  std::chrono::milliseconds execute_timeout_{};
  std::chrono::milliseconds statement_timeout_{};
  bool is_readonly_master_expected_{false};
};
 
}  // namespace testsuite
 
USERVER_NAMESPACE_END