在linux下实现CSingleLock函数

最近做的项目涉及到多线程处理的问题，因为考虑到在windows下实现的效率，在里面是有了CSingleLock，现在需要将项目扩展到可以跨平台。所以就面临两个选择，第一就是将所有涉及到操作系统平台的地方都使用第三方的库来替换，优点是省事，缺点是之前针对windows做的一些优化就白费了；第二个就是对windows的相关函数小修改甚至不修改，然后增加linux平台下的函数，此种方式优点是保持之前windows下优化的部分，缺点就是自己写相应的函数比较费时，而且需要测试。经过评估项目本身的特性，觉得还是第二种方式要适合一点。本文就给出CSingleLock的实现源码和简单测试程序。

闲话少说，先贴上代码

#ifndef TMUTEX_H
#define TMUTEX_H

#include <pthread.h>
#include <stdio.h>

class CThreadMutex
{
public:
    CThreadMutex()
    {
        pthread_mutex_init(&mtx, NULL);
    }
    
    ~CThreadMutex()
    {
        printf("release 1===================== \n");
        pthread_mutex_destroy(&mtx);
    }
    
    inline void lock()
    {
        pthread_mutex_lock(&mtx);
    }    

    inline void unlock()
    {
        printf("unlock 1===================== \n");
        pthread_mutex_unlock(&mtx);
    }

private:
    pthread_mutex_t mtx;
};


//class NullMutex
//{
//public:
//    inline void lock()  {};
//    inline void unlock(){};
//};

class CSingleLock
{
public:
    CSingleLock(CThreadMutex &mtx) : m_mtx(mtx) , m_bIsLocked(false)
    {}
    
    void Lock()
    {
        m_bIsLocked = true;
        m_mtx.lock();
        printf("lock ++++++ \n");
    }

    void UnLock()
    {
        if (m_bIsLocked)
        {
            printf("++++++++++++++++++++++++++++ \n");
            m_bIsLocked = false;
            m_mtx.unlock();
            printf("Unlock ++++++ \n");
        }
    }

    ~CSingleLock()
    {
        printf("release 2===================== \n");
        m_mtx.unlock();
    }

protected:
    bool m_bIsLocked;
    CThreadMutex &m_mtx;

};

#endif

实现的基本思路就是对posix标准中的互斥锁封装成类 CThreadMutex，然后通过类CSingleLock进行调用。

测试程序如下：

#include <stdio.h>
#include <pthread.h>
#include "CSingleLock.h"
#include <unistd.h>

typedef CThreadMutex MUTEX_TYPE;
//typedef NullMutex MUTEX_TYPE;

MUTEX_TYPE g_mtx;

int a = 200;
int b = 100;

void *add(void *pPM)
{
    {
    CSingleLock gd(g_mtx);
    gd.Lock();
    
    a -= 50;
    
    sleep(15);

    b += 50;    
    
    printf("leave ==================== \n"); 

//gd.Unlock();  //显式的调用释放锁函数

    }
    
    printf("in sleep\n"); 
    sleep(20);
}


void *sub(void *pPM)
{
    sleep(2);
    CSingleLock gd(g_mtx);
    gd.Lock();
    printf("%d\n", a+b);

    //gd.Unlock();   //显式的调用释放锁函数</span>
}

int main(int argc, char **argv)
{
    pthread_t tid1, tid2;
    //pthread_t tid3, tid4;

    pthread_create(&tid1, NULL, add, NULL);
    pthread_create(&tid2, NULL, sub, NULL);

    pthread_join(tid1, NULL);
    pthread_join(tid2, NULL);

    return 0;  
}

          线程tid1和tid2都需要获取锁g_mtx，在线程tid1中，在函数包含的花括号中，获取该锁，在离开该花括号时，应该自动释放该锁（也就是说，在打印leave后就应该释放该锁），这时tid1开始sleep，tid2获取到该锁。当tid2结束后也自动释放该锁。当然上述锁的释放也可以显式的调用Unlock函数来释放。测试程序结果如下：（家里的ubuntu在更新，明日补上程序结果截图）