pw_async/public/pw_async/dispatcher.h

// Copyright 2022 The Pigweed Authors
//
// Licensed under the Apache License, Version 2.0 (the "License"); you may not
// use this file except in compliance with the License. You may obtain a copy of
// the License at
//
//     https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
// License for the specific language governing permissions and limitations under
// the License.
#pragma once

#include "pw_async/task.h"
#include "pw_chrono/system_clock.h"

namespace pw::async {

// Asynchronous Dispatcher abstract class. A default implementation is provided
// in dispatcher_basic.h.
//
// Dispatcher implements VirtualSystemClock so the Dispatcher's time can be
// injected into other modules under test. This is useful for consistently
// simulating time when using TestDispatcher (rather than using
// chrono::SimulatedSystemClock separately).
class Dispatcher : public chrono::VirtualSystemClock {
 public:
  ~Dispatcher() override = default;

  virtual void RequestStop() = 0;

  // Post caller owned |task|.
  virtual void PostTask(Task& task) = 0;

  // Post caller owned |task| to be run after |delay|.
  virtual void PostDelayedTask(Task& task,
                               chrono::SystemClock::duration delay) = 0;

  // Post caller owned |task| to be run at |time|.
  virtual void PostTaskForTime(Task& task,
                               chrono::SystemClock::time_point time) = 0;

  // Post caller owned |task| to be run immediately then rerun at a regular
  // |interval|.
  virtual void SchedulePeriodicTask(Task& task,
                                    chrono::SystemClock::duration interval) = 0;
  // Post caller owned |task| to be run at |start_time| then rerun at a regular
  // |interval|.
  virtual void SchedulePeriodicTask(
      Task& task,
      chrono::SystemClock::duration interval,
      chrono::SystemClock::time_point start_time) = 0;

  // Returns true if |task| is succesfully canceled.
  // If cancelation fails, the task may be running or completed.
  // Periodic tasks may be posted once more after they are canceled.
  virtual bool Cancel(Task& task) = 0;

  // Execute tasks until the Dispatcher enters a state where none are queued.
  virtual void RunUntilIdle() = 0;

  // Run the Dispatcher until Now() has reached `end_time`, executing all tasks
  // that come due before then.
  virtual void RunUntil(chrono::SystemClock::time_point end_time) = 0;

  // Run the Dispatcher until `duration` has elapsed, executing all tasks that
  // come due in that period.
  virtual void RunFor(chrono::SystemClock::duration duration) = 0;
};

}  // namespace pw::async