zoukankan      html  css  js  c++  java
  • C++ ObjectPool_VS2010_非线程安全及线程安全版本

    // Test.cpp : 定义控制台应用程序的入口点。
    //
    
    #include "stdafx.h"
    #include <memory>
    #include <iostream>
    #include <stack>
    #include <queue>
    using namespace std;
    
    #pragma once
    #include <memory>
    #include <iostream>
    #include <stack>
    using namespace std;
    
    template <class T> class PoolItemProxy;
    template <class T> class PoolItem ;
    template <class T> class ObjectPool;
    template <class T> class ObjectPoolTS ;
    //typedef PoolItem<T> PoolItem<T>;
    template <class T> class PoolItemProxy
    {
    private:
    	ObjectPool<T>* PoolPtr;
    	T* ItemPtr;
    
    	PoolItemProxy(ObjectPool<T>* poolPtr, T* itemPtr)
    	{
    		this->PoolPtr = poolPtr;
    		this->ItemPtr = itemPtr;
    		cout << "object from pool" << endl;
    
    	}
    
    public:
    	//PoolItemProxy(PoolItemProxy<T>& item)
    	//{
    	//	this->PoolPtr = item.m_pFromPool;
    	//	this->ItemPtr = item.release();
    	//}
    
    	PoolItemProxy(PoolItemProxy<T>& item)
    	{
    		this->PoolPtr = item.PoolPtr;
    		this->ItemPtr = item.ItemPtr;
    	}
    
    	virtual ~PoolItemProxy()
    	{
    		if(PoolPtr && ItemPtr)
    		{
    			PoolPtr->Release(*this);
    			ItemPtr = NULL;
    			PoolPtr = NULL;
    			cout << "release to pool!" << endl;
    		}
    	}
    
    	T* get()
    	{
    		return ItemPtr;
    	}
    
    
    
    	T* operator->()
    	{
    		return ItemPtr;
    	}
    
    
    	friend class PoolItemProxy<T>;
    	friend class PoolItem<T>;
    	friend class ObjectPool<T>;
    };
    
    
    template <class T> class ObjectPool
    {
    
    private: 
    	int m_nCountLower;
    	std::stack<T*> m_pool;
    public:
    	ObjectPool(int  nCountLower)
    	{
    		m_nCountLower  = nCountLower;
    		for(int i = 0; i < nCountLower; ++i)
    		{
    			m_pool.push(new T());
    		}
    	}
    
    	virtual ~ObjectPool()
    	{
    		while(!m_pool.empty())
    		{
    			delete m_pool.top();
    			m_pool.pop();
    		}
    	}
    
    
    	virtual PoolItem<T> Get()
    	{
    		if(m_pool.size() > 0)
    		{
    			PoolItem<T> item(new PoolItemProxy<T>((ObjectPool<T>*)this, m_pool.top()));
    			m_pool.pop();
    			return item;
    		}
    		else
    		{
    			PoolItem<T> item(new PoolItemProxy<T>((ObjectPool<T>*)this, new T()));
    			return item;
    		}
    	}
    
    	virtual void Release(PoolItemProxy<T>& item)
    	{
    		if(m_pool.size() >= m_nCountLower)
    		{
    			//直接Delete掉
    			delete item.ItemPtr;
    		}
    		else
    		{
    			m_pool.push(item.ItemPtr);
    			
    		}
    	}
    
    
    
    
    };
    
    template <class T> class PoolItem : public std::auto_ptr<PoolItemProxy<T>>
    {
    public:
    	PoolItem(PoolItemProxy<T>* pitem)
    		:auto_ptr<PoolItemProxy<T>>(pitem)
    	{
    
    	}
    
    	T* operator->()
    	{
    		return (auto_ptr<PoolItemProxy<T>>::operator ->())->ItemPtr;
    	}
    
    	T* get()
    	{
    		return (auto_ptr<PoolItemProxy<T>>::operator ->())->ItemPtr;
    	}
    
    };
    
    
    
    
    template <class T> class  ObjectPoolTS : public ObjectPool<T>
    {
    private:
    	CCriticalSection m_lock;
    public:
    	ObjectPoolTS(int nCountLower)
    		:ObjectPool<T>(nCountLower),m_lock()
    	{
    
    	}
    
    	virtual ~ObjectPoolTS()
    	{
    
    	}
    
    
    	virtual PoolItem<T> Get()
    	{
    
    		m_lock.Lock();
    		PoolItem<T>& ret = ObjectPool<T>::Get();
    		m_lock.Unlock();
    		return ret;
    
    	}
    
    	virtual void Release(PoolItemProxy<T>& item)
    	{
    		m_lock.Lock();
    		ObjectPool<T>::Release(item);
    		m_lock.Unlock();
    	}
    };
    
    class Data
    {
    	static int i;
    public:
    	Data()
    	{
    		m_i = ++i;
    		cout << "data created" << m_i << endl;
    	}
    
    	virtual ~Data()
    	{
    		cout << "date deleted"  << m_i << endl;
    	}
    
    	int m_i;
    };
    
    int Data::i = 0;
    
    void TestArgs1(PoolItem<Data>& item)
    {
    	cout << item->m_i << endl;
    
    }
    void TestArgs2(PoolItem<Data> item)
    {
    	cout << item->m_i << endl;
    
    }
    void TestArgs3(Data* item)
    {
    	cout << item->m_i << endl;
    }
    
    int _tmain(int argc, _TCHAR* argv[])
    {
    	
    
    	char c;
    	{
    		ObjectPoolTS<Data> pool(5);
    		{
    			PoolItem<Data> item = pool.Get();
    			PoolItem<Data> item1 = pool.Get();
    			PoolItem<Data> item2 = pool.Get();
    			PoolItem<Data> item3 = pool.Get();
    			PoolItem<Data> item4 = pool.Get();
    			PoolItem<Data> item5 = pool.Get();
    			cin.get(c);
    
    
    			PoolItem<Data> item6(item);
    			
    
    		}
    
    		{
    			PoolItem<Data> item = pool.Get();
    			PoolItem<Data> item1 = pool.Get();
    			PoolItem<Data> item2 = pool.Get();
    			cin.get(c);
    
    		}
    
    		{
    			PoolItem<Data> item = pool.Get();
    			TestArgs1(item);
    
    			TestArgs3(item.get());  //OK 
    
    			TestArgs2(item);
    
    			__assume(item.get() == NULL);
    		}
    
    		{
    			//OK
    			std::vector<PoolItem<Data>> arrNode;
    			arrNode.push_back(pool.Get());
    			arrNode.push_back(pool.Get());
    			cin.get(c);
    
    		}
    
    		cin.get(c);
    	}
    	
    
    	cin.get(c);
    }
    
    

    //程序输出

    data created1
    data created2
    data created3
    data created4
    data created5
    object from pool
    object from pool
    object from pool
    object from pool
    object from pool
    data created6
    object from pool

    release to pool!
    release to pool!
    release to pool!
    release to pool!
    release to pool!
    date deleted4
    release to pool!
    object from pool
    object from pool
    object from pool

    release to pool!
    release to pool!
    release to pool!
    object from pool
    3
    3
    3
    release to pool!
    object from pool
    object from pool

    release to pool!
    release to pool!


    date deleted2
    date deleted3
    date deleted1
    date deleted6
    date deleted5

  • 相关阅读:
    gerrit 在git review的时候碰上miss unkown + hash值
    centos7 rc.local脚本执行不成功
    python脚本之日期格式显示
    redis集群本地搭建
    php安装与注意事项
    nginx理解--如何处理一个请求
    数据同步 rsync+notify架构
    gitlab+gerrit+jenkins代码托管、审核、持续集成架构搭建
    RHEL6关于Header V3 DSA signature: NOKEY, key ID错误解决方法
    python脚本之traceroute生成路由跟踪图片
  • 原文地址:https://www.cnblogs.com/evlon/p/objectpool.html
Copyright © 2011-2022 走看看