catalog_diff_tool.h

Go to the documentation of this file.

#ifndef CVMFS_CATALOG_DIFF_TOOL_H_
#define CVMFS_CATALOG_DIFF_TOOL_H_

#include <string>

#include "directory_entry.h"
#include "file_chunk.h"
#include "shortstring.h"
#include "statistics.h"
#include "util/pointer.h"
#include "util/raii_temp_dir.h"
#include "xattr.h"

namespace download {
class DownloadManager;
}

template<typename RoCatalogMgr>
class CatalogDiffTool {
 public:
  CatalogDiffTool(RoCatalogMgr *old_catalog_mgr, RoCatalogMgr *new_catalog_mgr)
      : repo_path_("")
      , temp_dir_prefix_("")
      , download_manager_(NULL)
      , cache_dir_("")
      , old_catalog_mgr_(old_catalog_mgr)
      , new_catalog_mgr_(new_catalog_mgr)
      , needs_setup_(false) { }

  CatalogDiffTool(const std::string &repo_path,
                  const shash::Any &old_root_hash,
                  const shash::Any &new_root_hash,
                  const std::string &temp_dir_prefix,
                  download::DownloadManager *download_manager,
                  const std::string &cache_dir = "")
      : repo_path_(repo_path)
      , old_root_hash_(old_root_hash)
      , new_root_hash_(new_root_hash)
      , temp_dir_prefix_(temp_dir_prefix)
      , download_manager_(download_manager)
      , cache_dir_(cache_dir)
      , old_raii_temp_dir_()
      , new_raii_temp_dir_()
      , old_catalog_mgr_()
      , new_catalog_mgr_()
      , needs_setup_(true) { }

  virtual ~CatalogDiffTool() { }

  bool Init();

  bool Run(const PathString &path);

 protected:
  virtual bool IsIgnoredPath(const PathString & /* path */) { return false; }

  virtual bool IsReportablePath(const PathString & /* path */) { return true; }

  // Note that addition and modification can return false to indicate that
  // the recursion stops. In the merge tool, this happens at nested catalog
  // transition points:
  //  - For a new directory that is a nested catalog, we don't need to recurse
  //    further but just install that nested catalog in the parent
  //  - When a nested catalog is replaced, we likewise do not need to recurse
  //    further into the new nested catalog tree.

  virtual bool ReportAddition(const PathString &path,
                              const catalog::DirectoryEntry &entry,
                              const XattrList &xattrs,
                              const FileChunkList &chunks) = 0;
  virtual void ReportRemoval(const PathString &path,
                             const catalog::DirectoryEntry &entry) = 0;
  virtual bool ReportModification(const PathString &path,
                                  const catalog::DirectoryEntry &old_entry,
                                  const catalog::DirectoryEntry &new_entry,
                                  const XattrList &xattrs,
                                  const FileChunkList &chunks) = 0;

  const catalog::Catalog *GetOldCatalog() const {
    return old_catalog_mgr_->GetRootCatalog();
  }
  const catalog::Catalog *GetNewCatalog() const {
    return new_catalog_mgr_->GetRootCatalog();
  }
  RoCatalogMgr *GetOldCatalogMgr() { return old_catalog_mgr_.weak_ref(); }
  RoCatalogMgr *GetNewCatalogMgr() { return new_catalog_mgr_.weak_ref(); }

 private:
  RoCatalogMgr *OpenCatalogManager(const std::string &repo_path,
                                   const std::string &temp_dir,
                                   const shash::Any &root_hash,
                                   download::DownloadManager *download_manager,
                                   perf::Statistics *stats,
                                   const std::string &cache_dir);

  void DiffRec(const PathString &path);

  std::string repo_path_;
  shash::Any old_root_hash_;
  shash::Any new_root_hash_;
  std::string temp_dir_prefix_;

  download::DownloadManager *download_manager_;
  const std::string cache_dir_;  // path if local caching of catalogs

  perf::Statistics stats_old_;
  perf::Statistics stats_new_;

  UniquePtr<RaiiTempDir> old_raii_temp_dir_;
  UniquePtr<RaiiTempDir> new_raii_temp_dir_;

  UniquePtr<RoCatalogMgr> old_catalog_mgr_;
  UniquePtr<RoCatalogMgr> new_catalog_mgr_;

  const bool needs_setup_;
};

#include "catalog_diff_tool_impl.h"

#endif  // CVMFS_CATALOG_DIFF_TOOL_H_