coordination.hpp

#pragma once
 
#include <memory>
#include <string_view>
 
#include <ydb-cpp-sdk/client/coordination/coordination.h>
 
USERVER_NAMESPACE_BEGIN
 
namespace ydb {
 
namespace impl {
class Driver;
}  // namespace impl
 
class CoordinationSession final {
 public:
  /// @cond
  // For internal use only.
  explicit CoordinationSession(NYdb::NCoordination::TSession&& session);
  /// @endcond
 
  std::uint64_t GetSessionId();
 
  NYdb::NCoordination::ESessionState GetSessionState();
 
  NYdb::NCoordination::EConnectionState GetConnectionState();
 
  void Close();
 
  void Ping();
 
  void Reconnect();
 
  /// @warning Use `TAcquireSemaphoreSettings::OnAccepted` callback with care,
  /// it will be executed on a non-coroutine thread
  bool AcquireSemaphore(
      std::string_view name,
      const NYdb::NCoordination::TAcquireSemaphoreSettings& settings);
 
  bool ReleaseSemaphore(std::string_view name);
 
  /// @warning Use `TDescribeSemaphoreSettings::OnChanged` callback with care,
  /// it will be executed on a non-coroutine thread
  NYdb::NCoordination::TSemaphoreDescription DescribeSemaphore(
      std::string_view name,
      const NYdb::NCoordination::TDescribeSemaphoreSettings& settings);
 
  void CreateSemaphore(std::string_view name, std::uint64_t limit);
 
  void UpdateSemaphore(std::string_view name, std::string_view data);
 
  void DeleteSemaphore(std::string_view name);
 
 private:
  NYdb::NCoordination::TSession session_;
};
 
class CoordinationClient final {
 public:
  /// @cond
  // For internal use only.
  explicit CoordinationClient(std::shared_ptr<impl::Driver> driver);
  /// @endcond
 
  /// @warning Use `TSessionSettings::OnStateChanged` and
  /// `TSessionSettings::OnStopped` callbacks with care, they will be executed
  /// on a non-coroutine thread
  CoordinationSession StartSession(
      std::string_view path,
      const NYdb::NCoordination::TSessionSettings& settings);
 
  void CreateNode(std::string_view path,
                  const NYdb::NCoordination::TCreateNodeSettings& settings);
 
  void AlterNode(std::string_view path,
                 const NYdb::NCoordination::TAlterNodeSettings& settings);
 
  void DropNode(std::string_view path);
 
  NYdb::NCoordination::TNodeDescription DescribeNode(std::string_view path);
 
  /// Get native coordination client
  /// @warning Use with care! Facilities from
  /// `<core/include/userver/drivers/subscribable_futures.hpp>` can help with
  /// non-blocking wait operations.
  NYdb::NCoordination::TClient& GetNativeCoordinationClient();
 
 private:
  std::shared_ptr<impl::Driver> driver_;
  NYdb::NCoordination::TClient client_;
};
 
}  // namespace ydb
 
USERVER_NAMESPACE_END