diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-21 11:54:28 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-21 11:54:28 +0000 |
commit | e6918187568dbd01842d8d1d2c808ce16a894239 (patch) | |
tree | 64f88b554b444a49f656b6c656111a145cbbaa28 /src/cls/test_remote_reads/cls_test_remote_reads.cc | |
parent | Initial commit. (diff) | |
download | ceph-e6918187568dbd01842d8d1d2c808ce16a894239.tar.xz ceph-e6918187568dbd01842d8d1d2c808ce16a894239.zip |
Adding upstream version 18.2.2.upstream/18.2.2
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'src/cls/test_remote_reads/cls_test_remote_reads.cc')
-rw-r--r-- | src/cls/test_remote_reads/cls_test_remote_reads.cc | 87 |
1 files changed, 87 insertions, 0 deletions
diff --git a/src/cls/test_remote_reads/cls_test_remote_reads.cc b/src/cls/test_remote_reads/cls_test_remote_reads.cc new file mode 100644 index 000000000..33b0e9dc1 --- /dev/null +++ b/src/cls/test_remote_reads/cls_test_remote_reads.cc @@ -0,0 +1,87 @@ +/* + * This is an example RADOS object class that shows how to use remote reads. + */ + +#include "common/ceph_json.h" +#include "objclass/objclass.h" + +CLS_VER(1,0) +CLS_NAME(test_remote_reads) + +cls_handle_t h_class; +cls_method_handle_t h_test_read; +cls_method_handle_t h_test_gather; + +/** + * read data + */ +static int test_read(cls_method_context_t hctx, bufferlist *in, bufferlist *out) { + int r = cls_cxx_read(hctx, 0, 0, out); + if (r < 0) { + CLS_ERR("%s: error reading data", __PRETTY_FUNCTION__); + return r; + } + return 0; +} + +/** + * gather data from other objects using remote reads + */ +static int test_gather(cls_method_context_t hctx, bufferlist *in, bufferlist *out) { + std::map<std::string, bufferlist> src_obj_buffs; + int r = cls_cxx_get_gathered_data(hctx, &src_obj_buffs); + if (src_obj_buffs.empty()) { + // start remote reads + JSONParser parser; + bool b = parser.parse(in->c_str(), in->length()); + if (!b) { + CLS_ERR("%s: failed to parse json", __PRETTY_FUNCTION__); + return -EBADMSG; + } + auto *o_cls = parser.find_obj("cls"); + ceph_assert(o_cls); + std::string cls = o_cls->get_data_val().str; + + auto *o_method = parser.find_obj("method"); + ceph_assert(o_method); + std::string method = o_method->get_data_val().str; + + auto *o_pool = parser.find_obj("pool"); + ceph_assert(o_pool); + std::string pool = o_pool->get_data_val().str; + + auto *o_src_objects = parser.find_obj("src_objects"); + ceph_assert(o_src_objects); + auto src_objects_v = o_src_objects->get_array_elements(); + std::set<std::string> src_objects; + for (auto it = src_objects_v.begin(); it != src_objects_v.end(); it++) { + std::string oid_without_double_quotes = it->substr(1, it->size()-2); + src_objects.insert(oid_without_double_quotes); + } + r = cls_cxx_gather(hctx, src_objects, pool, cls.c_str(), method.c_str(), *in); + } else { + // write data gathered using remote reads + int offset = 0; + for (std::map<std::string, bufferlist>::iterator it = src_obj_buffs.begin(); it != src_obj_buffs.end(); it++) { + bufferlist bl= it->second; + r = cls_cxx_write(hctx, offset, bl.length(), &bl); + offset += bl.length(); + } + } + return r; +} + +CLS_INIT(test_remote_reads) +{ + CLS_LOG(0, "loading cls_test_remote_reads"); + + cls_register("test_remote_reads", &h_class); + + cls_register_cxx_method(h_class, "test_read", + CLS_METHOD_RD, + test_read, &h_test_read); + + cls_register_cxx_method(h_class, "test_gather", + CLS_METHOD_RD | CLS_METHOD_WR, + test_gather, &h_test_gather); +} |