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

流媒体/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 VThreadPool_cxx_Version =
  51.     "$Id: VThreadPool.cxx,v 1.3 2001/07/20 02:29:50 bko Exp $";
  52. #ifndef __vxworks
  53. #include <errno.h>
  54. #include <stdio.h>
  55. #include "VThreadPool.hxx"
  56. #include "cpLog.h"
  57. extern "C" 
  58. {
  59. static void* workerProc(void* args)
  60. {
  61.     return VThreadPool::workerProc(args);
  62. }
  63. }
  64. VThreadPool::VThreadPool(int numThreads)
  65.     :
  66.     numThreads_(numThreads),
  67.     queueClosed_(0),
  68.     shutdown_(0),
  69.     currQueueSize_(0)
  70. {
  71.     init();
  72. }
  73. void
  74. VThreadPool::init() throw (VThreadInitException&)
  75. {
  76.     int ret;
  77.     char buf[256];
  78.     if ((ret = pthread_mutex_init(&queueLock_, 0)) != 0)
  79.     {
  80.         sprintf(buf, "pthread_mutext_init %s", strerror(errno));
  81.         cpLog(LOG_ALERT, buf);
  82.         throw VThreadInitException(buf, __FILE__, __LINE__, errno);
  83.     }
  84.     if ((ret = pthread_cond_init(&queueNotEmpty_, 0)) != 0)
  85.     {
  86.         sprintf(buf, "pthread_cond_init %s", strerror(errno));
  87.         cpLog(LOG_ALERT, buf);
  88.         throw VThreadInitException(buf, __FILE__, __LINE__, errno);
  89.     }
  90.     if ((ret = pthread_cond_init(&queueNotFull_, 0)) != 0)
  91.     {
  92.         sprintf(buf, "pthread_cond_init %s", strerror(errno));
  93.         cpLog(LOG_ALERT, buf);
  94.         throw VThreadInitException(buf, __FILE__, __LINE__, errno);
  95.     }
  96.     if ((ret = pthread_cond_init(&queueEmpty_, 0)) != 0)
  97.     {
  98.         sprintf(buf, "pthread_cond_init %s", strerror(errno));
  99.         cpLog(LOG_ALERT, buf);
  100.         throw VThreadInitException(buf, __FILE__, __LINE__, errno);
  101.     }
  102.     ///create threads
  103.     for (int i = 0 ; i < numThreads_; i++)
  104.     {
  105.         VThread* thread = new VThread();
  106.         workerThreads_.push_back(thread);
  107.         ret = thread->spawn(::workerProc, this);
  108.         if (ret != 0)
  109.         {
  110.             sprintf(buf, "pthread_create %s", strerror(errno));
  111.             cpLog(LOG_ALERT, buf);
  112.             throw VThreadInitException(buf, __FILE__, __LINE__, errno);
  113.         }
  114.     }
  115. }
  116. void
  117. VThreadPool::addFunctor(const VFunctor& functor) throw (VThreadInitException&)
  118. {
  119.     pthread_mutex_lock(&queueLock_);
  120.     if (shutdown_ || queueClosed_)
  121.     {
  122.         pthread_mutex_unlock(&queueLock_);
  123.         cpLog(LOG_DEBUG, "Pool shutting down..");
  124.         throw VThreadInitException("Pool shutting down..",
  125.                                    __FILE__, __LINE__, 0);
  126.     }
  127.     if (currQueueSize_ == 0)
  128.     {
  129.         queue_.push_back(functor);
  130.         pthread_cond_signal(&queueNotEmpty_);
  131.     }
  132.     else
  133.     {
  134.         queue_.push_back(functor);
  135.         pthread_cond_signal(&queueNotEmpty_);
  136.     }
  137.     currQueueSize_++;
  138.     pthread_mutex_unlock(&queueLock_);
  139. }
  140. void
  141. VThreadPool::shutdown(bool finish)
  142. {
  143.     int ret;
  144.     if ((ret = pthread_mutex_lock(&queueLock_)) != 0)
  145.     {
  146.         cpLog(LOG_DEBUG, "Failed to get the lock, still shutting down..");
  147. assert(0);
  148. return;
  149.     }
  150.     if (queueClosed_ || shutdown_)
  151.     {
  152.         pthread_mutex_unlock(&queueLock_);
  153.         return ;
  154.     }
  155.     queueClosed_ = 1;
  156.     if (finish)
  157.     {
  158.         // Wait till the entire queue is finished
  159.         while (currQueueSize_ != 0)
  160.         {
  161.             pthread_cond_wait(&queueEmpty_, &queueLock_);
  162.         }
  163.     }
  164.     shutdown_ = 1;
  165.     pthread_mutex_unlock(&queueLock_);
  166.     // Wakeup all worker threads so that they can recheck shutdown flag
  167.     pthread_cond_broadcast(&queueNotEmpty_);
  168.     pthread_cond_broadcast(&queueNotFull_);
  169.     // wait for workers to exit
  170.     for (ListOfThreads::iterator itr = workerThreads_.begin();
  171.             itr != workerThreads_.end(); itr++)
  172.     {
  173.         Sptr < VThread > vth = (*itr);
  174.         if ((ret = vth->join()) != 0)
  175.         {
  176.             cpLog(LOG_ALERT, "Failed to join thread (%d)", vth->getId());
  177.         }
  178.     }
  179. }
  180. void*
  181. VThreadPool::workerProc(void* args)
  182. {
  183.     VThreadPool& self = *(static_cast < VThreadPool* > (args));
  184.     cpLog(LOG_DEBUG, "Creating worker thread (%d)", pthread_self());
  185.     for (;;)
  186.     {
  187.         pthread_mutex_lock(&(self.queueLock_));
  188.         while ((self.currQueueSize_ == 0) && !(self.shutdown_))
  189.         {
  190.             pthread_cond_wait(&(self.queueNotEmpty_), &(self.queueLock_));
  191.         }
  192.         if (self.shutdown_)
  193.         {
  194.             pthread_mutex_unlock(&(self.queueLock_));
  195.             cpLog(LOG_DEBUG, "Thread (%d) exiting.." , pthread_self());
  196.             pthread_exit(0);
  197.         }
  198. VFunctor work = (self.queue_.front());
  199.         self.queue_.pop_front();
  200.         self.currQueueSize_--;
  201.         if (self.currQueueSize_ == 0)
  202.         {
  203.             pthread_cond_signal(&(self.queueEmpty_));
  204.         }
  205.         pthread_mutex_unlock(&(self.queueLock_));
  206.         try
  207.         {
  208. //            if (work)
  209. //            {
  210.                 cpLog(LOG_DEBUG, "Thread (%d) Doing work." , pthread_self());
  211.                 work.doWork();
  212. //            }
  213.         }
  214.         catch (VException& e)
  215.         {
  216.             cpLog(LOG_ALERT, "Work threw an exception: %s", e.getDescription().c_str());
  217.         }
  218.     }
  219.    return 0;
  220. }
  221. VThreadPool::~VThreadPool()
  222. {
  223.     cpLog(LOG_DEBUG, "VThreadPool::~VThreadPool");
  224.     shutdown();
  225. }
  226. #endif