线程互斥锁封装

  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

编译运行：