用ATL建立轻量级的COM对象（三）

第一部分:为什么要使用ATL。

第二部分:起步篇。

实现IUnknown

用纯粹的C++实现IUnknown相对来说比较简单。IUnknown实现之间的主要差别重点在于QueryInterface中将给出哪些接口。请看下列接口定义：

01.interface IMessageSource : IUnknown {

02.HRESULT GetNextMessage([out] OLECHAR **ppwsz);

03.}

04.interface IPager : IUnknown {

05.HRESULT SendMessage([in] const OLECHAR *pwsz);

06.}

07.interface IPager2 : IPager {

08.HRESULT SendUrgentMessage(void);

09.}

这些C++类定义实现了三个接口：

01.class CPager : public IMessageSource, public IPager2 {

02.LONG m_dwRef;

03.public:

04.CPager(void) :   

05.m_dwRef(0) {}

06.virtual ~CPager(void) {}

07.STDMETHODIMP

08.QueryInterface(REFIID, 

09.void**);

10.STDMETHODIMP_(ULONG)

11.AddRef(void);

12.STDMETHODIMP_(ULONG)

13.Release(void);

14.STDMETHODIMP GetNextMessage(OLECHAR **ppwsz);

15.STDMETHODIMP SendMessage(const COLECHAR * pwsz);

16.STDMETHODIMP SendUrgentMessage(void);

17.};

 如果在堆中创建对象（也就是说用new操作符在内部创建）并且只用单线程公寓（STA）模式运行，下面是合理的AddRef 和Release实现：

01.STDMETHODIMP_(ULONG) CPager::AddRef() {

02.return ++m_dwRef;

03.}

04.STDMETHODIMP_(ULONG) CPager::Release(){

05.ULONG result = -m_dwRef;

06.if (result == 0)

07.delete this;

08.return result;

09.}

 如果输出的对象是以多线程公寓（MTA）模式运行，则++和--操作符就必须用Win32的原子增量和减量（Increment/Decrement）例程调用来代替：

01.STDMETHODIMP_(ULONG) CPager::AddRef() {

02.return InterlockedIncrement(&m_dwRef);

03.}

04.STDMETHODIMP_(ULONG) CPager::Release(){

05.ULONG result = InterlockedDecrement(&m_dwRef);

06.if (result == 0)

07.delete this;

08.return result;

09.}

无论哪一种线程模式，下面的QueryInterface实现都是正确的：

01.STDMETHODIMP CPager::QueryInterface(REFIID riid, void **ppv) {

02.if (riid == IID_IUnknown)

03.*ppv = (IMessageSource*)this;

04.else if (riid == IID_IMessageSource)

05.*ppv = (IMessageSource*)this;

06.else if (riid == IID_IPager)

07.*ppv = (IPager*)this;

08.else if (riid == IID_IPager2)

09.*ppv = (IPager2*)this;

10.else

11.return (*ppv = 0), E_NOINTERFACE;

12.((IUnknown*)*ppv)->AddRef();

13.return S_OK;

14.}

QueryInterface的最后四行代码对所有的对象都一样。其余的部分则根据这个对象类型层次上的类不同而有所不同。

如果IUnknown的实现能形成规律，那么ATL便可以提供一种机制将这些具有共性的程序语句从代码中提取出来。实际上ATL做到了这一点，方法是通过提供灵活和可扩展的类层次，使得开发人员只要正确地说明所使用的类集，就可确定线程，服务器锁定和对象生命期行为。

如果看一看ATL实现的IUnknown类层次，你碰到的第一个参数化行为就是线程。ATL允许你构造被优化的对象和服务器，在相互转换STA和MTA的用法时，不用修改源代码。缺省情况下，ATL工程构造的是安全的MTA（MTA-safe）对象，除了要具备构造单STA对象所需的一切外，还需要附加代码和状态管理。通过定义适当的预处理指令，你能改变缺省的线程行为，从而支持单STA或多个基于STA的工程。

使用如下这个指令：

1./D _ATL_SINGLE_THREADED

来编译工程可以改变服务器缺省的线程模型，让它只支持一个基于STA的线程。它适合于进程外的且不创建自拥有线程的基于STA的服务器情况，当你用这个选项时，所有对ATL全局状态的存取将都是不加锁的，并发的。尽管此选项似乎很有效，但它实质上限制了ATL服务器只能是一个单线程的。

使用如下这个指令：

1./D _ATL_APARTMENT_THREADED

来编译工程可以改变服务器缺省的线程模型支持多个基于STA的线程。它适合于建立注册表项ThreadingModel=Apartment的进程内服务器。如果要创建基于STA的进程外服务器且还要建立附加的基于STA的线程，那么这个指令也是必须的。使用这个选项导致ATL用能安全存取线程的临界区来保护它的全局状态。

使用如下这个指令：

1./D _ATL_FREE_THREADED

可以创建与任何线程环境兼容的服务器。也就是说ATL的全局状态将在临界区中被锁定，并且每个对象将拥有它自己的私有临界区来保护它的实例状态。如果没有定义这些指令，则ATL头文件假设为使用_ATL_FREE_THREADED。

为了在ATL中正确使用线程，必须理解ATL的线程模型概念。ATL定义了三种线程模型类来实现线程安全行为所需的少数内联操作。每一种线程模型都输出两个静态函数，Increment 和Decrement。它们都有一个长整指针参数，并且对指定的线程模型实现最快的，合法的增减操作。每一种线程模型都输出两个嵌套类型定义（typedefs），即AutoCriticalSection 和CriticalSection，它们要么打包Win32的CRITICAL_SECTION，要么就是出于兼容性考虑的空类，没有实际实现。记住，所用临界区实际使用的实际类型依赖于特定的线程模型。

ATL实现的三种线程模型分别是CComMultiThreadModel，CComSingleThreadModel和 CComMultiThreadModelNoCS。CComMultiThreadModel使用InterlockedIncrement/InterlockedDecrement和实的CRITICAL_SECTIONS。CComSingleThreadModel使用更有效的++和-操作符及虚的CRITICAL_SECTIONS。

混合的CComMultiThreadModelNoCS除了使用虚的CRITICAL_SECTIONS外，还有InterlockedIncrement/InterlockedDecrement。第三种模型对于存在于MTAs中，但不需要任何数据成员的锁定的对象很有用。

ATL提供了两个类型定义，CComObjectThreadModel 和 CComGlobalsThreadModel，通过条件编译来保证对象和全局变量各自的效率及行为安全。依据所定义的三种预编译指令之一，每一类型名对应着以上描述的三种线程模型类之一。下表说明了这种对应关系，它依赖于ATL所使用的预处理指令。

只要给定了上述的线程模型类型层次，你就能将相应的参数化线程行为添加到任何COM类。请看下列代码：

01.参数化的线程

02. 

03.class CPager : public IPager {

04.LONG m_dwRef;

05.typedef CComObjectThreadModel _ThreadModel;

06._ThreadModel::CComAutoCriticalSection m_critsec;

07.:   :   :   :

08.STDMETHODIMP_(ULONG) CPager::AddRef() {

09.return _ThreadModel::Increment(&m_dwRef);

10.}

11.STDMETHODIMP_(ULONG) CPager::Release(){

12.ULONG res = _ThreadModel::Decrement(&m_dwRef);

13.if (res == 0)

14.delete this;

15.return res;

16.}

17.STDMEHTHODIMP SendUrgentMessage() {

18.// 保证只有一个线程 

19.m_critsec.Lock();

20.// 实现任务

21.this->GenerateMessage();

22.this->WakeUpUser();

23.// 允许其它线程

24.m_critsec.Unlock();

25.return S_OK;

26.}

27.};

使用缺省选项（_ATL_FREE_THREADED）的编译则将一个实临界区添加到对象，并执行Lock和Unlock方法将内联调用映射到EnterCriticalSection/LeaveCriticalSection API函数。同时，AddRef和Release方法将使用InterlockedIncrement/InterlockedDecrement来安全地改变这个对象的引用计数。

如果前面的代码是用_ATL_APARTMENT_THREADED 或者 _ATL_SINGLE_ THREADED选项编译的，则m_critsee数据成员将为空，Lock和Unlock内联例程将变成虚操作，并且AddRef和Release方法将使用++和--操作符。如果这个对象是一个ActiveX控件且其线程模型为Apartment （ThreadingModel=Apartment）的进程内服务器，则这将是小而快的代码。

有时构造以MTA模式（即它的AddRef和Release必须是线程安全的）运行，但不需要附加锁定的对象很有用。对于这种类型的对象，混合型的CComMultiThreadModelNoCS很适合。

通过将类的类型定义从：

1.typedef CComObjectThreadModel _ThreadModel;

细化到

1.typedef CComMultiThreadModelNoCS _ThreadModel;

那么针对每一个对象，你不必付出CRITICAL_SECTION的开销（CComAutoCriticalSection 会映射到 CComFakeCriticalSection）就可以得到线程安全的AddRef和Release（将Increment 和 Decrement方法映射到InterlockedIncrement和InterlockedDecrement）。（待续）

http://www.vckbase.com/index.php/wv/180.html