interval.hpp

//                  ReactivePlusPlus library
//
//          Copyright Aleksey Loginov 2022 - present.
// Distributed under the Boost Software License, Version 1.0.
//    (See accompanying file LICENSE_1_0.txt or copy at
//          https://www.boost.org/LICENSE_1_0.txt)
//
// Project home: https://github.com/victimsnino/ReactivePlusPlus
//
 
#pragma once
 
#include <rpp/sources/fwd.hpp>
#include <rpp/sources/create.hpp>
#include <rpp/schedulers/fwd.hpp>
#include <rpp/schedulers/trampoline_scheduler.hpp>
 
#include <type_traits>
 
 
IMPLEMENTATION_FILE(interval_tag);
 
namespace rpp::observable
{
    template<schedulers::constraint::scheduler TScheduler /*= schedulers::trampoline */>
    auto interval(schedulers::duration period, const TScheduler& scheduler /* = TScheduler{} */) requires rpp::details::is_header_included<rpp::details::interval_tag, TScheduler>
    {
        return interval(period, period, scheduler);
    }
 
    template<schedulers::constraint::scheduler TScheduler /*= schedulers::trampoline*/>
    auto interval(schedulers::duration first_delay, schedulers::duration period, const TScheduler& scheduler /* = TScheduler{} */) requires rpp::details::is_header_included<rpp::details::interval_tag, TScheduler>
    {
        return source::create<size_t>([first_delay, period, scheduler](auto&& subscriber)
        {
            auto worker = scheduler.create_worker(subscriber.get_subscription());
            worker.schedule(first_delay, [counter = size_t{}, subscriber = std::forward<decltype(subscriber)>(subscriber), period]() mutable-> schedulers::optional_duration
            {
                subscriber.on_next(counter++);
                return period;
            });
        });
    }
} // namespace rpp::observable