osmutex.h

来自「linux程序」· C头文件 代码 · 共 48 行

#ifndef _COMPUNICATE_OSMUTEX_H_
#define _COMPUNICATE_OSMUTEX_H_

#include <stdlib.h>
#include "OSHead.h"

#ifdef __REDHATLINUX__
    #include <pthread.h>
#endif    

// class OSCond;
	class OSMutex
    {
    public:
        OSMutex();
        ~OSMutex();

		UInt32 TryLock();
        void Lock();
        void Unlock();
    private:

    #ifdef __Win32__
        CRITICAL_SECTION mMutex;
    #endif
    
    #ifdef __REDHATLINUX__
        pthread_mutex_t mMutex;
        // pthread_mutexattr_t attr;
    #endif

    //        friend class OSCond;
    };

    class OSMutexLocker
    {
        public:
        OSMutexLocker(OSMutex *inMutexP) : mMutex(inMutexP) { if (mMutex != NULL) mMutex->Lock(); }
        ~OSMutexLocker() {  if (mMutex != NULL) mMutex->Unlock(); }
        void Lock()         { if (mMutex != NULL) mMutex->Lock(); }
        void Unlock()       { if (mMutex != NULL) mMutex->Unlock(); }

        private:
        OSMutex*    mMutex;
    };

#endif //_OSMUTEX_H_