tier.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 <45d/Quota.hpp>
#include <boost/filesystem.hpp>
#include <mutex>
#include <queue>
#include <rocksdb/db.h>
namespace fs = boost::filesystem;
 
class File;
 
class Tier {
private:
    ffd::Quota quota_; 
    ffd::Bytes usage_; 
 
    ffd::Bytes sim_usage_;
    std::string id_;
    fs::path path_; 
 
    std::vector<File *> incoming_files_;
    void copy_ownership_and_perms(const fs::path &old_path, const fs::path &new_path) const;
    std::mutex usage_mt_; 
public:
    Tier(std::string id, const fs::path &path, const ffd::Quota &quota);
    Tier(Tier &&other)
        : quota_(std::move(other.quota_))
        , usage_(std::move(other.usage_))
        , sim_usage_(std::move(other.sim_usage_))
        , id_(std::move(other.id_))
        , path_(std::move(other.path_))
        , incoming_files_(std::move(other.incoming_files_))
        , usage_mt_() {}
    ~Tier() = default;
    void add_file_size(ffd::Bytes size);
    void subtract_file_size(ffd::Bytes size);
    void size_delta(ffd::Bytes old_size, ffd::Bytes new_size);
    void add_file_size_sim(ffd::Bytes size);
    void subtract_file_size_sim(ffd::Bytes size);
    void quota_percent(double quota_percent);
    double quota_percent(void) const;
    ffd::Quota quota(void) const;
    bool full_test(const ffd::Bytes &file_size) const;
    const fs::path &path(void) const;
    const std::string &id(void) const;
    void enqueue_file_ptr(File *fptr);
    void transfer_files(int buff_sz, const fs::path &run_path);
    bool move_file(const fs::path &old_path,
                   const fs::path &new_path,
                   int buff_sz,
                   bool *conflicted = nullptr,
                   std::string orig_tier = "") const;
    void usage(ffd::Bytes usage);
    double usage_percent(void) const;
    ffd::Bytes usage_bytes(void) const;
    void reset_sim(void);
    ffd::Bytes capacity(void) const;
};