[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