一个简单的线程锁------pthread和win32的临界区(Critical Section)

 1 //一个简单的线程锁
 2 #ifndef _THREADLOCK_H 
 3 #define _THREADLOCK_H 
 4 
 5 #ifdef WIN32 
 6 #include <Windows.h> 
 7 #else 
 8 #include <pthread.h> 
 9 #endif 
10 
11 #ifdef WIN32 
12 #define LOCK_MUTEXT CRITICAL_SECTION
13 #define INIT_LOCK(X) InitializeCriticalSection(X)
14 #define DO_LOCK(X) EnterCriticalSection(X)
15 #define DO_UNLOCK(X) LeaveCriticalSection(X)
16 #define DESTROY_LOCK(X) DeleteCriticalSection(X)
17 #else 
18 #define LOCK_MUTEXT pthread_mutex_t
19 #define INIT_LOCK(X) pthread_mutex_init(X,NULL)
20 #define DO_LOCK(X) pthread_mutex_lock(X)
21 #define DO_UNLOCK(X) pthread_mutex_unlock(X)
22 #define DESTROY_LOCK(X) pthread_mutex_destroy(X)
23 #endif 
24 
25 class CThreadLock 
26 { 
27 public: 
28     CThreadLock(void) 
29     { 
30         Init(); 
31     }; 
32 
33     ~CThreadLock() 
34     { 
35         Close(); 
36     }; 
37 
38     void Init() 
39     { 
40         INIT_LOCK(&m_lock);
41     }; 
42 
43     void Close() 
44     { 
45         DESTROY_LOCK(&m_lock);
46     } 
47 
48     void Lock() 
49     { 
50         DO_LOCK(&m_lock);
51     }; 
52 
53     void UnLock() 
54     { 
55         DO_UNLOCK(&m_lock);
56     }; 
57 
58 private: 
59     LOCK_MUTEXT m_lock; 
60 }; 
61 
62 //自动加锁的类 
63 class CAutoLock 
64 { 
65 public: 
66     CAutoLock(CThreadLock* pThreadLock)  
67     {  
68         m_pThreadLock = pThreadLock; 
69         if(NULL != m_pThreadLock) 
70         { 
71             m_pThreadLock->Lock(); 
72         } 
73     }; 
74     ~CAutoLock()  
75     { 
76         if(NULL != m_pThreadLock) 
77         { 
78             m_pThreadLock->UnLock(); 
79         } 
80     }; 
81     
82 private: 
83     CThreadLock* m_pThreadLock; 
84 }; 
85 
86 #endif