root/trunk/0051_RWLock/RWLockTest.cpp

User picture

Author: Setsu

Revision: 348 («Previous)


File Size: 4.38 KB

(August 06, 2010 09:30 UTC) Almost 2 years ago

Typo.

 
Show/hide line numbers
//
//  RWLockTest.cpp
//
//  Created by Setsu on 7/25/2010.
//  Copyright 2010 RoundSquare Inc. All rights reserved.
//
// 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
//----------------------------------------
#include <Poco/Logger.h>
#include <Poco/PatternFormatter.h>
#include <Poco/FormattingChannel.h>
#include <Poco/ConsoleChannel.h>
#include <Poco/Format.h>
#include <Poco/Runnable.h>
#include <Poco/Thread.h>
#include <Poco/RWLock.h>
#include <Poco/Stopwatch.h>

#include <string>

#include "ScopedLogMessage.h"

//----------------------------------------
//	const
//----------------------------------------
const int kNumLoops = 1000;
const int kNumRunnable = 5;

//----------------------------------------
//	RWLockRunnable
//----------------------------------------
class RWLockRunnable: public Poco::Runnable
{
public:
	RWLockRunnable(Poco::RWLock& lock, volatile int& counter) :
		m_Lock(lock)
	,	m_Counter(counter)
	,	m_Ok(true)
	{
	}
	
	void run()
	{
		int lastCount = 0;
		for(int i=0; i<kNumLoops; ++i)
		{
			{	// scope for read lock
				Poco::ScopedReadRWLock lock(m_Lock);

				lastCount = m_Counter;
				for(int k=0; k<100; ++k)
				{
					if(m_Counter != lastCount)
					{
						m_Ok = false;
					}
					Poco::Thread::yield();
				}
			}

			{	// scope for write lock
				Poco::ScopedWriteRWLock lock(m_Lock);

				for(int k=0; k<100; ++k)
				{
					--m_Counter;
					Poco::Thread::yield();
				}
				for(int k=0; k<100; ++k)
				{
					++m_Counter;
					Poco::Thread::yield();
				}
				++m_Counter;
				if(m_Counter <= lastCount)
				{
					m_Ok = false;
				}
			}
		}
	}
	
	bool ok() const
	{
		return m_Ok;
	}
	
private:
	Poco::RWLock&	m_Lock;
	volatile int&	m_Counter;
	bool			m_Ok;
};

//----------------------------------------
//	PrepareConsoleLogger
//----------------------------------------
void PrepareConsoleLogger(const std::string& name, int level=Poco::Message::PRIO_INFORMATION)
{
	Poco::FormattingChannel* pFCConsole = new Poco::FormattingChannel(new Poco::PatternFormatter("%t"));
	pFCConsole->setChannel(new Poco::ConsoleChannel);
	pFCConsole->open();

	Poco::Logger::create(name, pFCConsole, level);
}

//----------------------------------------
//	main
//----------------------------------------
int main(int /*argc*/, char** /*argv*/)
{
	PrepareConsoleLogger(Poco::Logger::ROOT, Poco::Message::PRIO_INFORMATION);

	ScopedLogMessage msg("RWLockTest ", "start", "end");

	Poco::Stopwatch stopwatch;
	stopwatch.start();

	Poco::RWLock lock;
	int counter = 0;

	RWLockRunnable* pRunnable[kNumRunnable];
	Poco::Thread thread[kNumRunnable];

	for(int i=0; i<kNumRunnable; ++i)
	{
		pRunnable[i] = new RWLockRunnable(lock, counter);
		thread[i].start(*pRunnable[i]);
	}

	for(int i=0; i<kNumRunnable; ++i)
	{
		thread[i].join();
		if(!pRunnable[i]->ok())
		{
			msg.Message(Poco::format(" RWLockRunnable[%i] failed!", i));
		}
		delete pRunnable[i];
	}

	stopwatch.stop();

	msg.Message(Poco::format(" counter = %i (expected: %i)", counter, kNumLoops*kNumRunnable));
	msg.Message(Poco::format(" Elapsed time = %.3fmSec", (1000.0 * stopwatch.elapsed()) / stopwatch.resolution()));

	return 0;
}