task_impl.cpp

Go to the documentation of this file.

// Copyright 2023 owent
 
#include <libcopp/utils/config/libcopp_build_features.h>
 
#include <libcopp/coroutine/coroutine_context.h>
#include <libcopp/coroutine/coroutine_context_fiber.h>
 
#include <libcotask/impl/task_action_impl.h>
#include <libcotask/impl/task_impl.h>
 
// clang-format off
#include <libcopp/utils/config/stl_include_prefix.h>  // NOLINT(build/include_order)
// clang-format on
#include <assert.h>
#include <algorithm>
#include <cstdlib>
// clang-format off
#include <libcopp/utils/config/stl_include_suffix.h>  // NOLINT(build/include_order)
// clang-format on
 
LIBCOPP_COTASK_NAMESPACE_BEGIN
namespace impl {
LIBCOPP_COTASK_API task_impl::task_impl()
    : action_(nullptr), id_(0), finish_priv_data_(nullptr), status_(EN_TS_CREATED) {
  id_allocator_t id_alloc_;
  ((void)id_alloc_);
  id_ = id_alloc_.allocate();
}
 
LIBCOPP_COTASK_API task_impl::~task_impl() {
  assert(status_ <= EN_TS_CREATED || status_ >= EN_TS_DONE);
 
  // free resource
  id_allocator_t id_alloc_;
  ((void)id_alloc_);
  id_alloc_.deallocate(id_);
}
 
LIBCOPP_COTASK_API bool task_impl::is_canceled() const LIBCOPP_MACRO_NOEXCEPT { return EN_TS_CANCELED == get_status(); }
 
LIBCOPP_COTASK_API bool task_impl::is_completed() const LIBCOPP_MACRO_NOEXCEPT { return is_exiting(); }
 
LIBCOPP_COTASK_API bool task_impl::is_faulted() const LIBCOPP_MACRO_NOEXCEPT { return EN_TS_KILLED <= get_status(); }
 
LIBCOPP_COTASK_API bool task_impl::is_timeout() const LIBCOPP_MACRO_NOEXCEPT { return EN_TS_TIMEOUT == get_status(); }
 
LIBCOPP_COTASK_API bool task_impl::is_exiting() const LIBCOPP_MACRO_NOEXCEPT { return EN_TS_DONE <= get_status(); }
 
LIBCOPP_COTASK_API int task_impl::on_finished() { return 0; }
 
LIBCOPP_COTASK_API task_impl *task_impl::this_task() {
  LIBCOPP_COPP_NAMESPACE_ID::coroutine_context_base *this_co =
      LIBCOPP_COPP_NAMESPACE_ID::coroutine_context_base::get_this_coroutine_base();
  if (nullptr == this_co) {
    return nullptr;
  }
 
  if (false == this_co->check_flags(ext_coroutine_flag_t::EN_ECFT_COTASK)) {
    return nullptr;
  }
 
  return *reinterpret_cast<task_impl **>(this_co->get_private_buffer());
}
 
LIBCOPP_COTASK_API void task_impl::_set_action(action_ptr_type action) { action_ = action; }
 
LIBCOPP_COTASK_API task_impl::action_ptr_type task_impl::_get_action() { return action_; }
 
LIBCOPP_COTASK_API bool task_impl::_cas_status(EN_TASK_STATUS &expected, EN_TASK_STATUS desired) {
  uint32_t expected_int = expected;
  bool ret =
      status_.compare_exchange_weak(expected_int, desired, LIBCOPP_COPP_NAMESPACE_ID::util::lock::memory_order_acq_rel,
                                    LIBCOPP_COPP_NAMESPACE_ID::util::lock::memory_order_acquire);
  expected = static_cast<EN_TASK_STATUS>(expected_int);
  return ret;
}
 
#if defined(LIBCOPP_MACRO_ENABLE_STD_EXCEPTION_PTR) && LIBCOPP_MACRO_ENABLE_STD_EXCEPTION_PTR
LIBCOPP_COTASK_API int task_impl::_notify_finished(std::list<std::exception_ptr> &unhandled, void *priv_data) {
#else
LIBCOPP_COTASK_API int task_impl::_notify_finished(void *priv_data) {
#endif
  finish_priv_data_ = priv_data;
 
#if defined(LIBCOPP_MACRO_ENABLE_STD_EXCEPTION_PTR) && LIBCOPP_MACRO_ENABLE_STD_EXCEPTION_PTR
  try {
#endif
    _get_action()->on_finished(*this);
    int ret = on_finished();
#if defined(LIBCOPP_MACRO_ENABLE_STD_EXCEPTION_PTR) && LIBCOPP_MACRO_ENABLE_STD_EXCEPTION_PTR
    return ret;
  } catch (...) {
    unhandled.emplace_back(std::current_exception());
    return LIBCOPP_COPP_NAMESPACE_ID::COPP_EC_HAS_UNHANDLE_EXCEPTION;
  }
#else
  return ret;
#endif
}
}  // namespace impl
LIBCOPP_COTASK_NAMESPACE_END