// Copyright 2019 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef LIB_ASYNC_CPP_EXECUTOR_H_
#define LIB_ASYNC_CPP_EXECUTOR_H_

#include <lib/async/dispatcher.h>
#include <lib/async/task.h>
#include <lib/fit/thread_safety.h>
#include <lib/fpromise/promise.h>
#include <lib/fpromise/scheduler.h>
#include <lib/zx/handle.h>
#include <lib/zx/time.h>

#include <mutex>

namespace async {

// Execution context for an asynchronous task that runs within the scope
// of an |async_dispatcher_t|'s dispatch loop, such as a |async::Promise|.
class Context : public fpromise::context {
 public:
  // Gets the executor's |async_dispatcher_t|, never null.
  virtual async_dispatcher_t* dispatcher() const = 0;

 protected:
  ~Context() override = default;
};

// An asynchronous task executor that wraps an |async_dispatcher_t|.
//
// This allows asynchronous tasks, such as promises, to be evaluated alongside
// other asynchronous operations managed by the |async_dispatcher_t|.
class Executor final : public fpromise::executor {
 public:
  // Wraps the specified dispatcher.
  //
  // |dispatcher| must not be null and it must outlive the executor itself.
  explicit Executor(async_dispatcher_t* dispatcher);

  // Destroys the executor along with all of its remaining scheduled tasks
  // that have yet to complete.
  ~Executor() override;

  // Gets the executor's |async_dispatcher_t|, never null.
  async_dispatcher_t* dispatcher() const { return dispatcher_->dispatcher(); }

  // Schedules a task for eventual execution by the executor.
  //
  // This method is thread-safe.
  void schedule_task(fpromise::pending_task task) override;

  Executor(const Executor&) = delete;
  Executor(Executor&&) = delete;
  Executor& operator=(const Executor&) = delete;
  Executor& operator=(Executor&&) = delete;

  // Returns a promise that will complete after the specified duration.
  //
  // The countdown starts when this method is called.
  fpromise::promise<> MakeDelayedPromise(zx::duration duration);

  // Returns a promise that will complete on or after |deadline|.
  fpromise::promise<> MakePromiseForTime(zx::time deadline);

  // Makes a promise that waits for one or more signals on a handle.
  //
  // |object|, |trigger|, and |options| must be valid according to the
  // corresponding arguments to |async::WaitOnce()|.
  //
  // |object| must remain valid at least until |trigger| is sent. The returned promise will only
  // have access to the data that was sent up to the point that |object| received |trigger|.
  fpromise::promise<zx_packet_signal_t, zx_status_t> MakePromiseWaitHandle(
      zx::unowned_handle object, zx_signals_t trigger = ZX_SIGNAL_NONE, uint32_t options = 0);

 private:
  // The dispatcher runs tasks, provides the suspended task resolver, and
  // provides the task context.
  //
  // The lifetime of this object is somewhat complex since there are pointers
  // to it from multiple sources which are released in different ways.
  //
  // - |Executor| holds a pointer in |dispatcher_| which it releases after
  //   calling |Shutdown()| to inform the dispatcher of its own demise
  // - |suspended_task| holds a pointer to the dispatcher's resolver
  //   interface and the number of outstanding pointers corresponds to the
  //   number of outstanding suspended task tickets tracked by |scheduler_|.
  // - |async_dispatcher_t| holds a pointer to the dispatcher's async task
  //   interface whenever dispatch is pending as indicated by |dispatch_pending_|.
  //
  // The dispatcher deletes itself once all pointers have been released.
  // See also |PurgeTasksAndMaybeDeleteSelfLocked()|.
  class DispatcherImpl final : public fpromise::suspended_task::resolver,
                               public async::Context,
                               public async_task_t {
   public:
    DispatcherImpl(async_dispatcher_t* dispatcher, Executor* executor);

    void Shutdown();
    void ScheduleTask(fpromise::pending_task task);

    // |executor()| and |dispatcher()| are presented on the |async::Context|
    // so they are only accessible while |task_running_| is true which
    // implies that |executor_| and |dispatcher_| have not been destroyed.
    Executor* executor() const override { return executor_; }
    async_dispatcher_t* dispatcher() const override { return dispatcher_; }

    // Suspends the currently running task.  This method is presented
    // on the |async::Context| so it can only be called while
    // |task_running_| is true as above.
    fpromise::suspended_task suspend_task() override;

    // These methods implement the suspended task token contract.
    // They may be called on any thread at any time.
    fpromise::suspended_task::ticket duplicate_ticket(
        fpromise::suspended_task::ticket ticket) override;
    void resolve_ticket(fpromise::suspended_task::ticket ticket, bool resume_task) override;

   private:
    ~DispatcherImpl() override;

    // Callback from |async_dispatcher_t*|.
    // Invokes |Dispatch()| to run all runnable tasks.
    static void Dispatch(async_dispatcher_t* dispatcher, async_task_t* task, zx_status_t status);
    void Dispatch(zx_status_t status);

    // Runs the specified task.  Called by |Dispatch()|.
    void RunTask(fpromise::pending_task* task);

    // Attempts to schedule a call to |Dispatch()| on the async dispatcher.
    // Returns true if a dispatch is pending.
    bool ScheduleDispatchLocked() FIT_REQUIRES(guarded_.mutex_);

    // Moves all tasks from |incoming_tasks_| to the |scheduler_| runnable queue.
    void AcceptIncomingTasksLocked() FIT_REQUIRES(guarded_.mutex_);

    // When |was_shutdown_| or |loop_failure_| is true, purges any tasks
    // that remain and deletes the dispatcher if all outstanding references
    // to it have gone away.  Should be called at points where one of these
    // conditions changes.  Takes ownership of the lock and drops it.
    void PurgeTasksAndMaybeDeleteSelfLocked(std::unique_lock<std::mutex> lock)
        FIT_REQUIRES(guarded_.mutex_);

    async_dispatcher_t* const dispatcher_;
    Executor* const executor_;

    // The queue of runnable tasks.
    // Only accessed by |RunTask()| and |suspend_task()| which happens
    // on the dispatch thread.
    fpromise::subtle::scheduler::task_queue runnable_tasks_;

    // The current suspended task ticket or 0 if none.
    // Only accessed by |RunTask()| and |suspend_task()| which happens
    // on the dispatch thread.
    fpromise::suspended_task::ticket current_task_ticket_ = 0;

    // A bunch of state that is guarded by a mutex.
    struct {
      std::mutex mutex_;

      // True if the executor is about to be destroyed.
      bool was_shutdown_ FIT_GUARDED(mutex_) = false;

      // True if the underlying async_dispatcher_t reported an error.
      bool loop_failure_ FIT_GUARDED(mutex_) = false;

      // True if a call to |Dispatch()| is pending.
      bool dispatch_pending_ FIT_GUARDED(mutex_) = false;

      // True while |RunTask| is running a task.
      bool task_running_ FIT_GUARDED(mutex_) = false;

      // Holds tasks that have been scheduled on this dispatcher.
      fpromise::subtle::scheduler scheduler_ FIT_GUARDED(mutex_);

      // Newly scheduled tasks which have yet to be added to the
      // runnable queue.  This allows the dispatch to distinguish between
      // newly scheduled tasks and resumed tasks so it can manage them
      // separately.  See comments in |Dispatch()|.
      fpromise::subtle::scheduler::task_queue incoming_tasks_ FIT_GUARDED(mutex_);
    } guarded_;
  };

  DispatcherImpl* dispatcher_;
};

}  // namespace async

#endif  // LIB_ASYNC_CPP_EXECUTOR_H_