VThreadPoolConnectTest.cxx
上传用户:sy_wanhua
上传日期:2013-07-25
资源大小:3048k
文件大小:4k
源码类别:

流媒体/Mpeg4/MP4

开发平台:

C/C++

  1. /* ====================================================================
  2.  * The Vovida Software License, Version 1.0 
  3.  * 
  4.  * Copyright (c) 2000 Vovida Networks, Inc.  All rights reserved.
  5.  * 
  6.  * Redistribution and use in source and binary forms, with or without
  7.  * modification, are permitted provided that the following conditions
  8.  * are met:
  9.  * 
  10.  * 1. Redistributions of source code must retain the above copyright
  11.  *    notice, this list of conditions and the following disclaimer.
  12.  * 
  13.  * 2. Redistributions in binary form must reproduce the above copyright
  14.  *    notice, this list of conditions and the following disclaimer in
  15.  *    the documentation and/or other materials provided with the
  16.  *    distribution.
  17.  * 
  18.  * 3. The names "VOCAL", "Vovida Open Communication Application Library",
  19.  *    and "Vovida Open Communication Application Library (VOCAL)" must
  20.  *    not be used to endorse or promote products derived from this
  21.  *    software without prior written permission. For written
  22.  *    permission, please contact vocal@vovida.org.
  23.  *
  24.  * 4. Products derived from this software may not be called "VOCAL", nor
  25.  *    may "VOCAL" appear in their name, without prior written
  26.  *    permission of Vovida Networks, Inc.
  27.  * 
  28.  * THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESSED OR IMPLIED
  29.  * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  30.  * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, TITLE AND
  31.  * NON-INFRINGEMENT ARE DISCLAIMED.  IN NO EVENT SHALL VOVIDA
  32.  * NETWORKS, INC. OR ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT DAMAGES
  33.  * IN EXCESS OF $1,000, NOR FOR ANY INDIRECT, INCIDENTAL, SPECIAL,
  34.  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  35.  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  36.  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
  37.  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  38.  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
  39.  * USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
  40.  * DAMAGE.
  41.  * 
  42.  * ====================================================================
  43.  * 
  44.  * This software consists of voluntary contributions made by Vovida
  45.  * Networks, Inc. and many individuals on behalf of Vovida Networks,
  46.  * Inc.  For more information on Vovida Networks, Inc., please see
  47.  * <http://www.vovida.org/>.
  48.  *
  49.  */
  50. static const char* const VThreadPoolConcurrencyTest_cxx_Version =
  51.     "$Id: VThreadPoolConnectTest.cxx,v 1.1 2001/07/20 02:29:50 bko Exp $";
  52. #include "VThreadPool.cxx"
  53. #include "Condition.hxx"
  54. #include "cpLog.h"
  55. #include "Tcp_ServerSocket.hxx"
  56. #include <map>
  57. #include "VMutex.h"
  58. #include "Lock.hxx"
  59. using Vocal::Threads::Lock;
  60. VMutex myMutex;
  61. class ConnContainer
  62. {
  63.     public:
  64. Connection conn;
  65. bool active;
  66. ConnContainer()
  67.     : active(false)
  68. {
  69. }
  70. ConnContainer(const Connection& c)
  71.     : conn(c),
  72.       active(false)
  73. {
  74. }
  75. ConnContainer(const ConnContainer& c)
  76.     : conn(c.conn),
  77.       active(c.active)
  78. {
  79. }
  80. const ConnContainer& operator=(const ConnContainer& c)
  81. {
  82.     if(&c != this)
  83.     {
  84. conn = c.conn;
  85. active = c.active;
  86.     }
  87.     return *this;
  88. }
  89. };
  90. map <int, ConnContainer> myMap;
  91. void doWork(void* msg)
  92. {
  93.     ConnContainer* conn;
  94.     conn = reinterpret_cast<ConnContainer*>(msg);
  95.     assert(conn);
  96.     cpLog(LOG_ERR, "got connection");
  97.     char buf[256];
  98.     int num;
  99.     conn->conn.readLine(buf, 256, num);
  100.     vusleep(100);
  101.     int fd = conn->conn.getConnId();
  102.     conn->conn.close();
  103.     Lock myLock(myMutex);
  104.     myMap.erase(fd);
  105. }
  106. int main()
  107. {
  108.     VThreadPool pool(10);
  109.     TcpServerSocket server(5000);
  110.     for(;;)
  111.     {
  112. if(server.getServerConn().isReadReady())
  113. {
  114.     ConnContainer conn;
  115.     server.accept(conn.conn);
  116.     myMutex.lock();
  117.     myMap[conn.conn.getConnId()] = conn;
  118.     myMutex.unlock();
  119. }
  120. myMutex.lock();
  121. for(map <int, ConnContainer>::iterator i = myMap.begin(); 
  122.     i != myMap.end() ; 
  123.     ++i)
  124. {
  125.     ConnContainer* ptr = &((*i).second);
  126.     if(ptr->conn.isLive() && ptr->conn.isReadReady())
  127.     {
  128. VFunctor func(doWork, ptr);
  129. pool.addFunctor(func);
  130.     }
  131. }
  132. myMutex.unlock();
  133.     }
  134. }