Linux多线程程序设计中,互斥锁机制十分关键,它主要起到资源保护的作用。在某些场合,对锁的操作,锁的初始化、上锁、解锁和销毁,过程会显得较为繁琐,再比如,上了锁之后再函数退出后有没有解锁,然后又在其他线程企图上锁,这样就造成了死锁。对这些操作可用c++对其做了简单的封装: 类的构造函数上锁,类的析构函数则解锁。运用到c++的多态、纯虚函数和继承、封装的思想。
UML图:
代码如下:
//sysLock.h
#include <pthread.h>
#include <stdio.h>
/* 互斥锁基类,定义2个纯虚函数 */
class BaseLock
{
public:
virtual void Lock() = 0;
virtual void UnLock() = 0;
};
/* 线程互斥锁类 */
class PthreadLock : public BaseLock
{
private:
pthread_mutex_t mutex; /* 锁的句柄 */
bool createMutex(); /* 初始化锁操作 */
void destroyMutex(); /* 销毁锁操作 */
public:
PthreadLock();
~PthreadLock();
void Lock();
void UnLock();
};
/* 自动锁 */
class AutoLock
{
BaseLock* AutoLockObj;
public:
AutoLock(BaseLock* LockObj);
~AutoLock();
};
#endif /* __SYSLOCK_H__ */
//sysLock.cpp
#include <sysLock.h>
/* 构造函数调用初始化锁的私有函数 */
PthreadLock::PthreadLock()
{
createMutex();
}
/* 析构函数调用销毁锁的私有函数 */
PthreadLock::~PthreadLock()
{
destroyMutex();
}
/* 初始化锁 */
bool PthreadLock::createMutex()
{
printf("Mutex::createMutex\n");
return pthread_mutex_init(&mutex, NULL) == 0;
}
/* 销毁锁 */
void PthreadLock::destroyMutex()
{
printf("Mutex::destroyMutex\n");
pthread_mutex_destroy(&mutex);
}
/* 上锁 */
void PthreadLock::Lock()
{
printf("Mutex::Lock\n");
pthread_mutex_lock(&mutex);
}
/* 解锁 */
void PthreadLock::UnLock()
{
printf("Mutex::unLock()\n");
pthread_mutex_unlock(&mutex);
}
/************************************************************/
/************************************************************/
/* 自动上锁 */
AutoLock::AutoLock(BaseLock* LockObj) : AutoLockObj(LockObj)
{
AutoLockObj->Lock();
}
/* 自动解锁 */
AutoLock::~AutoLock()
{
AutoLockObj->UnLock();
}
测试用例:
//main.c
#include <sysLock.h>
#include <unistd.h>
int main()
{
PthreadLock mutex;
AutoLock al(&mutex); /* 调用此句后对上锁,直至main函数退出 */
for (int i = 0; i < 5; i++)
{
printf("*");
fflush(stdout);
sleep(1);
}
printf("\n");
return 0;
}
编写简单的Makefile:
all:
g++ -c sysLock.cpp -I./
g++ main.cpp sysLock.o -I./
clean:
rm *.o
编译运行: