在Win32平台上进行多线程编程,常会用到锁。下边用C++实现了互斥对象(Mutex)锁和临界区(CRITICAL_SECTION)锁,以加深理解和今后方便使用。代码已在VS2005环境下编译测试通过。
Lock.h
- #ifndef _Lock_H
- #define _Lock_H
- #include <windows.h>
- //锁接口类
- class ILock
- {
- public:
- virtual ~ILock() {}
- virtual void Lock() const = 0;
- virtual void Unlock() const = 0;
- };
- //互斥对象锁类
- class Mutex : public ILock
- {
- public:
- Mutex();
- ~Mutex();
- virtual void Lock() const;
- virtual void Unlock() const;
- private:
- HANDLE m_mutex;
- };
- //临界区锁类
- class CriSection : public ILock
- {
- public:
- CriSection();
- ~CriSection();
- virtual void Lock() const;
- virtual void Unlock() const;
- private:
- CRITICAL_SECTION m_critclSection;
- };
- //锁
- class CMyLock
- {
- public:
- CMyLock(const ILock&);
- ~CMyLock();
- private:
- const ILock& m_lock;
- };
- #endif
Lock.cpp
- #include "Lock.h"
- //---------------------------------------------------------------------------
- //创建一个匿名互斥对象
- Mutex::Mutex()
- {
- m_mutex = ::CreateMutex(NULL, FALSE, NULL);
- }
- //销毁互斥对象,释放资源
- Mutex::~Mutex()
- {
- ::CloseHandle(m_mutex);
- }
- //确保拥有互斥对象的线程对被保护资源的独自访问
- void Mutex::Lock() const
- {
- DWORD d = WaitForSingleObject(m_mutex, INFINITE);
- }
- //释放当前线程拥有的互斥对象,以使其它线程可以拥有互斥对象,对被保护资源进行访问
- void Mutex::Unlock() const
- {
- ::ReleaseMutex(m_mutex);
- }
- //---------------------------------------------------------------------------
- //初始化临界资源对象
- CriSection::CriSection()
- {
- ::InitializeCriticalSection(&m_critclSection);
- }
- //释放临界资源对象
- CriSection::~CriSection()
- {
- ::DeleteCriticalSection(&m_critclSection);
- }
- //进入临界区,加锁
- void CriSection::Lock() const
- {
- ::EnterCriticalSection((LPCRITICAL_SECTION)&m_critclSection);
- }
- //离开临界区,解锁
- void CriSection::Unlock() const
- {
- ::LeaveCriticalSection((LPCRITICAL_SECTION)&m_critclSection);
- }
- //---------------------------------------------------------------------------
- //利用C++特性,进行自动加锁
- CMyLock::CMyLock(const ILock& m) : m_lock(m)
- {
- m_lock.Lock();
- }
- //利用C++特性,进行自动解锁
- CMyLock::~CMyLock()
- {
- m_lock.Unlock();
- }
下边是测试代码
- // MyLock.cpp : 定义控制台应用程序的入口点。
- //
- #include <iostream>
- #include <process.h>
- #include <time.h>
- #include "Lock.h"
- using namespace std;
- #define ENABLE_MUTEX
- #define ENABLE_CRITICAL_SECTION
- #if defined (ENABLE_MUTEX)
- //创建一个互斥对象类型锁
- Mutex g_Lock;
- #elif defined (ENABLE_CRITICAL_SECTION)
- //创建一个临界区类型锁
- CriSection g_Lock;
- #endif
- void LockCompare(int &iNum)
- {
- CMyLock lock1(g_Lock);
- iNum++;
- }
- //线程函数
- unsigned int __stdcall StartThread(void *pParam)
- {
- char *pMsg = (char *)pParam;
- if (!pMsg)
- {
- return (unsigned int)1;
- }
- CMyLock lock2(g_Lock);
- clock_t tStart,tEnd;
- tStart = clock();
- int iNum = 0;
- for (int i = 0; i < 100000; i++)
- {
- LockCompare(iNum);
- }
- tEnd = clock();
- #if defined (ENABLE_MUTEX)
- cout<<"The lock type is mutex, time = "<<(tEnd - tStart)<<" ms."<<endl;
- #elif defined (ENABLE_CRITICAL_SECTION)
- cout<<"The lock type is critical section, time = "<<(tEnd - tStart)<<" ms."<<endl;
- #endif
- return (unsigned int)0;
- }
- int main(int argc, char* argv[])
- {
- HANDLE hThread1, hThread2;
- unsigned int uiThreadId1, uiThreadId2;
- char *pMsg1 = "First print thread.";
- char *pMsg2 = "Second print thread.";
- //创建两个工作线程,分别打印不同的消息
- hThread1 = (HANDLE)_beginthreadex(NULL, 0, &StartThread, (void *)pMsg1, 0, &uiThreadId1);
- hThread2 = (HANDLE)_beginthreadex(NULL, 0, &StartThread, (void *)pMsg2, 0, &uiThreadId2);
- //等待线程结束
- DWORD dwRet = WaitForSingleObject(hThread1,INFINITE);
- if ( dwRet == WAIT_TIMEOUT )
- {
- TerminateThread(hThread1,0);
- }
- dwRet = WaitForSingleObject(hThread2,INFINITE);
- if ( dwRet == WAIT_TIMEOUT )
- {
- TerminateThread(hThread2,0);
- }
- //关闭线程句柄,释放资源
- ::CloseHandle(hThread1);
- ::CloseHandle(hThread2);
- system("pause");
- return 0;
- }
在线程函数StartThread中,循环100000次,对保护资源“iNum ”反复加锁,解锁。编译,运行5次,将每次打印的线程锁切换耗时时间记录下来。之后,将测试代码中的宏 #define ENABLE_MUTEX 注释掉,禁掉互斥锁,启用临界区锁,重新编译代码,运行5次。下边是分别是互斥锁和临界区锁耗时记录(不同机器上耗时会不同):
互斥锁
线程Id |
耗时 / ms |
总计 |
||||
1 |
141 |
125 |
125 |
125 |
125 |
641 |
2 |
140 |
125 |
140 |
125 |
156 |
686 |
临界区锁
线程Id |
耗时 / ms |
总计 |
||||
1 |
15 |
16 |
31 |
31 |
31 |
124 |
2 |
31 |
31 |
31 |
16 |
31 |
140 |
互斥锁总共耗时:641+686=1327 ms,而临界区锁:124+140=264 ms。显而易见,临界区锁耗时比互斥锁耗时节约了大概5倍的时间。
总结:1、在同一个进程的多线程同步锁,宜用临界区锁,它比较节约线程上下文切换带来的系统开销。但因临界区工作在用户模式下,所以不能对不同进程中的多线程进行同步。2、因互斥对象锁属于内核对象,所以在进行多线程同步时速度会比较慢,但是可以在不同进程的多个线程之间进行同步。