thread_pool.cpp
上传用户:yhdzpy8989
上传日期:2007-06-13
资源大小:13604k
文件大小:4k
源码类别:

生物技术

开发平台:

C/C++

  1. /*
  2.  * ===========================================================================
  3.  * PRODUCTION $Log: thread_pool.cpp,v $
  4.  * PRODUCTION Revision 1000.2  2004/06/03 19:28:20  gouriano
  5.  * PRODUCTION PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.4
  6.  * PRODUCTION
  7.  * ===========================================================================
  8.  */
  9. /*  $Id: thread_pool.cpp,v 1000.2 2004/06/03 19:28:20 gouriano Exp $
  10. * ===========================================================================
  11. *
  12. *                            PUBLIC DOMAIN NOTICE
  13. *               National Center for Biotechnology Information
  14. *
  15. *  This software/database is a "United States Government Work" under the
  16. *  terms of the United States Copyright Act.  It was written as part of
  17. *  the author's official duties as a United States Government employee and
  18. *  thus cannot be copyrighted.  This software/database is freely available
  19. *  to the public for use. The National Library of Medicine and the U.S.
  20. *  Government have not placed any restriction on its use or reproduction.
  21. *
  22. *  Although all reasonable efforts have been taken to ensure the accuracy
  23. *  and reliability of the software and data, the NLM and the U.S.
  24. *  Government do not and cannot warrant the performance or results that
  25. *  may be obtained by using this software or data. The NLM and the U.S.
  26. *  Government disclaim all warranties, express or implied, including
  27. *  warranties of performance, merchantability or fitness for any particular
  28. *  purpose.
  29. *
  30. *  Please cite the author in any work or product based on this material.
  31. *
  32. * ===========================================================================
  33. *
  34. * Author:  Aaron Ucko
  35. *
  36. * File Description:
  37. *   Pools of generic request-handling threads.
  38. */
  39. #include <ncbi_pch.hpp>
  40. #include <corelib/ncbistd.hpp>
  41. #include <util/thread_pool.hpp>
  42. BEGIN_NCBI_SCOPE
  43. class CFatalRequest : public CStdRequest
  44. {
  45. public:
  46.     CFatalRequest(CSemaphore* sem) : m_Sem(sem) {}
  47. protected:
  48.     // Kill the current thread
  49.     virtual void Process(void);
  50. private:
  51.     CSemaphore* m_Sem;
  52. };
  53. void CFatalRequest::Process(void)
  54.     if (m_Sem) {
  55.         m_Sem->Post();
  56.     }
  57.     CThread::Exit(0);
  58. }
  59. void CStdPoolOfThreads::KillAllThreads(bool wait)
  60. {
  61.     unsigned int   n;
  62.     {{
  63.         CMutexGuard guard(m_Mutex);
  64.         m_MaxThreads = 0;  // Forbid spawning new threads
  65.         n = m_ThreadCount.Get(); // Capture for use without mutex
  66.     }}
  67.     auto_ptr<CSemaphore> sem;
  68.     if (wait) {
  69.         sem.reset(new CSemaphore(0, n));
  70.     }
  71.     CRef<CStdRequest> poison(new CFatalRequest(sem.get()));
  72.     for (unsigned int i = 0;  i < n;  i++) {
  73.         AcceptRequest(poison);
  74.     }
  75.     if (wait) {
  76.         // Wait for all threads to post to the semaphore
  77.         for (unsigned int i = 0;  i < n  &&  m_ThreadCount.Get();  i++) {
  78.             sem->Wait();
  79.         }
  80.     }
  81. }
  82. END_NCBI_SCOPE
  83. /*
  84. * ===========================================================================
  85. *
  86. * $Log: thread_pool.cpp,v $
  87. * Revision 1000.2  2004/06/03 19:28:20  gouriano
  88. * PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.4
  89. *
  90. * Revision 1.4  2004/06/02 17:50:49  ucko
  91. * CPoolOfThreads: change type of m_Delta and m_ThreadCount to
  92. * CAtomicCounter to reduce need for m_Mutex.
  93. * KillAllThreads: make poison a CRef in the first place to make certain
  94. * it doesn't get GCed prematurely.
  95. * Move CVS log to end.
  96. *
  97. * Revision 1.3  2004/05/17 21:06:02  gorelenk
  98. * Added include of PCH ncbi_pch.hpp
  99. *
  100. * Revision 1.2  2002/11/04 21:29:22  grichenk
  101. * Fixed usage of const CRef<> and CRef<> constructor
  102. *
  103. * Revision 1.1  2001/12/11 19:54:45  ucko
  104. * Introduce thread pools.
  105. *
  106. * ===========================================================================
  107. */