临界区(锁)windows/linux通用

来源:互联网 发布:性价比高的显卡知乎 编辑:程序博客网 时间:2024/06/06 07:35

//---------------------------------------------------------------------------

//                                      lock.h

//---------------------------------------------------------------------------

#pragma once


#ifdef WIN32

    #include <Windows.h>

    #include <process.h>


    class CLock

    {

    public:

        CLock()        { InitializeCriticalSection(&m_CriticalSection);           }

        ~CLock()       { DeleteCriticalSection(&m_CriticalSection);               }

        int Lock()     { EnterCriticalSection(&m_CriticalSection); return true;   }

        int Unlock()   { LeaveCriticalSection(&m_CriticalSection); return true;   }


    private:

        CRITICAL_SECTION m_CriticalSection;

    };


#else   // POSIX - linux


    #include <pthread.h>


    class CLock

    {

    public:

        CLock()        { pthread_mutex_init(&m_Mutex, NULL);                                       }

        ~CLock()       { pthread_mutex_destroy(&m_Mutex);                                          }

        int Lock()     { int nRetCode = pthread_mutex_lock(&m_Mutex);      return (nRetCode == 0); }

        int Unlock()   { int nRetCode = pthread_mutex_unlock(&m_Mutex);    return (nRetCode == 0); }


    private:

        pthread_mutex_t m_Mutex;

    };


#endif


//---------------------------------------------------------------------------

//                                      test.cpp

//---------------------------------------------------------------------------


int main(int argc, char** argv)

{

    CLock lock;


    lock.Lock();

    lock.Unlock();

    return 0;

}

 

 

转载自 http://zhidao.baidu.com/question/462443960.html

原创粉丝点击