v1/__memory_resource/synchronized_pool_resource.h

0001 //===----------------------------------------------------------------------===//
0002 //
0003 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
0004 // See https://llvm.org/LICENSE.txt for license information.
0005 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
0006 //
0007 //===----------------------------------------------------------------------===//
0008
0009 #ifndef _LIBCPP___MEMORY_RESOURCE_SYNCHRONIZED_POOL_RESOURCE_H
0010 #define _LIBCPP___MEMORY_RESOURCE_SYNCHRONIZED_POOL_RESOURCE_H
0011
0012 #include <__config>
0013 #include <__cstddef/size_t.h>
0014 #include <__memory_resource/memory_resource.h>
0015 #include <__memory_resource/pool_options.h>
0016 #include <__memory_resource/unsynchronized_pool_resource.h>
0017 #include <__mutex/mutex.h>
0018 #include <__mutex/unique_lock.h>
0019
0020 #if !defined(_LIBCPP_HAS_NO_PRAGMA_SYSTEM_HEADER)
0021 #  pragma GCC system_header
0022 #endif
0023
0024 #if _LIBCPP_STD_VER >= 17
0025
0026 _LIBCPP_BEGIN_NAMESPACE_STD
0027
0028 namespace pmr {
0029
0030 // [mem.res.pool.overview]
0031
0032 class _LIBCPP_AVAILABILITY_PMR _LIBCPP_EXPORTED_FROM_ABI synchronized_pool_resource : public memory_resource {
0033 public:
0034   _LIBCPP_HIDE_FROM_ABI synchronized_pool_resource(const pool_options& __opts, memory_resource* __upstream)
0035       : __unsync_(__opts, __upstream) {}
0036
0037   _LIBCPP_HIDE_FROM_ABI synchronized_pool_resource()
0038       : synchronized_pool_resource(pool_options(), get_default_resource()) {}
0039
0040   _LIBCPP_HIDE_FROM_ABI explicit synchronized_pool_resource(memory_resource* __upstream)
0041       : synchronized_pool_resource(pool_options(), __upstream) {}
0042
0043   _LIBCPP_HIDE_FROM_ABI explicit synchronized_pool_resource(const pool_options& __opts)
0044       : synchronized_pool_resource(__opts, get_default_resource()) {}
0045
0046   synchronized_pool_resource(const synchronized_pool_resource&) = delete;
0047
0048   _LIBCPP_HIDE_FROM_ABI_VIRTUAL ~synchronized_pool_resource() override = default;
0049
0050   synchronized_pool_resource& operator=(const synchronized_pool_resource&) = delete;
0051
0052   _LIBCPP_HIDE_FROM_ABI void release() {
0053 #  if _LIBCPP_HAS_THREADS
0054     unique_lock<mutex> __lk(__mut_);
0055 #  endif
0056     __unsync_.release();
0057   }
0058
0059   _LIBCPP_HIDE_FROM_ABI memory_resource* upstream_resource() const { return __unsync_.upstream_resource(); }
0060
0061   _LIBCPP_HIDE_FROM_ABI pool_options options() const { return __unsync_.options(); }
0062
0063 protected:
0064   _LIBCPP_HIDE_FROM_ABI_VIRTUAL void* do_allocate(size_t __bytes, size_t __align) override {
0065 #  if _LIBCPP_HAS_THREADS
0066     unique_lock<mutex> __lk(__mut_);
0067 #  endif
0068     return __unsync_.allocate(__bytes, __align);
0069   }
0070
0071   _LIBCPP_HIDE_FROM_ABI_VIRTUAL void do_deallocate(void* __p, size_t __bytes, size_t __align) override {
0072 #  if _LIBCPP_HAS_THREADS
0073     unique_lock<mutex> __lk(__mut_);
0074 #  endif
0075     return __unsync_.deallocate(__p, __bytes, __align);
0076   }
0077
0078   bool do_is_equal(const memory_resource& __other) const noexcept override; // key function
0079
0080 private:
0081 #  if _LIBCPP_HAS_THREADS
0082   mutex __mut_;
0083 #  endif
0084   unsynchronized_pool_resource __unsync_;
0085 };
0086
0087 } // namespace pmr
0088
0089 _LIBCPP_END_NAMESPACE_STD
0090
0091 #endif // _LIBCPP_STD_VER >= 17
0092
0093 #endif // _LIBCPP___MEMORY_RESOURCE_SYNCHRONIZED_POOL_RESOURCE_H