Mercurial > hg > nginx

comparison src/core/ngx_rwlock.c @ 6101:682d8222c6b1

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Core: read/write locks.
author Ruslan Ermilov <ru@nginx.com>
date Sat, 21 Mar 2015 14:05:08 +0300
parents
children 79ddb0bdb273
comparison
equal deleted inserted replaced
6100:c44459611d91 6101:682d8222c6b1
1
2 /*
3 * Copyright (C) Ruslan Ermilov
4 * Copyright (C) Nginx, Inc.
5 */
6
7
8 #include <ngx_config.h>
9 #include <ngx_core.h>
10
11
12 #if (NGX_HAVE_ATOMIC_OPS)
13
14
15 #define NGX_RWLOCK_SPIN 2048
16 #define NGX_RWLOCK_WLOCK ((ngx_atomic_uint_t) -1)
17
18
19 void
20 ngx_rwlock_wlock(ngx_atomic_t *lock)
21 {
22 ngx_uint_t i, n;
23
24 for ( ;; ) {
25
26 if (*lock == 0 && ngx_atomic_cmp_set(lock, 0, NGX_RWLOCK_WLOCK)) {
27 return;
28 }
29
30 if (ngx_ncpu > 1) {
31
32 for (n = 1; n < NGX_RWLOCK_SPIN; n <<= 1) {
33
34 for (i = 0; i < n; i++) {
35 ngx_cpu_pause();
36 }
37
38 if (*lock == 0
39 && ngx_atomic_cmp_set(lock, 0, NGX_RWLOCK_WLOCK))
40 {
41 return;
42 }
43 }
44 }
45
46 ngx_sched_yield();
47 }
48 }
49
50
51 void
52 ngx_rwlock_rlock(ngx_atomic_t *lock)
53 {
54 ngx_uint_t i, n;
55 ngx_atomic_uint_t readers;
56
57 for ( ;; ) {
58 readers = *lock;
59
60 if (readers != NGX_RWLOCK_WLOCK
61 && ngx_atomic_cmp_set(lock, readers, readers + 1))
62 {
63 return;
64 }
65
66 if (ngx_ncpu > 1) {
67
68 for (n = 1; n < NGX_RWLOCK_SPIN; n <<= 1) {
69
70 for (i = 0; i < n; i++) {
71 ngx_cpu_pause();
72 }
73
74 readers = *lock;
75
76 if (readers != NGX_RWLOCK_WLOCK
77 && ngx_atomic_cmp_set(lock, readers, readers + 1))
78 {
79 return;
80 }
81 }
82 }
83
84 ngx_sched_yield();
85 }
86 }
87
88
89 void
90 ngx_rwlock_unlock(ngx_atomic_t *lock)
91 {
92 ngx_atomic_uint_t readers;
93
94 readers = *lock;
95
96 if (readers == NGX_RWLOCK_WLOCK) {
97 *lock = 0;
98 return;
99 }
100
101 for ( ;; ) {
102
103 if (ngx_atomic_cmp_set(lock, readers, readers - 1)) {
104 return;
105 }
106
107 readers = *lock;
108 }
109 }
110
111
112 #endif