今天来到公司,把《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;