include/grpcpp/impl/codegen/slice.h

Go to the documentation of this file.

/*
 *
 * Copyright 2015 gRPC authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */
 
#ifndef GRPCPP_IMPL_CODEGEN_SLICE_H
#define GRPCPP_IMPL_CODEGEN_SLICE_H
 
// IWYU pragma: private, include <grpcpp/support/slice.h>
 
#include <grpc/impl/codegen/slice.h>
#include <grpcpp/impl/codegen/config.h>
#include <grpcpp/impl/codegen/core_codegen_interface.h>
#include <grpcpp/impl/codegen/string_ref.h>
 
namespace grpc {
 
class Slice final {
 public:
  Slice() : slice_(g_core_codegen_interface->grpc_empty_slice()) {}
  ~Slice() { g_core_codegen_interface->grpc_slice_unref(slice_); }
 
  enum AddRef { ADD_REF };
  Slice(grpc_slice slice, AddRef)
      : slice_(g_core_codegen_interface->grpc_slice_ref(slice)) {}
 
  enum StealRef { STEAL_REF };
  Slice(grpc_slice slice, StealRef) : slice_(slice) {}
 
  explicit Slice(size_t len)
      : slice_(g_core_codegen_interface->grpc_slice_malloc(len)) {}
 
  Slice(const void* buf, size_t len)
      : slice_(g_core_codegen_interface->grpc_slice_from_copied_buffer(
            reinterpret_cast<const char*>(buf), len)) {}
 
  /* NOLINTNEXTLINE(google-explicit-constructor) */
  Slice(const std::string& str)
      : slice_(g_core_codegen_interface->grpc_slice_from_copied_buffer(
            str.c_str(), str.length())) {}
 
  enum StaticSlice { STATIC_SLICE };
 
  Slice(const void* buf, size_t len, StaticSlice)
      : slice_(g_core_codegen_interface->grpc_slice_from_static_buffer(
            reinterpret_cast<const char*>(buf), len)) {}
 
  Slice(const Slice& other)
      : slice_(g_core_codegen_interface->grpc_slice_ref(other.slice_)) {}
 
  Slice(Slice&& other) noexcept : slice_(other.slice_) {
    other.slice_ = g_core_codegen_interface->grpc_empty_slice();
  }
 
  Slice& operator=(Slice other) {
    std::swap(slice_, other.slice_);
    return *this;
  }
 
  Slice(void* buf, size_t len, void (*destroy)(void*), void* user_data)
      : slice_(g_core_codegen_interface->grpc_slice_new_with_user_data(
            buf, len, destroy, user_data)) {}
 
  Slice(void* buf, size_t len, void (*destroy)(void*))
      : Slice(buf, len, destroy, buf) {}
 
  Slice(void* buf, size_t len, void (*destroy)(void*, size_t))
      : slice_(g_core_codegen_interface->grpc_slice_new_with_len(buf, len,
                                                                 destroy)) {}
 
  size_t size() const { return GRPC_SLICE_LENGTH(slice_); }
 
  const uint8_t* begin() const { return GRPC_SLICE_START_PTR(slice_); }
 
  const uint8_t* end() const { return GRPC_SLICE_END_PTR(slice_); }
 
  Slice sub(size_t begin, size_t end) const {
    return Slice(g_core_codegen_interface->grpc_slice_sub(slice_, begin, end),
                 STEAL_REF);
  }
 
  grpc_slice c_slice() const {
    return g_core_codegen_interface->grpc_slice_ref(slice_);
  }
 
 private:
  friend class ByteBuffer;
 
  grpc_slice slice_;
};
 
inline grpc::string_ref StringRefFromSlice(const grpc_slice* slice) {
  return grpc::string_ref(
      reinterpret_cast<const char*>(GRPC_SLICE_START_PTR(*slice)),
      GRPC_SLICE_LENGTH(*slice));
}
 
inline std::string StringFromCopiedSlice(grpc_slice slice) {
  return std::string(reinterpret_cast<char*>(GRPC_SLICE_START_PTR(slice)),
                     GRPC_SLICE_LENGTH(slice));
}
 
inline grpc_slice SliceReferencingString(const std::string& str) {
  return g_core_codegen_interface->grpc_slice_from_static_buffer(str.data(),
                                                                 str.length());
}
 
inline grpc_slice SliceFromCopiedString(const std::string& str) {
  return g_core_codegen_interface->grpc_slice_from_copied_buffer(str.data(),
                                                                 str.length());
}
 
}  // namespace grpc
 
#endif  // GRPCPP_IMPL_CODEGEN_SLICE_H