《C/C++ 面试 100 例》（十六）自动锁

一、锁

• 锁 是 多线程编程 中一个很常用的概念，这里不多加介绍其原理，有兴趣可以参考 临界区 进行更多的了解；
• 锁 一般会提供三个接口：加锁（Lock）、解锁（UnLock）、尝试加锁（TryLock）；
• 本文将利用 C++ 的 构造函数 和 析构函数，把 加锁 和 解锁 在一行代码中实现；

二、自动锁

1、普通锁实现

• 这里利用 CRITICAL_SECTION 相关的接口实现了一系列普通锁的接口封装；

class ILock
{
public:
    ILock(){}
    virtual ~ILock(){}
    virtual void Lock(void) = 0;
    virtual bool TryLock(void) = 0;
    virtual void UnLock(void) = 0;
};

class NLock : public ILock
{
public:
    NLock (){
        InitializeCriticalSection(&m_kSection);
    }
    virtual ~NLock (){
        DeleteCriticalSection(&m_kSection);
    }
    virtual void Lock(void){
        EnterCriticalSection(&m_kSection);
    }
    virtual bool TryLock(void){
        return (TryEnterCriticalSection(&m_kSection) > 0);
    }
    virtual void UnLock(void){
        LeaveCriticalSection(&m_kSection);
    }
protected:
    CRITICAL_SECTION m_kSection;
};

2、自动锁实现

• 1）构造函数：接收传入的 锁指针，并且存成成员变量；
• 2）析构函数：对成员变量的锁执行解锁操作；

class AutoLock   // 注意：这里不继承 ILock
{
public:
    AutoLock(ILock *pkLock) {
        m_pkLock = pkLock;
        m_pkLock->Lock();
    }
    ~AutoLock() {
        m_pkLock->UnLock();
    }
protected:
    ILock*	m_pkLock;
};

3、自动锁使用

int main() {
	NLock m_kLock;
	{
		AutoLock kAutoLock(&m_kLock);   // 1）构造函数中执行 m_kLock 的加锁
		// 执行对应加锁后的操作
	}
	// 2）退出 kAutoLock 作用域，自动调用析构，执行 m_kLock 解锁
	return 0;
}

• 1）构造函数中执行 m_kLock 的加锁；
• 2）退出 kAutoLock 作用域，自动调用析构，执行 m_kLock 解锁；