FileCache.h

// -*- indent-tabs-mode: nil -*-

#ifndef FILECACHE_H_
#define FILECACHE_H_

#include <sstream>
#include <vector>
#include <map>
#include <set>
#include <arc/DateTime.h>
#include <arc/Logger.h>

#include "FileCacheHash.h"

namespace Arc {


  struct CacheParameters {
    std::string cache_path;
    std::string cache_link_path;
  };


  class FileCache {
   private:
    std::map <std::string, struct CacheParameters> _cache_map;
    std::vector<struct CacheParameters> _caches;
    std::vector<struct CacheParameters> _draining_caches;
    std::vector<struct CacheParameters> _readonly_caches;
    std::set<std::string> _urls_unlocked;
    std::string _id;
    uid_t _uid;
    gid_t _gid;

    static const std::string CACHE_DATA_DIR;
    static const std::string CACHE_JOB_DIR;
    static const int CACHE_DIR_LENGTH;
    static const int CACHE_DIR_LEVELS;
    static const std::string CACHE_META_SUFFIX;
    static const int CACHE_DEFAULT_AUTH_VALIDITY;
    static const int CACHE_LOCK_TIMEOUT;
    static const int CACHE_META_LOCK_TIMEOUT;

    bool _init(const std::vector<std::string>& caches,
               const std::vector<std::string>& draining_caches,
               const std::vector<std::string>& readonly_caches,
               const std::string& id,
               uid_t job_uid,
               gid_t job_gid);
    bool _checkMetaFile(const std::string& filename, const std::string& url, bool& is_locked);
    bool _createMetaFile(const std::string& meta_file, const std::string& content, bool& is_locked);
    std::string _getMetaFileName(const std::string& url);
    std::string _getHash(const std::string& url) const;
    struct CacheParameters _chooseCache(const std::string& url) const;
    float _getCacheInfo(const std::string& path) const;
    bool _cleanFilesAndReturnFalse(const std::string& hard_link_file, bool& locked);

    static Logger logger;

   public:

    FileCache(const std::string& cache_path,
              const std::string& id,
              uid_t job_uid,
              gid_t job_gid);


    FileCache(const std::vector<std::string>& caches,
              const std::string& id,
              uid_t job_uid,
              gid_t job_gid);


    FileCache(const std::vector<std::string>& caches,
              const std::vector<std::string>& draining_caches,
              const std::string& id,
              uid_t job_uid,
              gid_t job_gid);


    FileCache(const std::vector<std::string>& caches,
              const std::vector<std::string>& draining_caches,
              const std::vector<std::string>& readonly_caches,
              const std::string& id,
              uid_t job_uid,
              gid_t job_gid);

    FileCache(): _uid(0),_gid(0) {
      _caches.clear();
    }


    bool Start(const std::string& url,
               bool& available,
               bool& is_locked,
               bool delete_first = false);


    bool Stop(const std::string& url);


    bool StopAndDelete(const std::string& url);


    std::string File(const std::string& url);


    bool Link(const std::string& link_path,
              const std::string& url,
              bool copy,
              bool executable,
              bool holding_lock,
              bool& try_again);


    bool Release() const;


    bool AddDN(const std::string& url, const std::string& DN, const Time& expiry_time);


    bool CheckDN(const std::string& url, const std::string& DN);


    bool CheckCreated(const std::string& url);


    Time GetCreated(const std::string& url);

    operator bool() {
      return (!_caches.empty());
    };

    bool operator==(const FileCache& a);

  };

} // namespace Arc

#endif /*FILECACHE_H_*/