summaryrefslogtreecommitdiffstats
path: root/src/common/Cond.h
blob: 59ea9350187c815888d7b52461e72276cf41d587 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- 
// vim: ts=8 sw=2 smarttab
/*
 * Ceph - scalable distributed file system
 *
 * Copyright (C) 2004-2006 Sage Weil <sage@newdream.net>
 *
 * This is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License version 2.1, as published by the Free Software 
 * Foundation.  See file COPYING.
 * 
 */


#ifndef __COND_H
#define __COND_H

#include <time.h>

#include "Mutex.h"
#include "Clock.h"

#include "include/Context.h"

#include <pthread.h>

class Cond {
  // my bits
  pthread_cond_t _c;

  // don't allow copying.
  void operator=(Cond &C) {}
  Cond( const Cond &C ) {}

 public:
  Cond() {
    int r = pthread_cond_init(&_c,NULL);
    assert(r == 0);
  }
  virtual ~Cond() { 
    pthread_cond_destroy(&_c); 
  }

  int Wait(Mutex &mutex)  { 
    int r = pthread_cond_wait(&_c, &mutex._m);
    return r;
  }

  int Wait(Mutex &mutex, char* s)  { 
    //cout << "Wait: " << s << endl;
    int r = pthread_cond_wait(&_c, &mutex._m);
    return r;
  }

  int WaitUntil(Mutex &mutex, utime_t when) {
    struct timespec ts;
    g_clock.make_timespec(when, &ts);
    //cout << "timedwait for " << ts.tv_sec << " sec " << ts.tv_nsec << " nsec" << endl;
    int r = pthread_cond_timedwait(&_c, &mutex._m, &ts);
    return r;
  }
  int WaitInterval(Mutex &mutex, utime_t interval) {
    utime_t when = g_clock.now();
    when += interval;
    return WaitUntil(mutex, when);
  }

  int Signal() { 
    //int r = pthread_cond_signal(&_c);
    int r = pthread_cond_broadcast(&_c);
    return r;
  }
  int SignalOne() { 
    int r = pthread_cond_signal(&_c);
    return r;
  }
  int SignalAll() { 
    //int r = pthread_cond_signal(&_c);
    int r = pthread_cond_broadcast(&_c);
    return r;
  }
};

class C_Cond : public Context {
  Cond *cond;
  bool *done;
  int *rval;
public:
  C_Cond(Cond *c, bool *d, int *r=0) : cond(c), done(d), rval(r) {
    *done = false;
  }
  void finish(int r) {
    if (rval) *rval = r;
    *done = true;
    cond->Signal();
  }
};

class C_SafeCond : public Context {
  Mutex *lock;
  Cond *cond;
  bool *done;
  int *rval;
public:
  C_SafeCond(Mutex *l, Cond *c, bool *d, int *r=0) : lock(l), cond(c), done(d), rval(r) {
    *done = false;
  }
  void finish(int r) {
    lock->Lock();
    if (rval) *rval = r;
    *done = true;
    cond->Signal();
    lock->Unlock();
  }
};

#endif