///////////////////////////////////////////////////////////////////////////////

// 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;

}

}