litehtml/support/webpage/http_requests_pool.h

#ifndef LITEBROWSER_HTTP_REQUESTS_POOL_H
#define LITEBROWSER_HTTP_REQUESTS_POOL_H

#include <memory>
#include <mutex>
#include <list>
#include <condition_variable>
#include "http_request.h"

namespace litebrowser
{
	class http_requests_pool;

	class pool_thread
	{
		http_requests_pool* m_pool;
		std::shared_ptr<http_request> m_request;
		std::mutex m_request_mutex;
		std::thread m_thread;
	public:
		explicit pool_thread(http_requests_pool* pool) : m_pool(pool)
		{
			m_thread = std::thread(&pool_thread::thread_loop, this);
		}
		std::shared_ptr<http_request> get_request()
		{
			std::unique_lock<std::mutex> request_lock(m_request_mutex);
			return m_request;
		}
		std::thread& thread() { return m_thread; }
	private:
		void thread_loop();
	};

	class http_requests_pool : public std::enable_shared_from_this<http_requests_pool>
	{
		friend class pool_thread;
	protected:
		bool m_cancel;
		std::mutex m_queue_mutex;
		std::condition_variable m_mutex_condition;
		std::list<std::shared_ptr<http_request>> m_queue;
		std::vector<std::shared_ptr<pool_thread>> m_threads;
		std::function<void()> m_queue_empty_cb;
		std::function<void()> m_cb_update_state;
	public:
		explicit http_requests_pool(int pool_size, const std::function<void()>& cb_update_state);
		~http_requests_pool()
		{
			stop();
		}

		void enqueue(const std::string& url,
					 const std::function<void(void* data, size_t len, size_t downloaded, size_t total)>& cb_on_data,
					 const std::function<void(u_int32_t http_status, u_int32_t err_code, const std::string& err_text, const std::string& url)>& cb_on_finish);
		void stop();
		bool is_downloading();
	private:
	};
}

#endif //LITEBROWSER_HTTP_REQUESTS_POOL_H