adhoc.hpp

/*
 *    Copyright (C) 2019-2021 Joshua Boudreau <jboudreau@45drives.com>
 *
 *    This file is part of autotier.
 *
 *    autotier is free software: you can redistribute it and/or modify
 *    it under the terms of the GNU General Public License as published by
 *    the Free Software Foundation, either version 3 of the License, or
 *    (at your option) any later version.
 *
 *    autotier is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *    GNU General Public License for more details.
 *
 *    You should have received a copy of the GNU General Public License
 *    along with autotier.  If not, see <https://www.gnu.org/licenses/>.
 */
 
#pragma once
 
#include "base.hpp"
#include "concurrentQueue.hpp"
#include "tools.hpp"
 
#include <45d/socket/UnixSocketServer.hpp>
 
class TierEngineAdhoc : virtual public TierEngineBase {
public:
    TierEngineAdhoc(const fs::path &config_path, const ConfigOverrides &config_overrides);
    ~TierEngineAdhoc();
    void set_socket_permissions(void);
    void process_adhoc_requests(void);
    void process_oneshot(const AdHoc &work);
    void process_pin_unpin(const AdHoc &work);
    void process_status(const AdHoc &work);
    void process_config(void);
    void process_list_pins(void);
    void process_list_popularity(void);
    void process_which_tier(AdHoc &work);
    template<typename... Args>
    void enqueue_work(Args &&...args) {
        if ([](int job, const std::vector<std::string> &) {return job;} (args...) == ONESHOT) {
            if (oneshot_in_queue_)
                return;
            oneshot_in_queue_ = true;
        }
        adhoc_work_.emplace(args...);
        sleep_cv_.notify_one();
    }
    void execute_queued_work(void);
    void pin_files(const std::vector<std::string> &args);
    void unpin_files(const std::vector<std::string> &args);
    void shutdown_socket_server(void);
private:
    bool oneshot_in_queue_;
    ffd::UnixSocketServer socket_server_; 
};