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

生物技术

开发平台:

C/C++

  1. /*
  2.  * ===========================================================================
  3.  * PRODUCTION $Log: threaded_server.cpp,v $
  4.  * PRODUCTION Revision 1000.1  2004/06/01 18:45:32  gouriano
  5.  * PRODUCTION PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R6.11
  6.  * PRODUCTION
  7.  * ===========================================================================
  8.  */
  9. /*  $Id: threaded_server.cpp,v 1000.1 2004/06/01 18:45:32 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. *   Framework for a multithreaded network server
  38. */
  39. #include <ncbi_pch.hpp>
  40. #include <connect/threaded_server.hpp>
  41. #include <connect/ncbi_socket.hpp>
  42. #include <util/thread_pool.hpp>
  43. BEGIN_NCBI_SCOPE
  44. class CSocketRequest : public CStdRequest
  45. {
  46. public:
  47.     CSocketRequest(CThreadedServer& server, SOCK sock)
  48.         : m_Server(server), m_Sock(sock) {}
  49.     virtual void Process(void)
  50.         { m_Server.Process(m_Sock); }
  51. private:
  52.     CThreadedServer& m_Server;
  53.     SOCK             m_Sock;
  54. };
  55. void CThreadedServer::Run(void)
  56. {
  57.     SetParams();
  58.     if (m_InitThreads <= 0  ||  m_MaxThreads < m_InitThreads
  59.         ||  m_MaxThreads > 1000  ||  m_Port > 65535) {
  60.         NCBI_THROW(CThreadedServerException, eBadParameters,
  61.                    "CThreadedServer::Run: Bad parameters");
  62.     }
  63.     CStdPoolOfThreads pool(m_MaxThreads, m_QueueSize, m_SpawnThreshold);
  64.     pool.Spawn(m_InitThreads);
  65.     CListeningSocket lsock(m_Port);
  66.     if (lsock.GetStatus() != eIO_Success) {
  67.         NCBI_THROW(CThreadedServerException, eCouldntListen,
  68.                    "CThreadedServer::Run: Unable to create listening socket: "
  69.                    + string(strerror(errno)));
  70.     }
  71.     for (;;) {
  72.         CSocket    sock;
  73.         EIO_Status status = lsock.Accept(sock);
  74.         sock.SetOwnership(eNoOwnership); // Process[Overflow] will close it
  75.         if (status == eIO_Success) {
  76.             try {
  77.                 pool.AcceptRequest
  78.                     (CRef<ncbi::CStdRequest>
  79.                      (new CSocketRequest(*this, sock.GetSOCK())));
  80.                 if (pool.IsFull()  &&  m_TemporarilyStopListening) {
  81.                     lsock.Close();
  82.                     pool.WaitForRoom();
  83.                     lsock.Listen(m_Port);
  84.                 }
  85.             } catch (CBlockingQueueException) {
  86.                 _ASSERT(!m_TemporarilyStopListening);
  87.                 ProcessOverflow(sock.GetSOCK());
  88.             }            
  89.         } else {
  90.             ERR_POST("accept failed: " << IO_StatusStr(status));
  91.         }
  92.     }
  93. }
  94. END_NCBI_SCOPE
  95. /*
  96. * ===========================================================================
  97. *
  98. * $Log: threaded_server.cpp,v $
  99. * Revision 1000.1  2004/06/01 18:45:32  gouriano
  100. * PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R6.11
  101. *
  102. * Revision 6.11  2004/05/17 20:58:13  gorelenk
  103. * Added include of PCH ncbi_pch.hpp
  104. *
  105. * Revision 6.10  2003/08/12 19:28:11  ucko
  106. * Throw CThreadedServerException for abnormal exits.
  107. *
  108. * Revision 6.9  2002/11/04 21:29:02  grichenk
  109. * Fixed usage of const CRef<> and CRef<> constructor
  110. *
  111. * Revision 6.8  2002/09/17 18:42:30  ucko
  112. * Use CSocket now that SetOwnership exists.
  113. *
  114. * Revision 6.7  2002/09/13 17:13:27  ucko
  115. * Use CListeningSocket instead of LSOCK, but stick with SOCK to avoid
  116. * double closes.
  117. *
  118. * Revision 6.6  2002/09/13 15:16:25  ucko
  119. * Update for new CBlockingQueue exception setup.
  120. *
  121. * Revision 6.5  2002/08/20 19:23:44  ucko
  122. * Check return status from LSOCK_Create() in CThreadedServer::Run().
  123. * Move CVS log to end of file.
  124. *
  125. * Revision 6.4  2002/01/25 15:39:29  ucko
  126. * Completely reorganized threaded servers.
  127. *
  128. * Revision 6.3  2002/01/24 20:19:18  ucko
  129. * Add magic TemporarilyStopListening overflow processor
  130. * More cleanups
  131. *
  132. * Revision 6.2  2002/01/24 18:35:56  ucko
  133. * Allow custom queue-overflow handling.
  134. * Clean up SOCKs and CONNs when done with them.
  135. *
  136. * Revision 6.1  2001/12/11 19:55:22  ucko
  137. * Introduce thread-pool-based servers.
  138. *
  139. * ===========================================================================
  140. */