ceph/src/rgw/rgw_compression.h

   1 // -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
   2 // vim: ts=8 sw=2 smarttab ft=cpp
   3
   4 #ifndef CEPH_RGW_COMPRESSION_H
   5 #define CEPH_RGW_COMPRESSION_H
   6
   7 #include <vector>
   8
   9 #include "compressor/Compressor.h"
  10 #include "rgw_putobj.h"
  11 #include "rgw_op.h"
  12 #include "rgw_compression_types.h"
  13
  14 int rgw_compression_info_from_attr(const bufferlist& attr,
  15                                    bool& need_decompress,
  16                                    RGWCompressionInfo& cs_info);
  17 int rgw_compression_info_from_attrset(const map<string, bufferlist>& attrs,
  18                                       bool& need_decompress,
  19                                       RGWCompressionInfo& cs_info);
  20
  21 class RGWGetObj_Decompress : public RGWGetObj_Filter
  22 {
  23   CephContext* cct;
  24   CompressorRef compressor;
  25   RGWCompressionInfo* cs_info;
  26   bool partial_content;
  27   vector<compression_block>::iterator first_block, last_block;
  28   off_t q_ofs, q_len;
  29   uint64_t cur_ofs;
  30   bufferlist waiting;
  31 public:
  32   RGWGetObj_Decompress(CephContext* cct_,
  33                        RGWCompressionInfo* cs_info_,
  34                        bool partial_content_,
  35                        RGWGetObj_Filter* next);
  36   ~RGWGetObj_Decompress() override {}
  37
  38   int handle_data(bufferlist& bl, off_t bl_ofs, off_t bl_len) override;
  39   int fixup_range(off_t& ofs, off_t& end) override;
  40
  41 };
  42
  43 class RGWPutObj_Compress : public rgw::putobj::Pipe
  44 {
  45   CephContext* cct;
  46   bool compressed{false};
  47   CompressorRef compressor;
  48   boost::optional<int32_t> compressor_message;
  49   std::vector<compression_block> blocks;
  50 public:
  51   RGWPutObj_Compress(CephContext* cct_, CompressorRef compressor,
  52                      rgw::putobj::DataProcessor *next)
  53     : Pipe(next), cct(cct_), compressor(compressor) {}
  54
  55   int process(bufferlist&& data, uint64_t logical_offset) override;
  56
  57   bool is_compressed() { return compressed; }
  58   vector<compression_block>& get_compression_blocks() { return blocks; }
  59   boost::optional<int32_t> get_compressor_message() { return compressor_message; }
  60
  61 }; /* RGWPutObj_Compress */
  62
  63 #endif /* CEPH_RGW_COMPRESSION_H */