]>
Commit | Line | Data |
---|---|---|
7c673cae FG |
1 | // -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- |
2 | // vim: ts=8 sw=2 smarttab | |
3 | ||
4 | #ifndef CEPH_RGW_LOADGEN_H | |
5 | #define CEPH_RGW_LOADGEN_H | |
6 | ||
7 | #include <map> | |
8 | #include <string> | |
9 | ||
10 | #include "rgw_client_io.h" | |
11 | ||
12 | ||
13 | struct RGWLoadGenRequestEnv { | |
14 | int port; | |
15 | uint64_t content_length; | |
16 | std::string content_type; | |
17 | std::string request_method; | |
18 | std::string uri; | |
19 | std::string query_string; | |
20 | std::string date_str; | |
21 | ||
22 | std::map<std::string, std::string> headers; | |
23 | ||
24 | RGWLoadGenRequestEnv() | |
25 | : port(0), | |
26 | content_length(0) { | |
27 | } | |
28 | ||
29 | void set_date(utime_t& tm); | |
30 | int sign(RGWAccessKey& access_key); | |
31 | }; | |
32 | ||
33 | /* XXX does RGWLoadGenIO actually want to perform stream/HTTP I/O, | |
34 | * or (e.g) are these NOOPs? */ | |
35 | class RGWLoadGenIO : public rgw::io::RestfulClient | |
36 | { | |
37 | uint64_t left_to_read; | |
38 | RGWLoadGenRequestEnv* req; | |
39 | RGWEnv env; | |
40 | ||
41 | void init_env(CephContext *cct) override; | |
42 | size_t read_data(char *buf, size_t len); | |
43 | size_t write_data(const char *buf, size_t len); | |
44 | ||
45 | public: | |
46 | explicit RGWLoadGenIO(RGWLoadGenRequestEnv* const req) | |
47 | : left_to_read(0), | |
48 | req(req) { | |
49 | } | |
50 | ||
51 | size_t send_status(int status, const char *status_name) override; | |
52 | size_t send_100_continue() override; | |
53 | size_t send_header(const boost::string_ref& name, | |
54 | const boost::string_ref& value) override; | |
55 | size_t complete_header() override; | |
56 | size_t send_content_length(uint64_t len) override; | |
57 | ||
58 | size_t recv_body(char* buf, size_t max) override { | |
59 | return read_data(buf, max); | |
60 | } | |
61 | ||
62 | size_t send_body(const char* buf, size_t len) override { | |
63 | return write_data(buf, len); | |
64 | } | |
65 | ||
66 | void flush() override; | |
67 | ||
68 | RGWEnv& get_env() noexcept override { | |
69 | return env; | |
70 | } | |
71 | ||
72 | size_t complete_request() override; | |
73 | }; | |
74 | ||
75 | #endif |