[EventEngine] Add virtual a MakeSlice method into MemoryAllocatorImpl and move its default implementation to GrpcMemoryAllocatorImpl
PiperOrigin-RevId: 589155443pull/35224/head
Vignesh Babu
1 year ago
committed by
Copybara-Service
parent
257efc0375
commit
f0d7df79da
22 changed files with 90 additions and 111 deletions
@ -1,74 +0,0 @@ |
|||||||
// Copyright 2021 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.
|
|
||||||
|
|
||||||
#include <grpc/support/port_platform.h> |
|
||||||
|
|
||||||
#include <stdint.h> |
|
||||||
#include <stdlib.h> |
|
||||||
|
|
||||||
#include <memory> |
|
||||||
#include <new> |
|
||||||
#include <utility> |
|
||||||
|
|
||||||
#include <grpc/event_engine/memory_allocator.h> |
|
||||||
#include <grpc/event_engine/memory_request.h> |
|
||||||
#include <grpc/slice.h> |
|
||||||
|
|
||||||
#include "src/core/lib/slice/slice_refcount.h" |
|
||||||
|
|
||||||
namespace grpc_event_engine { |
|
||||||
namespace experimental { |
|
||||||
|
|
||||||
namespace { |
|
||||||
|
|
||||||
// Reference count for a slice allocated by MemoryAllocator::MakeSlice.
|
|
||||||
// Takes care of releasing memory back when the slice is destroyed.
|
|
||||||
class SliceRefCount : public grpc_slice_refcount { |
|
||||||
public: |
|
||||||
SliceRefCount(std::shared_ptr<internal::MemoryAllocatorImpl> allocator, |
|
||||||
size_t size) |
|
||||||
: grpc_slice_refcount(Destroy), |
|
||||||
allocator_(std::move(allocator)), |
|
||||||
size_(size) { |
|
||||||
// Nothing to do here.
|
|
||||||
} |
|
||||||
~SliceRefCount() { allocator_->Release(size_); } |
|
||||||
|
|
||||||
private: |
|
||||||
static void Destroy(grpc_slice_refcount* p) { |
|
||||||
auto* rc = static_cast<SliceRefCount*>(p); |
|
||||||
rc->~SliceRefCount(); |
|
||||||
free(rc); |
|
||||||
} |
|
||||||
|
|
||||||
std::shared_ptr<internal::MemoryAllocatorImpl> allocator_; |
|
||||||
size_t size_; |
|
||||||
}; |
|
||||||
|
|
||||||
} // namespace
|
|
||||||
|
|
||||||
grpc_slice MemoryAllocator::MakeSlice(MemoryRequest request) { |
|
||||||
auto size = Reserve(request.Increase(sizeof(SliceRefCount))); |
|
||||||
void* p = malloc(size); |
|
||||||
new (p) SliceRefCount(allocator_, size); |
|
||||||
grpc_slice slice; |
|
||||||
slice.refcount = static_cast<SliceRefCount*>(p); |
|
||||||
slice.data.refcounted.bytes = |
|
||||||
static_cast<uint8_t*>(p) + sizeof(SliceRefCount); |
|
||||||
slice.data.refcounted.length = size - sizeof(SliceRefCount); |
|
||||||
return slice; |
|
||||||
} |
|
||||||
|
|
||||||
} // namespace experimental
|
|
||||||
} // namespace grpc_event_engine
|
|
||||||
Loading…
Reference in new issue