《WINDOWS核心编程》上的一个同步对象

      今天来到公司，把《WINDOWS核心编程》中的一个简单的同步对象改了一下，准备用在以后的开发中。
     起因是这样的，正在做一个聊天服务器，每个聊天房间里有一个双链表，存放的是当前用户对象，默认情况下，用户的发言都是广播，也就是说某个用户发言要转发给别的所有用户，对这个双链表进行遍历发送。同时，可能随时有新用户加入这个房间和退出这个房间，这时候需要对这个双链表的结点移除。典型的多线程读单线程写问题。刚好《WINDOWS核心编程》的第十章有一个这样的封装类，把拿来稍微修改了一下，从第256页最上面的代码的注释里可以看到：NOTE：It's possible that readers could never get access if there are always writes wanting to write。为了避免这个情况出现，我把 Done() 这个函数的实现代码稍微修改了一下,经过在SMP的机器上测试，发现确实避免了 Read 线程永远被调度不到的情况，而且让读线程和写线程之间的调度更加平均一些。原来的代码总是优先调度写线程，修改后的代码根据上一次完成的是读还是写操作来进行交叉调度，如果上一次刚完成了一系列的读，那么接下来就如果有写线程在等待就调度写线程，没有写线程等待才去调度读线程。如果上一次完成的操作是一个线程的写操作，接下来如果有读线程在等待就调度读线程，没有读线程等待的情况下才去调度写线程。
       另外，还做了个小小的优化，把原来代码中的InitializeCriticalSection用InitializeCriticalSectionAndSpinCount取代了，在SMP的机器上。这个函数还是有点用，性能的提高总是一点点的积累起来的。

 HANDLE hSem   = NULL;
 LONG   lCount = 1;
 BOOL   bRead  = TRUE;

 EnterCriticalSection(&m_cs);
 if (m_nActive > 0)
 {
  --m_nActive;
 }
 else
 {
  ++m_nActive;
  bRead = FALSE;
 }

 if (m_nActive == 0)
 {
  if (bRead)
  {
   if (m_nWaitWriters > 0)
   {
    m_nActive = -1;
    hSem      = m_hWriteSem;
    --m_nWaitWriters;
   }
   else if (m_nWaitReaders > 0)
   {
    lCount = m_nActive = m_nWaitReaders;
    m_nWaitReaders = 0;
    hSem           = m_hReadSem;
   }
  }
  else
  {
   if (m_nWaitReaders > 0)
   {
    lCount = m_nActive = m_nWaitReaders;
    m_nWaitReaders = 0;
    hSem           = m_hReadSem;
   }
   else if (m_nWaitWriters > 0)
   {
    m_nActive = -1;
    hSem      = m_hWriteSem;
    --m_nWaitWriters;
   }
  }
 }
 LeaveCriticalSection(&m_cs);

 if (hSem)
 {
  ReleaseSemaphore(hSem, lCount, NULL);
 }
 
 return;

本文地址：http://com.8s8s.com/it/it321.htm