172 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			C++
		
	
	
	
		
		
			
		
	
	
			172 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			C++
		
	
	
	
|  | //
 | ||
|  | // detail/object_pool.hpp
 | ||
|  | // ~~~~~~~~~~~~~~~~~~~~~~
 | ||
|  | //
 | ||
|  | // Copyright (c) 2003-2023 Christopher M. Kohlhoff (chris at kohlhoff dot com)
 | ||
|  | //
 | ||
|  | // Distributed under the Boost Software License, Version 1.0. (See accompanying
 | ||
|  | // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
 | ||
|  | //
 | ||
|  | 
 | ||
|  | #ifndef ASIO_DETAIL_OBJECT_POOL_HPP
 | ||
|  | #define ASIO_DETAIL_OBJECT_POOL_HPP
 | ||
|  | 
 | ||
|  | #if defined(_MSC_VER) && (_MSC_VER >= 1200)
 | ||
|  | # pragma once
 | ||
|  | #endif // defined(_MSC_VER) && (_MSC_VER >= 1200)
 | ||
|  | 
 | ||
|  | #include "asio/detail/noncopyable.hpp"
 | ||
|  | 
 | ||
|  | #include "asio/detail/push_options.hpp"
 | ||
|  | 
 | ||
|  | namespace asio { | ||
|  | namespace detail { | ||
|  | 
 | ||
|  | template <typename Object> | ||
|  | class object_pool; | ||
|  | 
 | ||
|  | class object_pool_access | ||
|  | { | ||
|  | public: | ||
|  |   template <typename Object> | ||
|  |   static Object* create() | ||
|  |   { | ||
|  |     return new Object; | ||
|  |   } | ||
|  | 
 | ||
|  |   template <typename Object, typename Arg> | ||
|  |   static Object* create(Arg arg) | ||
|  |   { | ||
|  |     return new Object(arg); | ||
|  |   } | ||
|  | 
 | ||
|  |   template <typename Object> | ||
|  |   static void destroy(Object* o) | ||
|  |   { | ||
|  |     delete o; | ||
|  |   } | ||
|  | 
 | ||
|  |   template <typename Object> | ||
|  |   static Object*& next(Object* o) | ||
|  |   { | ||
|  |     return o->next_; | ||
|  |   } | ||
|  | 
 | ||
|  |   template <typename Object> | ||
|  |   static Object*& prev(Object* o) | ||
|  |   { | ||
|  |     return o->prev_; | ||
|  |   } | ||
|  | }; | ||
|  | 
 | ||
|  | template <typename Object> | ||
|  | class object_pool | ||
|  |   : private noncopyable | ||
|  | { | ||
|  | public: | ||
|  |   // Constructor.
 | ||
|  |   object_pool() | ||
|  |     : live_list_(0), | ||
|  |       free_list_(0) | ||
|  |   { | ||
|  |   } | ||
|  | 
 | ||
|  |   // Destructor destroys all objects.
 | ||
|  |   ~object_pool() | ||
|  |   { | ||
|  |     destroy_list(live_list_); | ||
|  |     destroy_list(free_list_); | ||
|  |   } | ||
|  | 
 | ||
|  |   // Get the object at the start of the live list.
 | ||
|  |   Object* first() | ||
|  |   { | ||
|  |     return live_list_; | ||
|  |   } | ||
|  | 
 | ||
|  |   // Allocate a new object.
 | ||
|  |   Object* alloc() | ||
|  |   { | ||
|  |     Object* o = free_list_; | ||
|  |     if (o) | ||
|  |       free_list_ = object_pool_access::next(free_list_); | ||
|  |     else | ||
|  |       o = object_pool_access::create<Object>(); | ||
|  | 
 | ||
|  |     object_pool_access::next(o) = live_list_; | ||
|  |     object_pool_access::prev(o) = 0; | ||
|  |     if (live_list_) | ||
|  |       object_pool_access::prev(live_list_) = o; | ||
|  |     live_list_ = o; | ||
|  | 
 | ||
|  |     return o; | ||
|  |   } | ||
|  | 
 | ||
|  |   // Allocate a new object with an argument.
 | ||
|  |   template <typename Arg> | ||
|  |   Object* alloc(Arg arg) | ||
|  |   { | ||
|  |     Object* o = free_list_; | ||
|  |     if (o) | ||
|  |       free_list_ = object_pool_access::next(free_list_); | ||
|  |     else | ||
|  |       o = object_pool_access::create<Object>(arg); | ||
|  | 
 | ||
|  |     object_pool_access::next(o) = live_list_; | ||
|  |     object_pool_access::prev(o) = 0; | ||
|  |     if (live_list_) | ||
|  |       object_pool_access::prev(live_list_) = o; | ||
|  |     live_list_ = o; | ||
|  | 
 | ||
|  |     return o; | ||
|  |   } | ||
|  | 
 | ||
|  |   // Free an object. Moves it to the free list. No destructors are run.
 | ||
|  |   void free(Object* o) | ||
|  |   { | ||
|  |     if (live_list_ == o) | ||
|  |       live_list_ = object_pool_access::next(o); | ||
|  | 
 | ||
|  |     if (object_pool_access::prev(o)) | ||
|  |     { | ||
|  |       object_pool_access::next(object_pool_access::prev(o)) | ||
|  |         = object_pool_access::next(o); | ||
|  |     } | ||
|  | 
 | ||
|  |     if (object_pool_access::next(o)) | ||
|  |     { | ||
|  |       object_pool_access::prev(object_pool_access::next(o)) | ||
|  |         = object_pool_access::prev(o); | ||
|  |     } | ||
|  | 
 | ||
|  |     object_pool_access::next(o) = free_list_; | ||
|  |     object_pool_access::prev(o) = 0; | ||
|  |     free_list_ = o; | ||
|  |   } | ||
|  | 
 | ||
|  | private: | ||
|  |   // Helper function to destroy all elements in a list.
 | ||
|  |   void destroy_list(Object* list) | ||
|  |   { | ||
|  |     while (list) | ||
|  |     { | ||
|  |       Object* o = list; | ||
|  |       list = object_pool_access::next(o); | ||
|  |       object_pool_access::destroy(o); | ||
|  |     } | ||
|  |   } | ||
|  | 
 | ||
|  |   // The list of live objects.
 | ||
|  |   Object* live_list_; | ||
|  | 
 | ||
|  |   // The free list.
 | ||
|  |   Object* free_list_; | ||
|  | }; | ||
|  | 
 | ||
|  | } // namespace detail
 | ||
|  | } // namespace asio
 | ||
|  | 
 | ||
|  | #include "asio/detail/pop_options.hpp"
 | ||
|  | 
 | ||
|  | #endif // ASIO_DETAIL_OBJECT_POOL_HPP
 |