payload_processor.h

Go to the documentation of this file.

#ifndef CVMFS_RECEIVER_PAYLOAD_PROCESSOR_H_
#define CVMFS_RECEIVER_PAYLOAD_PROCESSOR_H_

#include <stdint.h>
#include <map>
#include <string>
#include <vector>

#include "pack.h"
#include "upload.h"
#include "util/raii_temp_dir.h"

namespace receiver {

struct FileInfo {
  FileInfo();
  explicit FileInfo(const ObjectPackBuild::Event& event);
  FileInfo(const FileInfo& other);
  FileInfo& operator=(const FileInfo& other);

  upload::UploadStreamHandle *handle;
  size_t total_size;
  size_t current_size;
  shash::ContextPtr hash_context;
  std::vector<unsigned char> hash_buffer;
};

class PayloadProcessor {
 public:
  enum Result { kSuccess, kPathViolation, kUploaderError, kOtherError };

  PayloadProcessor();
  virtual ~PayloadProcessor();

  Result Process(int fdin, const std::string& header_digest,
                 const std::string& path, uint64_t header_size);

  virtual void ConsumerEventCallback(const ObjectPackBuild::Event& event);

  virtual void OnUploadJobComplete(const upload::UploaderResults &results,
                                   void *buffer);

  int GetNumErrors() const { return num_errors_; }

  void SetStatistics(perf::Statistics *st);

 protected:
  // NOTE: These methods are made virtual such that they can be mocked for
  //       the purpose of unit testing
  virtual Result Initialize();
  virtual Result Finalize();

 private:
  typedef std::map<shash::Any, FileInfo>::iterator FileIterator;
  std::map<shash::Any, FileInfo> pending_files_;
  std::string current_repo_;
  UniquePtr<upload::AbstractUploader> uploader_;
  UniquePtr<RaiiTempDir> temp_dir_;
  int num_errors_;
  UniquePtr<perf::StatisticsTemplate> statistics_;
};

}  // namespace receiver

#endif  // CVMFS_RECEIVER_PAYLOAD_PROCESSOR_H_