client.hpp

/*
 * Copyright (c) 2024 Abdullatif Kalla.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE.txt file in the root directory of this source tree.
 */
 
#ifndef XCLOX_CLIENT_HPP
#define XCLOX_CLIENT_HPP
 
#include "query.hpp"
 
#include <list>
 
namespace xclox {
 
namespace ntp {
 
    class Client {
    public:
        using Callback = Query::Callback; 
        using Status = Query::Status; 
        using DefaultTimeout = Query::DefaultTimeout; 
 
 
        explicit Client(Callback callable)
            : m_callable(callable)
        {
        }
 
        ~Client()
        {
            m_pool.join();
        }
 
        void query(const std::string& server, const std::chrono::milliseconds& timeout = std::chrono::milliseconds(DefaultTimeout::ms))
        {
            std::lock_guard<std::mutex> lock(m_mutex);
            purgeQueryList();
            m_queryList.push_back(Query::start(m_pool, server, m_callable, timeout));
        }
 
        void setCallback(Callback callable)
        {
            m_callable = std::move(callable);
        }
 
        void cancel()
        {
            std::lock_guard<std::mutex> lock(m_mutex);
            for (auto query : m_queryList) {
                if (auto shared = query.lock()) {
                    shared->cancel();
                }
            }
            purgeQueryList();
        }
 
    private:
        void purgeQueryList()
        {
            m_queryList.remove_if(std::mem_fn(&std::weak_ptr<Query>::expired));
        }
 
        Callback m_callable;
        asio::thread_pool m_pool;
        std::mutex m_mutex;
        std::list<std::weak_ptr<Query>> m_queryList;
    };
 
} // namespace ntp
 
} // namespace xclox
 
#endif // XCLOX_CLIENT_HPP