WIN 程序员的 Linux 互斥类

作者：黄山松，发表于cnblogs：http://www.cnblogs.com/tomview/

对于一个 win 的程序员，要把在 win 下的程序移植到 linux 下，需要把一些平台相关的功能代码封装一下，这样在使用这些功能的时候，可以简单调用封装好的代码，方便在不同平台下使用。本文是一个非常简单的互斥类，通过使用这个互斥类，源代码在 linux 下和 win 下保持一致。

在 win 下，互斥的代码是这几个函数：

InitializeCriticalSection

EnterCriticalSection

LeaveCriticalSection

DeleteCriticalSection

TryEnterCriticalSection

在 Linux 下，可以用对应的下面函数实现：

pthread_mutex_init

pthread_mutex_destroy

pthread_mutex_lock

pthread_mutex_unlock

封装类 auto_entercs 之后，不管在 win 还是 linux 下都使用相同的 win 下的函数就可以了，如下：

用法：

#include "auto_entercs.h"

CRITICAL_SECTION m_cs;

InitializeCriticalSection(&m_cs);

DeleteCriticalSection(&m_cs);

在需要互斥的代码区域开头声明类 auto_entercs 的局部变量，类的构造函数中自动调用 EnterCriticalSection 获取控制权，在出了变量作用域的时候，类的析构函数自动调用 LeaveCriticalSection 释放控制权。

{
    //作用域的开始声明局部变量
    auto_entercs ace(&m_cs);
 
    //互斥区域代码。。。
 
    //离开作用域的时候自动释放控制权
 
}

auto_entercs.h 的代码如下：

#ifndef _X_HSS_CRITICAL_SECTION_HSS__
#define _X_HSS_CRITICAL_SECTION_HSS__
 
/**************************************************************************************************\
用于 win32 和 linux 下的通用的互斥类，如下的使用代码在 win 下和 linux 下使用同样的代码使用互斥
 
  用法：
 
  (0) 包含
  #include "auto_entercs.h"
 
  (1) 定义
  CRITICAL_SECTION m_cs;
 
  (2) 初始化
  InitializeCriticalSection(&m_cs);
 
  (3) 进入和离开互斥
  {
    auto_entercs ace(&m_cs);
    ....互斥区域
  }
 
  (4) 删除
  DeleteCriticalSection(&m_cs);
 
作者： 黄山松，http://www.cnblogs.com/tomview/
 
\**************************************************************************************************/
#ifdef WIN32
 
#include <windows.h>
 
#else
 
#include <pthread.h>
 
#define CRITICAL_SECTION            pthread_mutex_t
#define InitializeCriticalSection(p)            \
        {    \
        pthread_mutexattr_t attr;    \
        pthread_mutexattr_init(&attr);    \
        pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE_NP);    \
        pthread_mutex_init((p), &attr);    \
        }
#define DeleteCriticalSection(p)    pthread_mutex_destroy(p)
#define EnterCriticalSection(p)        pthread_mutex_lock(p)
#define LeaveCriticalSection(p)        pthread_mutex_unlock(p)
 
#endif
 
//auto_entercs 需要在互斥代码区域内声明局部变量，当代码执行出区域的时候，析构函数自动调用LeaveCriticalSection
class auto_entercs
{
public:
    auto_entercs(CRITICAL_SECTION* pcs)
    {
        m_pcs = pcs;
 
        if (m_pcs)
        {
            EnterCriticalSection(m_pcs);
        }
    }
 
    ~auto_entercs()
    {
        if (m_pcs)
        {
            LeaveCriticalSection(m_pcs);
        }
    }
 
    CRITICAL_SECTION* m_pcs;
};
 
#endif