scheduler.hpp

Go to the documentation of this file.

#pragma once
 
 
#include <mapbox/std/weak.hpp>
 
#include <functional>
#include <memory>
 
namespace mbgl {
 
class Mailbox;
 
class Scheduler {
public:
    virtual ~Scheduler() = default;
 
    virtual void schedule(std::function<void()>) = 0;
    virtual mapbox::base::WeakPtr<Scheduler> makeWeakPtr() = 0;
 
    std::function<void()> bindOnce(std::function<void()>);
 
    template <typename TaskFn, typename ReplyFn>
    void scheduleAndReplyValue(const TaskFn& task, const ReplyFn& reply) {
        assert(GetCurrent());
        scheduleAndReplyValue(task, reply, GetCurrent()->makeWeakPtr());
    }
 
    static Scheduler* GetCurrent();
    static void SetCurrent(Scheduler*);
 
    [[nodiscard]] static std::shared_ptr<Scheduler> GetBackground();
 
    [[nodiscard]] static std::shared_ptr<Scheduler> GetSequenced();
 
protected:
    template <typename TaskFn, typename ReplyFn>
    void scheduleAndReplyValue(const TaskFn& task,
                               const ReplyFn& reply,
                               mapbox::base::WeakPtr<Scheduler> replyScheduler) {
        auto scheduled = [replyScheduler = std::move(replyScheduler), task, reply] {
            auto lock = replyScheduler.lock();
            if (!replyScheduler) return;
            auto scheduledReply = [reply, result = task()] { reply(result); };
            replyScheduler->schedule(std::move(scheduledReply));
        };
 
        schedule(std::move(scheduled));
    }
};
 
}