FailbackPollerService.hpp

Go to the documentation of this file.

#ifndef __FAILBACKPOLLERSERVICE_HPP__
#define __FAILBACKPOLLERSERVICE_HPP__

#include "gpudb/GPUdb.hpp"

#include <iostream>
#include <thread>
#include <atomic>
#include <mutex>
#include <memory>
#include <vector>
#include <chrono>


namespace gpudb {
    class FailbackPollerService {
        public:
            static constexpr int DEFAULT_START_DELAY = 0;
            static constexpr int DEFAULT_POLLING_INTERVAL = 5; // seconds
        
        private:
        
            const gpudb::GPUdb& db;
            std::string primaryUrl;
            long pollingInterval;
            std::atomic<bool> is_running;
            std::mutex service_mutex;
            std::unique_ptr<std::thread> scheduler;
        
            FailbackPollerService(const gpudb::GPUdb& db, long pollingInterval) 
                : db(db), primaryUrl(db.getPrimaryURL()), pollingInterval(pollingInterval), is_running(false) {}
        
            void polling_task();
        
            bool poll(); 
        
            void reset_cluster_pointers();

            void handle_exception(const std::exception& e);

            inline void log_debug(const std::string& msg) { std::cout << msg << std::endl; }
            inline void log_info(const std::string& msg) { std::cout << msg << std::endl; }
            inline void log_warn(const std::string& msg) { std::cout << msg << std::endl; }
            inline void log_error(const std::string& msg) { std::cout << msg << std::endl; }
        
        
        public:
            static FailbackPollerService& get_instance(const gpudb::GPUdb& db, long pollingInterval = DEFAULT_POLLING_INTERVAL);

            // Delete copy constructor and assignment operator
            FailbackPollerService(const FailbackPollerService&) = delete;
            FailbackPollerService& operator=(const FailbackPollerService&) = delete;

            virtual ~FailbackPollerService();

            bool isRunning();
        
            void start();
        
            void stop();
        
            void restart();
        };
        
}

#endif