SF 1939071 and 1928786

This commit is contained in:
Aleksandar Fabijanic
2008-04-11 01:37:49 +00:00
parent 32c38a9197
commit 89ca50a581
14 changed files with 548 additions and 41 deletions

View File

@@ -1699,6 +1699,10 @@
RelativePath=".\src\ThreadPool.cpp" RelativePath=".\src\ThreadPool.cpp"
> >
</File> </File>
<File
RelativePath=".\src\ThreadTarget.cpp"
>
</File>
<File <File
RelativePath=".\src\Timer.cpp" RelativePath=".\src\Timer.cpp"
> >
@@ -1831,6 +1835,10 @@
RelativePath=".\include\Poco\ThreadPool.h" RelativePath=".\include\Poco\ThreadPool.h"
> >
</File> </File>
<File
RelativePath=".\include\Poco\ThreadTarget.h"
>
</File>
<File <File
RelativePath=".\include\Poco\Timer.h" RelativePath=".\include\Poco\Timer.h"
> >

View File

@@ -1682,6 +1682,10 @@
/> />
</FileConfiguration> </FileConfiguration>
</File> </File>
<File
RelativePath=".\src\ThreadEntry.cpp"
>
</File>
<File <File
RelativePath=".\src\ThreadLocal.cpp" RelativePath=".\src\ThreadLocal.cpp"
> >
@@ -1814,6 +1818,10 @@
RelativePath=".\include\Poco\Thread_WIN32.h" RelativePath=".\include\Poco\Thread_WIN32.h"
> >
</File> </File>
<File
RelativePath=".\include\Poco\ThreadEntry.h"
>
</File>
<File <File
RelativePath=".\include\Poco\ThreadLocal.h" RelativePath=".\include\Poco\ThreadLocal.h"
> >

View File

@@ -27,7 +27,7 @@ objects = ArchiveStrategy ASCIIEncoding AsyncChannel Base64Decoder Base64Encoder
StreamTokenizer String StringTokenizer SynchronizedObject \ StreamTokenizer String StringTokenizer SynchronizedObject \
Task TaskManager TaskNotification TeeStream Hash HashStatistic \ Task TaskManager TaskNotification TeeStream Hash HashStatistic \
TemporaryFile TextConverter TextEncoding TextIterator Thread ThreadLocal \ TemporaryFile TextConverter TextEncoding TextIterator Thread ThreadLocal \
ThreadPool ActiveDispatcher Timer Timespan Timestamp Timezone Token URI \ ThreadPool ThreadTarget ActiveDispatcher Timer Timespan Timestamp Timezone Token URI \
FileStreamFactory URIStreamFactory URIStreamOpener UTF16Encoding Windows1252Encoding \ FileStreamFactory URIStreamFactory URIStreamOpener UTF16Encoding Windows1252Encoding \
UTF8Encoding UnicodeConverter UUID UUIDGenerator Void Format \ UTF8Encoding UnicodeConverter UUID UUIDGenerator Void Format \
Pipe PipeImpl PipeStream SharedMemory FileStream Unicode UTF8String \ Pipe PipeImpl PipeStream SharedMemory FileStream Unicode UTF8String \

View File

@@ -5,7 +5,7 @@
// //
// Library: Foundation // Library: Foundation
// Package: Threading // Package: Threading
// Module: Thread // Module: RunnableAdapter
// //
// Definition of the RunnableAdapter template class. // Definition of the RunnableAdapter template class.
// //
@@ -59,11 +59,15 @@ class RunnableAdapter: public Runnable
public: public:
typedef void (C::*Callback)(); typedef void (C::*Callback)();
RunnableAdapter(C& object, Callback method): _pObject(&object), _method(method) RunnableAdapter(C& object, Callback method):
_pObject(&object),
_method(method)
{ {
} }
RunnableAdapter(const RunnableAdapter& ra): _pObject(ra._pObject), _method(ra._method) RunnableAdapter(const RunnableAdapter& ra):
_pObject(ra._pObject),
_method(ra._method)
{ {
} }

View File

@@ -68,6 +68,8 @@ class Foundation_API Thread: private ThreadImpl
/// The name of a thread can be changed at any time. /// The name of a thread can be changed at any time.
{ {
public: public:
using ThreadImpl::Callback;
enum Priority enum Priority
/// Thread priorities. /// Thread priorities.
{ {
@@ -108,9 +110,22 @@ public:
Priority getPriority() const; Priority getPriority() const;
/// Returns the thread's priority. /// Returns the thread's priority.
void setStackSize(std::size_t size);
/// Sets the thread's stack size in bytes.
/// Setting the stack size to 0 will use the default stack size.
/// Typically, the real stack size is rounded up to the nearest
/// page size multiple.
std::size_t getStackSize() const;
/// Returns the thread's stack size in bytes.
/// If the default stack size is used, 0 is returned.
void start(Runnable& target); void start(Runnable& target);
/// Starts the thread with the given target. /// Starts the thread with the given target.
void start(Callback target, void* pData = 0);
/// Starts the thread with the given target and parameter.
void join(); void join();
/// Waits until the thread completes execution. /// Waits until the thread completes execution.
/// If multiple threads try to join the same /// If multiple threads try to join the same
@@ -216,6 +231,18 @@ inline Thread* Thread::current()
} }
inline void Thread::setStackSize(std::size_t size)
{
setStackSizeImpl(size);
}
inline std::size_t Thread::getStackSize() const
{
return getStackSizeImpl();
}
} // namespace Poco } // namespace Poco

View File

@@ -0,0 +1,104 @@
//
// ThreadTarget.h
//
// $Id: //poco/svn/Foundation/include/Poco/ThreadTarget.h#2 $
//
// Library: Foundation
// Package: Threading
// Module: ThreadTarget
//
// Definition of the ThreadTarget class.
//
// Copyright (c) 2004-2006, Applied Informatics Software Engineering GmbH.
// and Contributors.
//
// Permission is hereby granted, free of charge, to any person or organization
// obtaining a copy of the software and accompanying documentation covered by
// this license (the "Software") to use, reproduce, display, distribute,
// execute, and transmit the Software, and to prepare derivative works of the
// Software, and to permit third-parties to whom the Software is furnished to
// do so, all subject to the following:
//
// The copyright notices in the Software and this entire statement, including
// the above license grant, this restriction and the following disclaimer,
// must be included in all copies of the Software, in whole or in part, and
// all derivative works of the Software, unless such copies or derivative
// works are solely in the form of machine-executable object code generated by
// a source language processor.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE, TITLE AND NON-INFRINGEMENT. IN NO EVENT
// SHALL THE COPYRIGHT HOLDERS OR ANYONE DISTRIBUTING THE SOFTWARE BE LIABLE
// FOR ANY DAMAGES OR OTHER LIABILITY, WHETHER IN CONTRACT, TORT OR OTHERWISE,
// ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
// DEALINGS IN THE SOFTWARE.
//
#ifndef Foundation_ThreadTarget_INCLUDED
#define Foundation_ThreadTarget_INCLUDED
#include "Poco/Foundation.h"
#include "Poco/Runnable.h"
namespace Poco {
class Foundation_API ThreadTarget: public Runnable
/// This adapter simplifies using static member functions as well as
/// standalone functions as targets for threads.
/// Note that it is possible to pass those entities directly to Thread::start().
/// This adapter is provided as a convenience for higher abstraction level
/// scenarios where Runnable abstract class is used.
///
/// Usage:
/// class MyObject
/// {
/// static void doSomething() {}
/// };
/// ThreadTarget ra(&MyObject::doSomething));
/// Thread thr;
/// thr.start(ra);
///
/// or:
///
/// void doSomething() {}
///
/// ThreadTarget ra(doSomething));
/// Thread thr;
/// thr.start(ra);
{
public:
typedef void (*Callback)();
ThreadTarget(Callback method);
ThreadTarget(const ThreadTarget& te);
~ThreadTarget();
ThreadTarget& operator = (const ThreadTarget& te);
void run();
private:
ThreadTarget();
Callback _method;
};
inline void ThreadTarget::run()
{
_method();
}
} // namespace Poco
#endif // Foundation_ThreadTarget_INCLUDED

View File

@@ -59,6 +59,8 @@ namespace Poco {
class Foundation_API ThreadImpl class Foundation_API ThreadImpl
{ {
public: public:
typedef void (*Callback)(void*);
enum Priority enum Priority
{ {
PRIO_LOWEST_IMPL, PRIO_LOWEST_IMPL,
@@ -68,13 +70,25 @@ public:
PRIO_HIGHEST_IMPL PRIO_HIGHEST_IMPL
}; };
struct CallbackData: public RefCountedObject
{
CallbackData(): callback(0), pData(0)
{
}
Callback callback;
void* pData;
};
ThreadImpl(); ThreadImpl();
~ThreadImpl(); ~ThreadImpl();
Runnable& targetImpl() const;
void setPriorityImpl(int prio); void setPriorityImpl(int prio);
int getPriorityImpl() const; int getPriorityImpl() const;
void setStackSizeImpl(std::size_t size);
std::size_t getStackSizeImpl() const;
void startImpl(Runnable& target); void startImpl(Runnable& target);
void startImpl(Callback target, void* pData = 0);
void joinImpl(); void joinImpl();
bool joinImpl(long milliseconds); bool joinImpl(long milliseconds);
@@ -91,17 +105,21 @@ private:
struct ThreadData: public RefCountedObject struct ThreadData: public RefCountedObject
{ {
ThreadData(): ThreadData():
pTarget(0), pRunnableTarget(0),
pCallbackTarget(0),
thread(0), thread(0),
prio(PRIO_NORMAL_IMPL), prio(PRIO_NORMAL_IMPL),
done(false) done(false),
stackSize(0)
{ {
} }
Runnable* pTarget; Runnable* pRunnableTarget;
pthread_t thread; AutoPtr<CallbackData> pCallbackTarget;
int prio; pthread_t thread;
Event done; int prio;
Event done;
std::size_t stackSize;
}; };
AutoPtr<ThreadData> _pData; AutoPtr<ThreadData> _pData;
@@ -142,12 +160,31 @@ inline void ThreadImpl::sleepImpl(long milliseconds)
} }
inline bool ThreadImpl::isRunningImpl() const
{
return _pData->pRunnableTarget != 0 ||
(_pData->pCallbackTarget.get() != 0 && _pData->pCallbackTarget->callback != 0);
}
inline void ThreadImpl::yieldImpl() inline void ThreadImpl::yieldImpl()
{ {
sched_yield(); sched_yield();
} }
inline void ThreadImpl::setStackSizeImpl(std::size_t size)
{
_pData->stackSize = size;
}
inline std::size_t ThreadImpl::getStackSizeImpl() const
{
return _pData->stackSize;
}
} // namespace Poco } // namespace Poco

View File

@@ -51,6 +51,24 @@ namespace Poco {
class Foundation_API ThreadImpl class Foundation_API ThreadImpl
{ {
public: public:
typedef void (*Callback)(void*);
#if defined(_DLL)
typedef DWORD (WINAPI *Entry)(LPVOID);
#else
typedef unsigned (__stdcall *Entry)(void*);
#endif
struct CallbackData
{
CallbackData(): callback(0), pData(0)
{
}
Callback callback;
void* pData;
};
enum Priority enum Priority
{ {
PRIO_LOWEST_IMPL = THREAD_PRIORITY_LOWEST, PRIO_LOWEST_IMPL = THREAD_PRIORITY_LOWEST,
@@ -65,7 +83,10 @@ public:
void setPriorityImpl(int prio); void setPriorityImpl(int prio);
int getPriorityImpl() const; int getPriorityImpl() const;
void setStackSizeImpl(std::size_t size);
std::size_t getStackSizeImpl() const;
void startImpl(Runnable& target); void startImpl(Runnable& target);
void startImpl(Callback target, void* pData = 0);
void joinImpl(); void joinImpl();
bool joinImpl(long milliseconds); bool joinImpl(long milliseconds);
@@ -76,15 +97,26 @@ public:
protected: protected:
#if defined(_DLL) #if defined(_DLL)
static DWORD WINAPI entry(LPVOID pThread); static DWORD WINAPI runnableEntry(LPVOID pThread);
#else #else
static unsigned __stdcall entry(void* pThread); static unsigned __stdcall runnableEntry(void* pThread);
#endif #endif
#if defined(_DLL)
static DWORD WINAPI functionEntry(LPVOID pThread);
#else
static unsigned __stdcall functionEntry(void* pThread);
#endif
void createImpl(Entry ent, void* pData);
void threadCleanup();
private: private:
Runnable* _pTarget; Runnable* _pRunnableTarget;
HANDLE _thread; CallbackData _callbackTarget;
int _prio; HANDLE _thread;
int _prio;
std::size_t _stackSize;
static DWORD _currentKey; static DWORD _currentKey;
}; };
@@ -111,6 +143,18 @@ inline void ThreadImpl::yieldImpl()
} }
inline void ThreadImpl::setStackSizeImpl(std::size_t size)
{
_stackSize = size;
}
inline std::size_t ThreadImpl::getStackSizeImpl() const
{
return _stackSize;
}
} // namespace Poco } // namespace Poco

View File

@@ -91,6 +91,12 @@ void Thread::start(Runnable& target)
} }
void Thread::start(Callback target, void* pData)
{
startImpl(target, pData);
}
void Thread::join() void Thread::join()
{ {
joinImpl(); joinImpl();

View File

@@ -0,0 +1,66 @@
//
// ThreadTarget.cpp
//
// $Id: //poco/svn/Foundation/src/ThreadTarget.cpp#2 $
//
// Library: Foundation
// Package: Threading
// Module: ThreadTarget
//
// Copyright (c) 2004-2006, Applied Informatics Software Engineering GmbH.
// and Contributors.
//
// Permission is hereby granted, free of charge, to any person or organization
// obtaining a copy of the software and accompanying documentation covered by
// this license (the "Software") to use, reproduce, display, distribute,
// execute, and transmit the Software, and to prepare derivative works of the
// Software, and to permit third-parties to whom the Software is furnished to
// do so, all subject to the following:
//
// The copyright notices in the Software and this entire statement, including
// the above license grant, this restriction and the following disclaimer,
// must be included in all copies of the Software, in whole or in part, and
// all derivative works of the Software, unless such copies or derivative
// works are solely in the form of machine-executable object code generated by
// a source language processor.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE, TITLE AND NON-INFRINGEMENT. IN NO EVENT
// SHALL THE COPYRIGHT HOLDERS OR ANYONE DISTRIBUTING THE SOFTWARE BE LIABLE
// FOR ANY DAMAGES OR OTHER LIABILITY, WHETHER IN CONTRACT, TORT OR OTHERWISE,
// ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
// DEALINGS IN THE SOFTWARE.
//
#include "Poco/ThreadTarget.h"
namespace Poco {
ThreadTarget::ThreadTarget(Callback method): _method(method)
{
}
ThreadTarget::ThreadTarget(const ThreadTarget& te): _method(te._method)
{
}
ThreadTarget& ThreadTarget::operator = (const ThreadTarget& te)
{
_method = te._method;
return *this;
}
ThreadTarget::~ThreadTarget()
{
}
} // namespace Poco

View File

@@ -39,7 +39,6 @@
#include "Poco/ErrorHandler.h" #include "Poco/ErrorHandler.h"
#include <signal.h> #include <signal.h>
// //
// Block SIGPIPE in main thread. // Block SIGPIPE in main thread.
// //
@@ -87,7 +86,7 @@ ThreadImpl::ThreadImpl():
ThreadImpl::~ThreadImpl() ThreadImpl::~ThreadImpl()
{ {
if (_pData->pTarget) if (isRunningImpl())
pthread_detach(_pData->thread); pthread_detach(_pData->thread);
} }
@@ -97,7 +96,7 @@ void ThreadImpl::setPriorityImpl(int prio)
if (prio != _pData->prio) if (prio != _pData->prio)
{ {
_pData->prio = prio; _pData->prio = prio;
if (_pData->pTarget) if (isRunningImpl())
{ {
struct sched_param par; struct sched_param par;
par.sched_priority = mapPrio(_pData->prio); par.sched_priority = mapPrio(_pData->prio);
@@ -110,12 +109,52 @@ void ThreadImpl::setPriorityImpl(int prio)
void ThreadImpl::startImpl(Runnable& target) void ThreadImpl::startImpl(Runnable& target)
{ {
if (_pData->pTarget) throw SystemException("thread already running"); if (_pData->pRunnableTarget) throw SystemException("thread already running");
_pData->pTarget = &target; pthread_attr_t attributes;
if (pthread_create(&_pData->thread, NULL, entry, this)) pthread_attr_init(&attributes);
if (_pData->stackSize != 0)
pthread_attr_setstacksize(&attributes, _pData->stackSize);
_pData->pRunnableTarget = &target;
if (pthread_create(&_pData->thread, &attributes, entry, this))
{ {
_pData->pTarget = 0; _pData->pRunnableTarget = 0;
throw SystemException("cannot start thread");
}
if (_pData->prio != PRIO_NORMAL_IMPL)
{
struct sched_param par;
par.sched_priority = mapPrio(_pData->prio);
if (pthread_setschedparam(_pData->thread, SCHED_OTHER, &par))
throw SystemException("cannot set thread priority");
}
}
void ThreadImpl::startImpl(Callback target, void* pData)
{
if (_pData->pCallbackTarget && _pData->pCallbackTarget->callback)
throw SystemException("thread already running");
pthread_attr_t attributes;
pthread_attr_init(&attributes);
if (_pData->stackSize != 0)
pthread_attr_setstacksize(&attributes, _pData->stackSize);
if (0 == _pData->pCallbackTarget.get())
_pData->pCallbackTarget = new CallbackData;
_pData->pCallbackTarget->callback = target;
_pData->pCallbackTarget->pData = pData;
if (pthread_create(&_pData->thread, &attributes, entry, this))
{
_pData->pCallbackTarget->callback = 0;
_pData->pCallbackTarget->pData = 0;
throw SystemException("cannot start thread"); throw SystemException("cannot start thread");
} }
@@ -151,12 +190,6 @@ bool ThreadImpl::joinImpl(long milliseconds)
} }
bool ThreadImpl::isRunningImpl() const
{
return _pData->pTarget != 0;
}
ThreadImpl* ThreadImpl::currentImpl() ThreadImpl* ThreadImpl::currentImpl()
{ {
if (_haveCurrentKey) if (_haveCurrentKey)
@@ -183,7 +216,10 @@ void* ThreadImpl::entry(void* pThread)
AutoPtr<ThreadData> pData = pThreadImpl->_pData; AutoPtr<ThreadData> pData = pThreadImpl->_pData;
try try
{ {
pData->pTarget->run(); if (pData->pRunnableTarget)
pData->pRunnableTarget->run();
else
pData->pCallbackTarget->callback(pData->pCallbackTarget->pData);
} }
catch (Exception& exc) catch (Exception& exc)
{ {
@@ -197,7 +233,14 @@ void* ThreadImpl::entry(void* pThread)
{ {
ErrorHandler::handle(); ErrorHandler::handle();
} }
pData->pTarget = 0; if (pData->pRunnableTarget)
pData->pRunnableTarget = 0;
else
{
pData->pCallbackTarget->callback = 0;
pData->pCallbackTarget->pData = 0;
}
pData->done.set(); pData->done.set();
return 0; return 0;
} }

View File

@@ -46,7 +46,11 @@ namespace Poco {
DWORD ThreadImpl::_currentKey = TLS_OUT_OF_INDEXES; DWORD ThreadImpl::_currentKey = TLS_OUT_OF_INDEXES;
ThreadImpl::ThreadImpl(): _pTarget(0), _thread(0), _prio(PRIO_NORMAL_IMPL) ThreadImpl::ThreadImpl():
_pRunnableTarget(0),
_thread(0),
_prio(PRIO_NORMAL_IMPL),
_stackSize(0)
{ {
if (_currentKey == TLS_OUT_OF_INDEXES) if (_currentKey == TLS_OUT_OF_INDEXES)
{ {
@@ -79,15 +83,35 @@ void ThreadImpl::setPriorityImpl(int prio)
void ThreadImpl::startImpl(Runnable& target) void ThreadImpl::startImpl(Runnable& target)
{ {
if (_thread) throw SystemException("thread already running"); if (isRunningImpl())
throw SystemException("thread already running");
_pTarget = &target; _pRunnableTarget = &target;
createImpl(runnableEntry, this);
}
void ThreadImpl::startImpl(Callback target, void* pData)
{
if (isRunningImpl())
throw SystemException("thread already running");
_callbackTarget.callback = target;
_callbackTarget.pData = pData;
createImpl(functionEntry, this);
}
void ThreadImpl::createImpl(Entry ent, void* pData)
{
#if defined(_DLL) #if defined(_DLL)
DWORD threadId; DWORD threadId;
_thread = CreateThread(NULL, 0, entry, this, 0, &threadId); _thread = CreateThread(NULL, _stackSize, ent, pData, 0, &threadId);
#else #else
unsigned threadId; unsigned threadId;
_thread = (HANDLE) _beginthreadex(NULL, 0, entry, this, 0, &threadId); _thread = (HANDLE) _beginthreadex(NULL, _stackSize, runnableEntry, this, 0, &threadId);
#endif #endif
if (!_thread) if (!_thread)
throw SystemException("cannot create thread"); throw SystemException("cannot create thread");
@@ -103,6 +127,7 @@ void ThreadImpl::joinImpl()
switch (WaitForSingleObject(_thread, INFINITE)) switch (WaitForSingleObject(_thread, INFINITE))
{ {
case WAIT_OBJECT_0: case WAIT_OBJECT_0:
threadCleanup();
return; return;
default: default:
throw SystemException("cannot join thread"); throw SystemException("cannot join thread");
@@ -119,6 +144,7 @@ bool ThreadImpl::joinImpl(long milliseconds)
case WAIT_TIMEOUT: case WAIT_TIMEOUT:
return false; return false;
case WAIT_OBJECT_0: case WAIT_OBJECT_0:
threadCleanup();
return true; return true;
default: default:
throw SystemException("cannot join thread"); throw SystemException("cannot join thread");
@@ -137,6 +163,13 @@ bool ThreadImpl::isRunningImpl() const
} }
void ThreadImpl::threadCleanup()
{
if (!_thread) return;
if (CloseHandle(_thread)) _thread = 0;
}
ThreadImpl* ThreadImpl::currentImpl() ThreadImpl* ThreadImpl::currentImpl()
{ {
if (_currentKey == TLS_OUT_OF_INDEXES) if (_currentKey == TLS_OUT_OF_INDEXES)
@@ -147,15 +180,43 @@ ThreadImpl* ThreadImpl::currentImpl()
#if defined(_DLL) #if defined(_DLL)
DWORD WINAPI ThreadImpl::entry(LPVOID pThread) DWORD WINAPI ThreadImpl::runnableEntry(LPVOID pThread)
#else #else
unsigned __stdcall ThreadImpl::entry(void* pThread) unsigned __stdcall ThreadImpl::runnableEntry(void* pThread)
#endif #endif
{ {
TlsSetValue(_currentKey, pThread); TlsSetValue(_currentKey, pThread);
try try
{ {
reinterpret_cast<ThreadImpl*>(pThread)->_pTarget->run(); reinterpret_cast<ThreadImpl*>(pThread)->_pRunnableTarget->run();
}
catch (Exception& exc)
{
ErrorHandler::handle(exc);
}
catch (std::exception& exc)
{
ErrorHandler::handle(exc);
}
catch (...)
{
ErrorHandler::handle();
}
return 0;
}
#if defined(_DLL)
DWORD WINAPI ThreadImpl::functionEntry(LPVOID pThread)
#else
unsigned __stdcall ThreadImpl::functionEntry(void* pThread)
#endif
{
TlsSetValue(_currentKey, pThread);
try
{
ThreadImpl* pTI = reinterpret_cast<ThreadImpl*>(pThread);
pTI->_callbackTarget.callback(pTI->_callbackTarget.pData);
} }
catch (Exception& exc) catch (Exception& exc)
{ {

View File

@@ -35,11 +35,13 @@
#include "CppUnit/TestSuite.h" #include "CppUnit/TestSuite.h"
#include "Poco/Thread.h" #include "Poco/Thread.h"
#include "Poco/Runnable.h" #include "Poco/Runnable.h"
#include "Poco/ThreadTarget.h"
#include "Poco/Event.h" #include "Poco/Event.h"
#include <iostream>
using Poco::Thread; using Poco::Thread;
using Poco::Runnable; using Poco::Runnable;
using Poco::ThreadTarget;
using Poco::Event; using Poco::Event;
@@ -74,6 +76,13 @@ public:
_event.set(); _event.set();
} }
static void staticFunc()
{
++_staticVar;
}
static int _staticVar;
private: private:
bool _ran; bool _ran;
std::string _threadName; std::string _threadName;
@@ -81,6 +90,21 @@ private:
}; };
int MyRunnable::_staticVar = 0;
void freeFunc()
{
++MyRunnable::_staticVar;
}
void freeFunc(void* pData)
{
MyRunnable::_staticVar += *reinterpret_cast<int*>(pData);
}
ThreadTest::ThreadTest(const std::string& name): CppUnit::TestCase(name) ThreadTest::ThreadTest(const std::string& name): CppUnit::TestCase(name)
{ {
} }
@@ -195,6 +219,75 @@ void ThreadTest::testJoin()
} }
void ThreadTest::testThreadTarget()
{
ThreadTarget te(&MyRunnable::staticFunc);
Thread thread;
assert (!thread.isRunning());
int tmp = MyRunnable::_staticVar;
thread.start(te);
thread.join();
assert (tmp + 1 == MyRunnable::_staticVar);
ThreadTarget te1(freeFunc);
assert (!thread.isRunning());
tmp = MyRunnable::_staticVar;
thread.start(te1);
thread.join();
assert (tmp + 1 == MyRunnable::_staticVar);
}
void ThreadTest::testThreadFunction()
{
Thread thread;
assert (!thread.isRunning());
int tmp = MyRunnable::_staticVar;
thread.start(freeFunc, &tmp);
thread.join();
assert (tmp * 2 == MyRunnable::_staticVar);
assert (!thread.isRunning());
tmp = MyRunnable::_staticVar = 0;
thread.start(freeFunc, &tmp);
thread.join();
assert (0 == MyRunnable::_staticVar);
}
void ThreadTest::testThreadStackSize()
{
Thread thread;
assert (0 == thread.getStackSize());
thread.setStackSize(50000000);
assert (50000000 == thread.getStackSize());
int tmp = MyRunnable::_staticVar;
thread.start(freeFunc, &tmp);
thread.join();
assert (tmp * 2 == MyRunnable::_staticVar);
thread.setStackSize(1);
assert (1 == thread.getStackSize());
tmp = MyRunnable::_staticVar;
thread.start(freeFunc, &tmp);
thread.join();
assert (tmp * 2 == MyRunnable::_staticVar);
thread.setStackSize(0);
assert (0 == thread.getStackSize());
tmp = MyRunnable::_staticVar;
thread.start(freeFunc, &tmp);
thread.join();
assert (tmp * 2 == MyRunnable::_staticVar);
}
void ThreadTest::setUp() void ThreadTest::setUp()
{ {
} }
@@ -214,6 +307,9 @@ CppUnit::Test* ThreadTest::suite()
CppUnit_addTest(pSuite, ThreadTest, testCurrent); CppUnit_addTest(pSuite, ThreadTest, testCurrent);
CppUnit_addTest(pSuite, ThreadTest, testThreads); CppUnit_addTest(pSuite, ThreadTest, testThreads);
CppUnit_addTest(pSuite, ThreadTest, testJoin); CppUnit_addTest(pSuite, ThreadTest, testJoin);
CppUnit_addTest(pSuite, ThreadTest, testThreadTarget);
CppUnit_addTest(pSuite, ThreadTest, testThreadFunction);
CppUnit_addTest(pSuite, ThreadTest, testThreadStackSize);
return pSuite; return pSuite;
} }

View File

@@ -51,6 +51,9 @@ public:
void testCurrent(); void testCurrent();
void testThreads(); void testThreads();
void testJoin(); void testJoin();
void testThreadTarget();
void testThreadFunction();
void testThreadStackSize();
void setUp(); void setUp();
void tearDown(); void tearDown();