ceph/src/rgw/rgw_loadgen.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 #pragma once
   5
   6 #include <map>
   7 #include <string>
   8
   9 #include "rgw_client_io.h"
  10
  11
  12 struct RGWLoadGenRequestEnv {
  13   int port;
  14   uint64_t content_length;
  15   std::string content_type;
  16   std::string request_method;
  17   std::string uri;
  18   std::string query_string;
  19   std::string date_str;
  20
  21   std::map<std::string, std::string> headers;
  22
  23   RGWLoadGenRequestEnv()
  24     : port(0),
  25       content_length(0) {
  26   }
  27
  28   void set_date(utime_t& tm);
  29   int sign(const DoutPrefixProvider *dpp, RGWAccessKey& access_key);
  30 };
  31
  32 /* XXX does RGWLoadGenIO actually want to perform stream/HTTP I/O,
  33  * or (e.g) are these NOOPs? */
  34 class RGWLoadGenIO : public rgw::io::RestfulClient
  35 {
  36   uint64_t left_to_read;
  37   RGWLoadGenRequestEnv* req;
  38   RGWEnv env;
  39
  40   int init_env(CephContext *cct) override;
  41   size_t read_data(char *buf, size_t len);
  42   size_t write_data(const char *buf, size_t len);
  43
  44 public:
  45   explicit RGWLoadGenIO(RGWLoadGenRequestEnv* const req)
  46     : left_to_read(0),
  47       req(req) {
  48   }
  49
  50   size_t send_status(int status, const char *status_name) override;
  51   size_t send_100_continue() override;
  52   size_t send_header(const std::string_view& name,
  53                      const std::string_view& value) override;
  54   size_t complete_header() override;
  55   size_t send_content_length(uint64_t len) override;
  56
  57   size_t recv_body(char* buf, size_t max) override {
  58     return read_data(buf, max);
  59   }
  60
  61   size_t send_body(const char* buf, size_t len) override {
  62     return write_data(buf, len);
  63   }
  64
  65   void flush() override;
  66
  67   RGWEnv& get_env() noexcept override {
  68     return env;
  69   }
  70
  71   size_t complete_request() override;
  72 };