/*
 * =====================================================================================
 *
 *       Filename:  timed_queue.h
 *
 *    Description:  
 *
 *        Version:  1.0
 *        Created:  2021年04月12日 09时36分04秒
 *       Revision:  none
 *       Compiler:  gcc
 *
 *         Author:  Li Chao (), lichao@aiotlink.com
 *   Organization:  
 *
 * =====================================================================================
 */
#ifndef TIMED_QUEUE_Y2YLRBS3
#define TIMED_QUEUE_Y2YLRBS3

#include "bh_util.h"
#include <chrono>
#include <list>
#include <string>

template <class Data, class ClockType = std::chrono::steady_clock>
class TimedData
{
public:
	typedef ClockType Clock;
	typedef typename Clock::time_point TimePoint;
	typedef typename Clock::duration Duration;

	TimedData(const TimePoint &expire, const Data &data) :
	    expire_(expire), data_(data) {}
	TimedData(const TimePoint &expire, Data &&data) :
	    expire_(expire), data_(std::move(data)) {}
	bool Expired() const { return Clock::now() > expire_; }
	const TimePoint &expire() const { return expire_; }
	Data &data() { return data_; }
	Data const &data() const { return data_; }

private:
	TimePoint expire_;
	Data data_;
};

template <class Data, class ClockType = std::chrono::steady_clock>
class TimedQueue
{
	typedef TimedData<Data, ClockType> Record;

public:
	typedef typename Record::Clock Clock;
	typedef typename Record::TimePoint TimePoint;
	typedef typename Record::Duration Duration;

private:
	typedef std::list<Record> Queue;
	Synced<Queue> queue_;

public:
	void Push(Data &&data, const TimePoint &expire) { queue_->emplace_back(expire, std::move(data)); }
	void Push(Data const &data, const TimePoint &expire) { queue_->emplace_back(expire, data); }

	void Push(Data &&data, Duration const &timeout) { Push(std::move(data), Clock::now() + timeout); }
	void Push(Data const &data, Duration const &timeout) { Push(data, Clock::now() + timeout); }

	template <class Func>
	void CheckAll(Func const &func)
	{
		queue_.Apply([&](Queue &q) {
			if (q.empty()) {
				return;
			}
			auto it = q.begin();
			do {
				if (it->Expired()) {
					it = q.erase(it);
				} else if (func(it->data())) {
					it = q.erase(it);
				} else {
					++it;
				}
			} while (it != q.end());
		});
	}
};

#endif // end of include guard: TIMED_QUEUE_Y2YLRBS3