ceph/src/common/buffer_seastar.cc

   1 // -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
   2 // vim: ts=8 sw=2 smarttab
   3 /*
   4  * Ceph - scalable distributed file system
   5  *
   6  * Copyright (C) 2017 Red Hat, Inc.
   7  *
   8  * This is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU Lesser General Public
  10  * License version 2.1, as published by the Free Software
  11  * Foundation.  See file COPYING.
  12  *
  13  */
  14
  15 #include <seastar/core/sharded.hh>
  16
  17 #include "include/buffer_raw.h"
  18 #include "buffer_seastar.h"
  19
  20 using temporary_buffer = seastar::temporary_buffer<char>;
  21
  22 namespace ceph::buffer {
  23
  24 class raw_seastar_foreign_ptr : public raw {
  25   seastar::foreign_ptr<temporary_buffer> ptr;
  26  public:
  27   raw_seastar_foreign_ptr(temporary_buffer&& buf)
  28     : raw(buf.get_write(), buf.size()), ptr(std::move(buf)) {}
  29   raw* clone_empty() override {
  30     return create(len).release();
  31   }
  32 };
  33
  34 raw* create_foreign(temporary_buffer&& buf) {
  35   return new raw_seastar_foreign_ptr(std::move(buf));
  36 }
  37
  38 class raw_seastar_local_ptr : public raw {
  39   temporary_buffer buf;
  40  public:
  41   raw_seastar_local_ptr(temporary_buffer&& buf)
  42     : raw(buf.get_write(), buf.size()), buf(std::move(buf)) {}
  43   raw* clone_empty() override {
  44     return create(len).release();
  45   }
  46 };
  47
  48 raw* create(temporary_buffer&& buf) {
  49   return new raw_seastar_local_ptr(std::move(buf));
  50 }
  51
  52 // buffer::ptr conversions
  53
  54 ptr::operator seastar::temporary_buffer<char>() &
  55 {
  56   return {c_str(), _len, seastar::make_object_deleter(*this)};
  57 }
  58
  59 ptr::operator seastar::temporary_buffer<char>() &&
  60 {
  61   auto data = c_str();
  62   auto length = _len;
  63   return {data, length, seastar::make_object_deleter(std::move(*this))};
  64 }
  65
  66 // buffer::list conversions
  67
  68 list::operator seastar::net::packet() &&
  69 {
  70   seastar::net::packet p;
  71   p.reserve(_buffers.size());
  72   for (auto& ptr : _buffers) {
  73     // append each ptr as a temporary_buffer
  74     p = seastar::net::packet(std::move(p), std::move(ptr));
  75   }
  76   clear();
  77   return p;
  78 }
  79
  80 } // namespace ceph::buffer
  81
  82 namespace {
  83
  84 using ceph::buffer::raw;
  85 class raw_seastar_local_shared_ptr : public raw {
  86   temporary_buffer buf;
  87 public:
  88   raw_seastar_local_shared_ptr(temporary_buffer& buf)
  89     : raw(buf.get_write(), buf.size()), buf(buf.share()) {}
  90   raw* clone_empty() override {
  91     return ceph::buffer::create(len).release();
  92   }
  93 };
  94 }
  95
  96 buffer::ptr seastar_buffer_iterator::get_ptr(size_t len)
  97 {
  98   buffer::raw* r = new raw_seastar_local_shared_ptr{buf};
  99   buffer::ptr p{r};
 100   p.set_length(len);
 101   return p;
 102 }
 103
 104 buffer::ptr const_seastar_buffer_iterator::get_ptr(size_t len)
 105 {
 106   return buffer::ptr{ buffer::copy(get_pos_add(len), len) };
 107 }