diff options
Diffstat (limited to 'libtransport/src/hicn/transport/utils/object_pool.h')
-rwxr-xr-x | libtransport/src/hicn/transport/utils/object_pool.h | 76 |
1 files changed, 76 insertions, 0 deletions
diff --git a/libtransport/src/hicn/transport/utils/object_pool.h b/libtransport/src/hicn/transport/utils/object_pool.h new file mode 100755 index 000000000..d4d8e18d6 --- /dev/null +++ b/libtransport/src/hicn/transport/utils/object_pool.h @@ -0,0 +1,76 @@ +/* + * Copyright (c) 2017-2019 Cisco and/or its affiliates. + * 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. + */ + +#pragma once + +// TODO +#include <hicn/transport/utils/spinlock.h> + +#include <deque> +#include <memory> +#include <mutex> + +namespace utils { + +template <typename T> +class ObjectPool { + class ObjectDeleter { + public: + ObjectDeleter(ObjectPool<T> *pool = nullptr) : pool_(pool) {} + + void operator()(T *t) { + if (pool_) { + pool_->add(t); + } else { + delete t; + } + } + + private: + ObjectPool<T> *pool_; + }; + + public: + using Ptr = std::unique_ptr<T, ObjectDeleter>; + + ObjectPool() {} + + std::pair<bool, Ptr> get() { + if (object_pool_.empty()) { + return std::make_pair<bool, Ptr>(false, makePtr(nullptr)); + } + + utils::SpinLock::Acquire locked(object_pool_lock_); + auto ret = std::move(object_pool_.front()); + object_pool_.pop_front(); + return std::make_pair<bool, Ptr>(true, std::move(ret)); + } + + void add(T *object) { + utils::SpinLock::Acquire locked(object_pool_lock_); + object_pool_.emplace_back(makePtr(object)); + } + + Ptr makePtr(T *object) { return Ptr(object, ObjectDeleter(this)); } + + private: + // No copies + ObjectPool(const ObjectPool &other) = delete; + + utils::SpinLock object_pool_lock_; + std::deque<Ptr> object_pool_; +}; + +} // namespace utils
\ No newline at end of file |