merge ObjectPool changes from develop

This commit is contained in:
Günter Obiltschnig
2017-11-08 20:43:46 +01:00
parent 88ce18e198
commit 56c9ccb7ca

View File

@@ -20,6 +20,7 @@
#include "Poco/Poco.h" #include "Poco/Poco.h"
#include "Poco/Mutex.h" #include "Poco/Mutex.h"
#include "Poco/Condition.h"
#include "Poco/AutoPtr.h" #include "Poco/AutoPtr.h"
#include "Poco/SharedPtr.h" #include "Poco/SharedPtr.h"
#include <vector> #include <vector>
@@ -38,7 +39,7 @@ class PoolableObjectFactory
/// a policy class to change the behavior of the ObjectPool when /// a policy class to change the behavior of the ObjectPool when
/// creating new objects, returning used objects back to the pool /// creating new objects, returning used objects back to the pool
/// and destroying objects, when the pool itself is destroyed or /// and destroying objects, when the pool itself is destroyed or
/// shrinked. /// shrunk.
{ {
public: public:
P createObject() P createObject()
@@ -155,7 +156,7 @@ class ObjectPool
/// removed from the pool, activated (using the factory) and returned. /// removed from the pool, activated (using the factory) and returned.
/// - Otherwise, if the peak capacity of the pool has not yet been reached, /// - Otherwise, if the peak capacity of the pool has not yet been reached,
/// a new object is created and activated, using the object factory, and returned. /// a new object is created and activated, using the object factory, and returned.
/// - If the peak capacity has already been reached, null is returned. /// - If the peak capacity has already been reached, null is returned after timeout.
/// ///
/// When an object is returned to the pool: /// When an object is returned to the pool:
/// - If the object is valid (checked by calling validateObject() /// - If the object is valid (checked by calling validateObject()
@@ -205,11 +206,11 @@ public:
} }
} }
P borrowObject() P borrowObject(long timeoutMilliseconds = 0)
/// Obtains an object from the pool, or creates a new object if /// Obtains an object from the pool, or creates a new object if
/// possible. /// possible.
/// ///
/// Returns null if no object is available. /// Returns null if no object is available after timeout.
/// ///
/// If activating the object fails, the object is destroyed and /// If activating the object fails, the object is destroyed and
/// the exception is passed on to the caller. /// the exception is passed on to the caller.
@@ -222,15 +223,29 @@ public:
_pool.pop_back(); _pool.pop_back();
return activateObject(pObject); return activateObject(pObject);
} }
else if (_size < _peakCapacity)
if (_size >= _peakCapacity)
{ {
if (timeoutMilliseconds == 0)
{
return 0;
}
while (_size >= _peakCapacity)
{
if ( !_availableCondition.tryWait(_mutex, timeoutMilliseconds))
{
// timeout
return 0;
}
}
}
// _size < _peakCapacity
P pObject = _factory.createObject(); P pObject = _factory.createObject();
activateObject(pObject); activateObject(pObject);
_size++; _size++;
return pObject; return pObject;
} }
else return 0;
}
void returnObject(P pObject) void returnObject(P pObject)
/// Returns an object to the pool. /// Returns an object to the pool.
@@ -254,6 +269,7 @@ public:
} }
_factory.destroyObject(pObject); _factory.destroyObject(pObject);
_size--; _size--;
_availableCondition.signal();
} }
std::size_t capacity() const std::size_t capacity() const
@@ -306,6 +322,7 @@ private:
std::size_t _size; std::size_t _size;
std::vector<P> _pool; std::vector<P> _pool;
mutable Poco::FastMutex _mutex; mutable Poco::FastMutex _mutex;
Poco::Condition _availableCondition;
}; };