jeanlemotan
42 lines
889 B
C++
42 lines
889 B
C++
///////////////////////////////////////////////////////////////////////////////
|
|
// Copyright (c) Lewis Baker
|
|
// Licenced under MIT license. See LICENSE.txt for details.
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
#include <cppcoro/cancellation_registration.hpp>
|
|
|
|
#include "cancellation_state.hpp"
|
|
|
|
#include <cassert>
|
|
|
|
cppcoro::cancellation_registration::~cancellation_registration()
|
|
{
|
|
if (m_state != nullptr)
|
|
{
|
|
m_state->deregister_callback(this);
|
|
m_state->release_token_ref();
|
|
}
|
|
}
|
|
|
|
void cppcoro::cancellation_registration::register_callback(cancellation_token&& token)
|
|
{
|
|
auto* state = token.m_state;
|
|
if (state != nullptr && state->can_be_cancelled())
|
|
{
|
|
m_state = state;
|
|
if (state->try_register_callback(this))
|
|
{
|
|
token.m_state = nullptr;
|
|
}
|
|
else
|
|
{
|
|
m_state = nullptr;
|
|
m_callback();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
m_state = nullptr;
|
|
}
|
|
}
|