initial code repo

[stor4nfv.git] / src / ceph / src / rgw / rgw_rest_client.h

diff --git a/src/ceph/src/rgw/rgw_rest_client.h b/src/ceph/src/rgw/rgw_rest_client.h
new file mode 100644 (file)
index 0000000..09393f8
--- /dev/null
+++ b/src/ceph/src/rgw/rgw_rest_client.h
@@ -0,0 +1,131 @@
+// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
+// vim: ts=8 sw=2 smarttab
+
+#ifndef CEPH_RGW_REST_CLIENT_H
+#define CEPH_RGW_REST_CLIENT_H
+
+#include "rgw_http_client.h"
+
+class RGWGetDataCB;
+
+class RGWRESTSimpleRequest : public RGWHTTPClient {
+protected:
+  int http_status;
+  int status;
+
+  string url;
+
+  map<string, string> out_headers;
+  param_vec_t params;
+
+  bufferlist::iterator *send_iter;
+
+  size_t max_response; /* we need this as we don't stream out response */
+  bufferlist response;
+
+  virtual int handle_header(const string& name, const string& val);
+  void append_param(string& dest, const string& name, const string& val);
+  void get_params_str(map<string, string>& extra_args, string& dest);
+
+  int sign_request(RGWAccessKey& key, RGWEnv& env, req_info& info);
+public:
+  RGWRESTSimpleRequest(CephContext *_cct, const string& _url, param_vec_t *_headers,
+               param_vec_t *_params) : RGWHTTPClient(_cct), http_status(0), status(0),
+                url(_url), send_iter(NULL),
+                max_response(0) {
+    set_headers(_headers);
+    set_params(_params);
+  }
+
+  void set_headers(param_vec_t *_headers) {
+    if (_headers)
+      headers = *_headers;
+  }
+
+  void set_params(param_vec_t *_params) {
+    if (_params)
+      params = *_params;
+  }
+
+  int receive_header(void *ptr, size_t len) override;
+  int receive_data(void *ptr, size_t len) override;
+  int send_data(void *ptr, size_t len) override;
+
+  bufferlist& get_response() { return response; }
+
+  int execute(RGWAccessKey& key, const char *method, const char *resource);
+  int forward_request(RGWAccessKey& key, req_info& info, size_t max_response, bufferlist *inbl, bufferlist *outbl);
+
+  map<string, string>& get_out_headers() { return out_headers; }
+
+  int get_http_status() { return http_status; }
+  int get_status();
+};
+
+
+class RGWRESTStreamWriteRequest : public RGWRESTSimpleRequest {
+  Mutex lock;
+  list<bufferlist> pending_send;
+  RGWGetDataCB *cb;
+  RGWHTTPManager http_manager;
+public:
+  int add_output_data(bufferlist& bl);
+  int send_data(void *ptr, size_t len) override;
+
+  RGWRESTStreamWriteRequest(CephContext *_cct, const string& _url, param_vec_t *_headers,
+               param_vec_t *_params) : RGWRESTSimpleRequest(_cct, _url, _headers, _params),
+                lock("RGWRESTStreamWriteRequest"), cb(NULL), http_manager(_cct) {}
+  ~RGWRESTStreamWriteRequest() override;
+  int put_obj_init(RGWAccessKey& key, rgw_obj& obj, uint64_t obj_size, map<string, bufferlist>& attrs);
+  int complete(string& etag, real_time *mtime);
+
+  RGWGetDataCB *get_out_cb() { return cb; }
+};
+
+class RGWRESTStreamRWRequest : public RGWRESTSimpleRequest {
+  Mutex lock;
+  RGWGetDataCB *cb;
+  bufferlist outbl;
+  bufferlist in_data;
+  size_t chunk_ofs;
+  size_t ofs;
+  RGWHTTPManager http_manager;
+  const char *method;
+  uint64_t write_ofs;
+protected:
+  int handle_header(const string& name, const string& val) override;
+public:
+  int send_data(void *ptr, size_t len) override;
+  int receive_data(void *ptr, size_t len) override;
+
+  RGWRESTStreamRWRequest(CephContext *_cct, const char *_method, const string& _url, RGWGetDataCB *_cb,
+               param_vec_t *_headers, param_vec_t *_params) : RGWRESTSimpleRequest(_cct, _url, _headers, _params),
+                lock("RGWRESTStreamReadRequest"), cb(_cb),
+                chunk_ofs(0), ofs(0), http_manager(_cct), method(_method), write_ofs(0) {
+  }
+  virtual ~RGWRESTStreamRWRequest() override {}
+  int send_request(RGWAccessKey& key, map<string, string>& extra_headers, rgw_obj& obj, RGWHTTPManager *mgr = NULL);
+  int send_request(RGWAccessKey *key, map<string, string>& extra_headers, const string& resource, bufferlist *send_data = NULL /* optional input data */, RGWHTTPManager *mgr = NULL);
+  int complete_request(string& etag, real_time *mtime, uint64_t *psize, map<string, string>& attrs);
+
+  void set_outbl(bufferlist& _outbl) {
+    outbl.swap(_outbl);
+  }
+
+  void set_in_cb(RGWGetDataCB *_cb) { cb = _cb; }
+};
+
+class RGWRESTStreamReadRequest : public RGWRESTStreamRWRequest {
+public:
+  RGWRESTStreamReadRequest(CephContext *_cct, const string& _url, RGWGetDataCB *_cb, param_vec_t *_headers,
+               param_vec_t *_params) : RGWRESTStreamRWRequest(_cct, "GET", _url, _cb, _headers, _params) {}
+};
+
+class RGWRESTStreamHeadRequest : public RGWRESTStreamRWRequest {
+public:
+  RGWRESTStreamHeadRequest(CephContext *_cct, const string& _url, RGWGetDataCB *_cb, param_vec_t *_headers,
+               param_vec_t *_params) : RGWRESTStreamRWRequest(_cct, "HEAD", _url, _cb, _headers, _params) {}
+};
+
+#endif
+