libgo/runtime/go-note.c


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

/* go-note.c -- implement notesleep, notewakeup and noteclear.

   Copyright 2009 The Go Authors. All rights reserved.
   Use of this source code is governed by a BSD-style
   license that can be found in the LICENSE file.  */

/* A note is a one-time notification.  noteclear clears the note.
   notesleep waits for a call to notewakeup.  notewakeup wakes up
   every thread waiting on the note.  */

#include "go-assert.h"
#include "runtime.h"

/* We use a single global lock and condition variable.  It would be
   better to use a futex on Linux.  */

static pthread_mutex_t note_lock = PTHREAD_MUTEX_INITIALIZER;
static pthread_cond_t note_cond = PTHREAD_COND_INITIALIZER;

/* noteclear is called before any calls to notesleep or
   notewakeup.  */

void
noteclear (Note* n)
{
  int32 i;

  i = pthread_mutex_lock (&note_lock);
  __go_assert (i == 0);

  n->woken = 0;

  i = pthread_mutex_unlock (&note_lock);
  __go_assert (i == 0);
}

/* Wait until notewakeup is called.  */

void
notesleep (Note* n)
{
  int32 i;

  i = pthread_mutex_lock (&note_lock);
  __go_assert (i == 0);

  while (!n->woken)
    {
      i = pthread_cond_wait (&note_cond, &note_lock);
      __go_assert (i == 0);
    }

  i = pthread_mutex_unlock (&note_lock);
  __go_assert (i == 0);
}

/* Wake up every thread sleeping on the note.  */

void
notewakeup (Note *n)
{
  int32 i;

  i = pthread_mutex_lock (&note_lock);
  __go_assert (i == 0);

  n->woken = 1;

  i = pthread_cond_broadcast (&note_cond);
  __go_assert (i == 0);

  i = pthread_mutex_unlock (&note_lock);
  __go_assert (i == 0);
}