ThreadPool.h

#pragma once
 
#include "Impl/ImmediateScheduler.h"
#include "Impl/cesium-async++.h"
#include "Library.h"
 
#include <memory>
 
namespace CesiumAsync {
 
class CESIUMASYNC_API ThreadPool {
public:
  ThreadPool(int32_t numberOfThreads);
 
private:
  struct Scheduler {
    Scheduler(int32_t numberOfThreads);
    void schedule(async::task_run_handle t);
 
    CesiumImpl::ImmediateScheduler<Scheduler> immediate{this};
 
    async::threadpool_scheduler scheduler;
  };
 
  static auto createPreRun(ThreadPool::Scheduler* pScheduler) {
    return
        [pScheduler]() { ThreadPool::_scope = pScheduler->immediate.scope(); };
  }
 
  static auto createPostRun() noexcept {
    return []() noexcept { ThreadPool::_scope.reset(); };
  }
 
  static thread_local CesiumImpl::ImmediateScheduler<Scheduler>::SchedulerScope
      _scope;
 
  std::shared_ptr<Scheduler> _pScheduler;
 
  template <typename T> friend class Future;
  template <typename T> friend class SharedFuture;
  friend class AsyncSystem;
};
 
} // namespace CesiumAsync