cancellation_token_source.hpp

#pragma once
 
#include "cancellation_token.hpp"
 
namespace Cvb
{
 
CVB_BEGIN_INLINE_NS
 
template <>
inline HandleGuard<CancellationTokenSource>::HandleGuard(void* handle) noexcept
  : HandleGuard<CancellationTokenSource>(handle, [](void* handle) { CVB_CALL_CAPI(ReleaseObject(handle)); })
{
}
 
 
class CancellationTokenSource
{
  public:
 
 
    CancellationTokenSource()
      : handle_(nullptr)
    {
      CExports::CVCANCELLATIONTOKENSOURCE sourceHandle = nullptr;
      CExports::CVCANCELLATIONTOKEN tokenHandle = nullptr;
      CVB_CALL_CAPI_CHECKED(CVCCreateCancellationToken(sourceHandle, tokenHandle));
      handle_ = HandleGuard<CancellationTokenSource>(sourceHandle);
      token_ = std::make_shared<CancellationToken>(HandleGuard<CancellationToken>(tokenHandle), CancellationToken::PrivateTag{});
    }
 
 
    void* Handle() const noexcept
    {
      return handle_.Handle();
    }
 
 
    CancellationTokenPtr Token() const noexcept
    {
      return token_;
    }
 
 
    void Cancel()
    {
      CVB_CALL_CAPI_CHECKED(CVCCancellationTokenSourceCancel(Handle()));
    }
 
  private:
 
    HandleGuard<CancellationTokenSource> handle_;
    CancellationTokenPtr token_;
};
 
CVB_END_INLINE_NS
 
}