开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Linux/Unix编程 > 查看源码
wait_queue.C
资源名称：threads-2.0.tar.gz　[点击查看]
上传用户：shtangtang
上传日期：2007-01-04
资源大小：167k
文件大小：2k
源码类别：

Linux/Unix编程

开发平台：

Unix_Linux

wait_queue.C：源码内容
  1. #include "thread.h"
  2. #include "thread_lists.h"
  4. #include "wait_queue.h"
  5. #include "shared.h"
  7. #define OFFS(n)         ((n)%__THREAD_QUEUE_MAX)
  9. wait_queue::wait_queue(int p_id)
  10. {
  11.   w_id = p_id;
  12.   init(attributes::process_shared);
  13. }
  15. wait_queue::wait_queue(attributes::scope p_scope)
  16. {
  17.   w_id = shared_mem::share.create_proj( 0 );
  18.   init(attributes::process_shared);
  19. }
  21. wait_queue::wait_queue()
  22. {
  23.   init(attributes::process_private);
  24. }
  26. wait_queue::~wait_queue()
  27. {
  28.   if ( _wq ) {
  29.     while( !empty() )
  30.       wake_up();
  31.     if ( w_scope == attributes::process_private )
  32.       delete _wq;
  33.     else
  34.       shared_mem::share.dealloc( _wq );
  35.     _wq = 0;
  36.   }
  37. }
  39. void wait_queue::init(attributes::scope p_scope)
  40. {
  41.   if ( w_id == -1 )
  42.     w_scope     = attributes::process_private;
  43.   else
  44.     w_scope     = p_scope;
  45.   if ( w_scope == attributes::process_shared ) {
  46.     _wq         = (storage *)shared_mem::share.alloc( sizeof(storage),w_id );
  47.     assert( _wq != (storage *)-1 );
  48.     if ( _wq->w_magic != 0x212611 ) {
  49.       _wq->top     = 0;
  50.       _wq->bottom  = 0;
  51.       _wq->w_magic = 0x212611;
  52.     }
  53.   } else {
  54.     _wq         = new storage;
  55.     _wq->top    = 0;
  56.     _wq->bottom = 0;
  57.   }
  58. }
  60. bool
  61. wait_queue::empty()
  62. {
  63.   return ( _wq->bottom == _wq->top );
  64. }
  66. void
  67. wait_queue::remove(int p_id)
  68. {
  69.   int pos;
  71.   _wq->w_sync.acquire();
  72.   if ( !empty() ) {
  73.     for( pos = _wq->bottom;pos != OFFS(_wq->top-1);pos = OFFS(pos+1) )
  74.       if ( _wq->pid[pos] == p_id ) {
  75. for( ;_wq->bottom != pos;pos = OFFS(pos-1) )
  76.   _wq->pid[pos] = _wq->pid[OFFS(pos-1)];
  77. _wq->bottom = OFFS(_wq->bottom+1);
  78. break;
  79.       }
  80.   }
  81.   _wq->w_sync.release();
  82. }
  84. void
  85. wait_queue::insert(int p_id)
  86. {
  87.   _wq->w_sync.acquire();
  88.   if ( _wq->bottom == OFFS(_wq->top+1) )
  89.     throw;
  90.   _wq->pid[_wq->top] = p_id;
  91.   _wq->top = OFFS(_wq->top+1);
  92.   _wq->w_sync.release();
  93. }
  95. void
  96. wait_queue::suspend_me()
  97. {
  98.   thread_list::iterator i = thread_list::__threads.locate(getpid());
  101.   if ( i == thread_list::__threads.end() )
  102.     return;
  103.   insert(getpid());
  104.   (*i)->suspend_with_cancelation();
  105.   //__sched_yield();
  106.   _wq->w_sync.acquire();
  107.   //we need to synchronize with the wakeup below.  Maybe __sched.yield()?
  108.   _wq->w_sync.release();
  109. }
  111. void
  112. wait_queue::wake_up()
  113. {
  114.   _wq->w_sync.acquire();
  115.   if ( !empty() ) {
  116.     thread_list::__threads.restart(_wq->pid[_wq->bottom]);
  117.     _wq->bottom = OFFS(_wq->bottom+1);
  118.   }
  119.   _wq->w_sync.release();
  120. }