afcc695edf
Summary: Introduce a 'Scheduler' interface which will allow to decouple from the existing used Folly scheduler. Reviewed By: fabiomassimo Differential Revision: D36245587 fbshipit-source-id: 2f28bc1612e37ae53060a134d1c8059231fbc8ad
41 lines
978 B
C++
41 lines
978 B
C++
/*
|
|
* Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
*
|
|
* This source code is licensed under the MIT license found in the
|
|
* LICENSE file in the root directory of this source tree.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "FlipperScheduler.h"
|
|
|
|
#include <folly/futures/Future.h>
|
|
#include <folly/io/async/EventBase.h>
|
|
#include <folly/io/async/ScopedEventBaseThread.h>
|
|
|
|
namespace facebook {
|
|
namespace flipper {
|
|
|
|
struct FollyScopedThreadScheduler : public Scheduler {
|
|
virtual void schedule(Func&& t) override {
|
|
thread_.getEventBase()->add(t);
|
|
}
|
|
|
|
virtual void scheduleAfter(Func&& t, unsigned int ms) override {
|
|
folly::makeFuture()
|
|
.via(thread_.getEventBase())
|
|
.delayed(std::chrono::milliseconds(ms))
|
|
.thenValue([t](auto&&) { t(); });
|
|
}
|
|
|
|
virtual bool isRunningInOwnThread() override {
|
|
return thread_.getEventBase()->isInEventBaseThread();
|
|
}
|
|
|
|
private:
|
|
folly::ScopedEventBaseThread thread_;
|
|
};
|
|
|
|
} // namespace flipper
|
|
} // namespace facebook
|