Mercurial > hg > nginx
annotate src/http/ngx_http_upstream_round_robin.c @ 1344:8f5b5641966c
fair upstream weight balancer
| author | Igor Sysoev <igor@sysoev.ru> |
|---|---|
| date | Sat, 28 Jul 2007 16:04:01 +0000 |
| parents | be2b895d31e0 |
| children | 6ff8c0c1e194 |
| rev | line source |
|---|---|
| 884 | 1 |
| 2 /* | |
| 3 * Copyright (C) Igor Sysoev | |
| 4 */ | |
| 5 | |
| 6 | |
| 7 #include <ngx_config.h> | |
| 8 #include <ngx_core.h> | |
| 9 #include <ngx_http.h> | |
| 10 | |
| 11 | |
|
1344
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
12 static ngx_uint_t |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
13 ngx_http_upstream_get_peer(ngx_http_upstream_rr_peers_t *peers); |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
14 |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
15 |
| 884 | 16 ngx_int_t |
| 17 ngx_http_upstream_init_round_robin(ngx_conf_t *cf, | |
| 18 ngx_http_upstream_srv_conf_t *us) | |
| 19 { | |
| 20 ngx_url_t u; | |
| 21 ngx_uint_t i, j, n; | |
| 22 ngx_http_upstream_server_t *server; | |
| 23 ngx_http_upstream_rr_peers_t *peers; | |
| 24 | |
| 25 us->peer.init = ngx_http_upstream_init_round_robin_peer; | |
| 26 | |
| 27 if (us->servers) { | |
| 28 n = 0; | |
| 29 server = us->servers->elts; | |
| 30 | |
| 31 for (i = 0; i < us->servers->nelts; i++) { | |
| 32 n += server[i].naddrs; | |
| 33 } | |
| 34 | |
| 35 peers = ngx_pcalloc(cf->pool, sizeof(ngx_http_upstream_rr_peers_t) | |
| 36 + sizeof(ngx_http_upstream_rr_peer_t) * (n - 1)); | |
| 37 if (peers == NULL) { | |
| 38 return NGX_ERROR; | |
| 39 } | |
| 40 | |
| 41 peers->number = n; | |
| 42 peers->name = &us->host; | |
| 43 | |
| 44 n = 0; | |
| 45 | |
| 46 for (i = 0; i < us->servers->nelts; i++) { | |
| 47 for (j = 0; j < server[i].naddrs; j++) { | |
| 48 peers->peer[n].sockaddr = server[i].addrs[j].sockaddr; | |
| 49 peers->peer[n].socklen = server[i].addrs[j].socklen; | |
| 50 peers->peer[n].name = server[i].addrs[j].name; | |
| 51 peers->peer[n].weight = server[i].weight; | |
| 52 peers->peer[n].current_weight = server[i].weight; | |
| 53 peers->peer[n].max_fails = server[i].max_fails; | |
| 54 peers->peer[n].fail_timeout = server[i].fail_timeout; | |
| 55 peers->peer[n].down = server[i].down; | |
| 56 n++; | |
| 57 } | |
| 58 } | |
| 59 | |
| 60 us->peer.data = peers; | |
| 61 | |
| 62 return NGX_OK; | |
| 63 } | |
| 64 | |
| 65 | |
| 66 /* an upstream implicitly defined by proxy_pass, etc. */ | |
| 67 | |
| 906 | 68 if (us->port == 0 && us->default_port == 0) { |
| 69 ngx_log_error(NGX_LOG_EMERG, cf->log, 0, | |
| 70 "no port in upstream \"%V\" in %s:%ui", | |
| 71 &us->host, us->file_name.data, us->line); | |
| 72 return NGX_ERROR; | |
| 73 } | |
| 74 | |
| 884 | 75 ngx_memzero(&u, sizeof(ngx_url_t)); |
| 76 | |
| 77 u.host = us->host; | |
| 916 | 78 u.port = (in_port_t) (us->port ? us->port : us->default_port); |
| 884 | 79 |
| 80 if (ngx_inet_resolve_host(cf, &u) != NGX_OK) { | |
| 81 if (u.err) { | |
| 82 ngx_log_error(NGX_LOG_EMERG, cf->log, 0, | |
|
896
f247db60fc85
fix fastcgi and memcached upstreams
Igor Sysoev <igor@sysoev.ru>
parents:
884
diff
changeset
|
83 "%s in upstream \"%V\" in %s:%ui", |
| 884 | 84 u.err, &us->host, us->file_name.data, us->line); |
| 85 } | |
| 86 | |
| 87 return NGX_ERROR; | |
| 88 } | |
| 89 | |
| 90 n = u.naddrs; | |
| 91 | |
| 92 peers = ngx_pcalloc(cf->pool, sizeof(ngx_http_upstream_rr_peers_t) | |
| 93 + sizeof(ngx_http_upstream_rr_peer_t) * (n - 1)); | |
| 94 if (peers == NULL) { | |
| 95 return NGX_ERROR; | |
| 96 } | |
| 97 | |
| 98 peers->number = n; | |
| 99 peers->name = &us->host; | |
| 100 | |
| 101 n = 0; | |
| 102 | |
| 103 for (i = 0; i < u.naddrs; i++) { | |
| 104 peers->peer[n].sockaddr = u.addrs[i].sockaddr; | |
| 105 peers->peer[n].socklen = u.addrs[i].socklen; | |
| 106 peers->peer[n].name = u.addrs[i].name; | |
| 107 peers->peer[n].weight = 1; | |
| 108 peers->peer[n].current_weight = 1; | |
| 109 peers->peer[n].max_fails = 1; | |
| 110 peers->peer[n].fail_timeout = 10; | |
| 111 n++; | |
| 112 } | |
| 113 | |
| 114 us->peer.data = peers; | |
| 115 | |
| 116 return NGX_OK; | |
| 117 } | |
| 118 | |
| 119 | |
| 120 ngx_int_t | |
| 121 ngx_http_upstream_init_round_robin_peer(ngx_http_request_t *r, | |
| 122 ngx_http_upstream_srv_conf_t *us) | |
| 123 { | |
| 124 ngx_uint_t n; | |
| 125 ngx_http_upstream_rr_peer_data_t *rrp; | |
| 126 | |
| 127 rrp = r->upstream->peer.data; | |
| 128 | |
| 129 if (rrp == NULL) { | |
| 130 rrp = ngx_palloc(r->pool, sizeof(ngx_http_upstream_rr_peer_data_t)); | |
| 131 if (rrp == NULL) { | |
| 132 return NGX_ERROR; | |
| 133 } | |
| 134 | |
| 135 r->upstream->peer.data = rrp; | |
| 136 } | |
| 137 | |
| 138 rrp->peers = us->peer.data; | |
| 139 rrp->current = 0; | |
| 140 | |
| 141 if (rrp->peers->number <= 8 * sizeof(uintptr_t)) { | |
| 142 rrp->tried = &rrp->data; | |
| 143 rrp->data = 0; | |
| 144 | |
| 145 } else { | |
| 146 n = (rrp->peers->number + (8 * sizeof(uintptr_t) - 1)) | |
| 147 / (8 * sizeof(uintptr_t)); | |
| 148 | |
| 149 rrp->tried = ngx_pcalloc(r->pool, n * sizeof(uintptr_t)); | |
| 150 if (rrp->tried == NULL) { | |
| 151 return NGX_ERROR; | |
| 152 } | |
| 153 } | |
| 154 | |
| 155 r->upstream->peer.get = ngx_http_upstream_get_round_robin_peer; | |
| 156 r->upstream->peer.free = ngx_http_upstream_free_round_robin_peer; | |
| 157 r->upstream->peer.tries = rrp->peers->number; | |
| 158 #if (NGX_HTTP_SSL) | |
|
1284
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
159 r->upstream->peer.set_session = |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
160 ngx_http_upstream_set_round_robin_peer_session; |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
161 r->upstream->peer.save_session = |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
162 ngx_http_upstream_save_round_robin_peer_session; |
| 884 | 163 #endif |
| 164 | |
| 165 return NGX_OK; | |
| 166 } | |
| 167 | |
| 168 | |
| 169 ngx_int_t | |
| 170 ngx_http_upstream_get_round_robin_peer(ngx_peer_connection_t *pc, void *data) | |
| 171 { | |
| 172 ngx_http_upstream_rr_peer_data_t *rrp = data; | |
| 173 | |
| 174 time_t now; | |
| 175 uintptr_t m; | |
| 176 ngx_uint_t i, n; | |
| 177 ngx_connection_t *c; | |
| 178 ngx_http_upstream_rr_peer_t *peer; | |
| 179 | |
| 180 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, pc->log, 0, | |
| 181 "get rr peer, try: %ui", pc->tries); | |
| 182 | |
| 183 now = ngx_time(); | |
| 184 | |
| 185 /* ngx_lock_mutex(rrp->peers->mutex); */ | |
| 186 | |
| 187 if (rrp->peers->last_cached) { | |
| 188 | |
| 189 /* cached connection */ | |
| 190 | |
| 191 c = rrp->peers->cached[rrp->peers->last_cached]; | |
| 192 rrp->peers->last_cached--; | |
| 193 | |
| 194 /* ngx_unlock_mutex(ppr->peers->mutex); */ | |
| 195 | |
| 196 #if (NGX_THREADS) | |
| 197 c->read->lock = c->read->own_lock; | |
| 198 c->write->lock = c->write->own_lock; | |
| 199 #endif | |
| 200 | |
| 201 pc->connection = c; | |
| 202 pc->cached = 1; | |
| 203 | |
| 204 return NGX_OK; | |
| 205 } | |
| 206 | |
| 207 pc->cached = 0; | |
| 208 pc->connection = NULL; | |
| 209 | |
| 210 if (rrp->peers->number == 1) { | |
| 211 peer = &rrp->peers->peer[0]; | |
| 212 | |
| 213 } else { | |
| 214 | |
| 215 /* there are several peers */ | |
| 216 | |
| 217 if (pc->tries == rrp->peers->number) { | |
| 218 | |
| 219 /* it's a first try - get a current peer */ | |
| 220 | |
| 221 for ( ;; ) { | |
|
1344
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
222 rrp->current = ngx_http_upstream_get_peer(rrp->peers); |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
223 |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
224 ngx_log_debug2(NGX_LOG_DEBUG_HTTP, pc->log, 0, |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
225 "get rr peer, current: %ui %i", |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
226 rrp->current, |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
227 rrp->peers->peer[rrp->current].current_weight); |
| 884 | 228 |
| 229 n = rrp->current / (8 * sizeof(uintptr_t)); | |
|
1206
451e34738078
fix worker endless loop on 64-bit platform if 33+ backend has fall
Igor Sysoev <igor@sysoev.ru>
parents:
916
diff
changeset
|
230 m = (uintptr_t) 1 << rrp->current % (8 * sizeof(uintptr_t)); |
| 884 | 231 |
| 232 if (!(rrp->tried[n] & m)) { | |
| 233 peer = &rrp->peers->peer[rrp->current]; | |
| 234 | |
| 235 if (!peer->down) { | |
| 236 | |
| 237 if (peer->max_fails == 0 | |
| 238 || peer->fails < peer->max_fails) | |
| 239 { | |
| 240 break; | |
| 241 } | |
| 242 | |
| 243 if (now - peer->accessed > peer->fail_timeout) { | |
| 244 peer->fails = 0; | |
| 245 break; | |
| 246 } | |
| 247 | |
|
1344
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
248 peer->current_weight = 0; |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
249 |
| 884 | 250 } else { |
| 251 rrp->tried[n] |= m; | |
| 252 } | |
| 253 | |
| 254 pc->tries--; | |
| 255 } | |
| 256 | |
| 257 if (pc->tries) { | |
| 258 continue; | |
| 259 } | |
| 260 | |
| 261 goto failed; | |
| 262 } | |
| 263 | |
| 264 peer->current_weight--; | |
| 265 | |
| 266 } else { | |
| 267 for ( ;; ) { | |
| 268 n = rrp->current / (8 * sizeof(uintptr_t)); | |
|
1206
451e34738078
fix worker endless loop on 64-bit platform if 33+ backend has fall
Igor Sysoev <igor@sysoev.ru>
parents:
916
diff
changeset
|
269 m = (uintptr_t) 1 << rrp->current % (8 * sizeof(uintptr_t)); |
| 884 | 270 |
| 271 if (!(rrp->tried[n] & m)) { | |
| 272 | |
| 273 peer = &rrp->peers->peer[rrp->current]; | |
| 274 | |
| 275 if (!peer->down) { | |
| 276 | |
| 277 if (peer->max_fails == 0 | |
| 278 || peer->fails < peer->max_fails) | |
| 279 { | |
| 280 break; | |
| 281 } | |
| 282 | |
| 283 if (now - peer->accessed > peer->fail_timeout) { | |
| 284 peer->fails = 0; | |
| 285 break; | |
| 286 } | |
| 287 | |
|
1344
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
288 peer->current_weight = 0; |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
289 |
| 884 | 290 } else { |
| 291 rrp->tried[n] |= m; | |
| 292 } | |
| 293 | |
| 294 pc->tries--; | |
| 295 } | |
| 296 | |
| 297 rrp->current++; | |
| 298 | |
| 299 if (rrp->current >= rrp->peers->number) { | |
| 300 rrp->current = 0; | |
| 301 } | |
| 302 | |
| 303 if (pc->tries) { | |
| 304 continue; | |
| 305 } | |
| 306 | |
| 307 goto failed; | |
| 308 } | |
| 309 | |
| 310 peer->current_weight--; | |
| 311 } | |
| 312 | |
| 313 rrp->tried[n] |= m; | |
| 314 } | |
| 315 | |
| 316 pc->sockaddr = peer->sockaddr; | |
| 317 pc->socklen = peer->socklen; | |
| 318 pc->name = &peer->name; | |
| 319 | |
| 320 /* ngx_unlock_mutex(rrp->peers->mutex); */ | |
| 321 | |
| 322 return NGX_OK; | |
| 323 | |
| 324 failed: | |
| 325 | |
| 326 /* all peers failed, mark them as live for quick recovery */ | |
| 327 | |
| 328 for (i = 0; i < rrp->peers->number; i++) { | |
| 329 rrp->peers->peer[i].fails = 0; | |
| 330 } | |
| 331 | |
| 332 /* ngx_unlock_mutex(rrp->peers->mutex); */ | |
| 333 | |
| 334 pc->name = rrp->peers->name; | |
| 335 | |
| 336 return NGX_BUSY; | |
| 337 } | |
| 338 | |
| 339 | |
|
1344
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
340 static ngx_uint_t |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
341 ngx_http_upstream_get_peer(ngx_http_upstream_rr_peers_t *peers) |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
342 { |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
343 ngx_uint_t i, n; |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
344 ngx_http_upstream_rr_peer_t *peer; |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
345 |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
346 peer = &peers->peer[0]; |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
347 |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
348 for ( ;; ) { |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
349 |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
350 for (i = 0; i < peers->number; i++) { |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
351 |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
352 if (peer[i].current_weight <= 0) { |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
353 continue; |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
354 } |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
355 |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
356 n = i; |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
357 |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
358 while (i < peers->number - 1) { |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
359 |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
360 i++; |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
361 |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
362 if (peer[i].current_weight <= 0) { |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
363 continue; |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
364 } |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
365 |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
366 if (peer[n].current_weight * 1000 / peer[i].current_weight |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
367 >= peer[n].weight * 1000 / peer[i].weight) |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
368 { |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
369 return n; |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
370 } |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
371 |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
372 n = i; |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
373 } |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
374 |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
375 if (peer[i].current_weight > 0) { |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
376 n = i; |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
377 } |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
378 |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
379 return n; |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
380 } |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
381 |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
382 for (i = 0; i < peers->number; i++) { |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
383 if (peer[i].fails == 0) { |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
384 peer[i].current_weight += peer[i].weight; |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
385 |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
386 } else { |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
387 /* 1 allows to go to quick recovery when all peers failed */ |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
388 peer[i].current_weight = 1; |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
389 } |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
390 } |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
391 } |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
392 } |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
393 |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
394 |
| 884 | 395 void |
| 396 ngx_http_upstream_free_round_robin_peer(ngx_peer_connection_t *pc, void *data, | |
| 397 ngx_uint_t state) | |
| 398 { | |
| 399 ngx_http_upstream_rr_peer_data_t *rrp = data; | |
| 400 | |
| 401 time_t now; | |
| 402 ngx_http_upstream_rr_peer_t *peer; | |
| 403 | |
| 404 ngx_log_debug2(NGX_LOG_DEBUG_HTTP, pc->log, 0, | |
| 405 "free rr peer %ui %ui", pc->tries, state); | |
| 406 | |
| 407 if (state == 0 && pc->tries == 0) { | |
| 408 return; | |
| 409 } | |
| 410 | |
| 411 /* TODO: NGX_PEER_KEEPALIVE */ | |
| 412 | |
| 413 if (rrp->peers->number == 1) { | |
| 414 pc->tries = 0; | |
| 415 return; | |
| 416 } | |
| 417 | |
| 418 if (state & NGX_PEER_FAILED) { | |
| 419 now = ngx_time(); | |
| 420 | |
| 421 peer = &rrp->peers->peer[rrp->current]; | |
| 422 | |
| 423 /* ngx_lock_mutex(rrp->peers->mutex); */ | |
| 424 | |
| 425 peer->fails++; | |
| 426 peer->accessed = now; | |
| 427 | |
|
1344
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
428 peer->current_weight -= peer->weight / peer->max_fails; |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
429 |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
430 ngx_log_debug2(NGX_LOG_DEBUG_HTTP, pc->log, 0, |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
431 "free rr peer failed: %ui %i", |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
432 rrp->current, peer->current_weight); |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
433 |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
434 if (peer->current_weight < 0) { |
|
8f5b5641966c
fair upstream weight balancer
Igor Sysoev <igor@sysoev.ru>
parents:
1284
diff
changeset
|
435 peer->current_weight = 0; |
| 884 | 436 } |
| 437 | |
| 438 /* ngx_unlock_mutex(rrp->peers->mutex); */ | |
| 439 } | |
| 440 | |
| 441 rrp->current++; | |
| 442 | |
| 443 if (rrp->current >= rrp->peers->number) { | |
| 444 rrp->current = 0; | |
| 445 } | |
| 446 | |
| 447 if (pc->tries) { | |
| 448 pc->tries--; | |
| 449 } | |
| 450 | |
| 451 /* ngx_unlock_mutex(rrp->peers->mutex); */ | |
| 452 } | |
| 453 | |
| 454 | |
| 455 #if (NGX_HTTP_SSL) | |
| 456 | |
|
1284
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
457 ngx_int_t |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
458 ngx_http_upstream_set_round_robin_peer_session(ngx_peer_connection_t *pc, |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
459 void *data) |
| 884 | 460 { |
| 461 ngx_http_upstream_rr_peer_data_t *rrp = data; | |
| 462 | |
|
1284
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
463 ngx_int_t rc; |
| 884 | 464 ngx_ssl_session_t *ssl_session; |
| 465 ngx_http_upstream_rr_peer_t *peer; | |
| 466 | |
|
1284
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
467 peer = &rrp->peers->peer[rrp->current]; |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
468 |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
469 /* TODO: threads only mutex */ |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
470 /* ngx_lock_mutex(rrp->peers->mutex); */ |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
471 |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
472 ssl_session = peer->ssl_session; |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
473 |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
474 rc = ngx_ssl_set_session(pc->connection, ssl_session); |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
475 |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
476 ngx_log_debug2(NGX_LOG_DEBUG_HTTP, pc->log, 0, |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
477 "set session: %p:%d", |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
478 ssl_session, ssl_session ? ssl_session->references : 0); |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
479 |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
480 /* ngx_unlock_mutex(rrp->peers->mutex); */ |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
481 |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
482 return rc; |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
483 } |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
484 |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
485 |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
486 void |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
487 ngx_http_upstream_save_round_robin_peer_session(ngx_peer_connection_t *pc, |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
488 void *data) |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
489 { |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
490 ngx_http_upstream_rr_peer_data_t *rrp = data; |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
491 |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
492 ngx_ssl_session_t *old_ssl_session, *ssl_session; |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
493 ngx_http_upstream_rr_peer_t *peer; |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
494 |
| 884 | 495 ssl_session = ngx_ssl_get_session(pc->connection); |
| 496 | |
| 497 if (ssl_session == NULL) { | |
| 498 return; | |
| 499 } | |
| 500 | |
|
1284
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
501 ngx_log_debug2(NGX_LOG_DEBUG_HTTP, pc->log, 0, |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
502 "save session: %p:%d", ssl_session, ssl_session->references); |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
503 |
| 884 | 504 peer = &rrp->peers->peer[rrp->current]; |
| 505 | |
|
1284
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
506 /* TODO: threads only mutex */ |
| 884 | 507 /* ngx_lock_mutex(rrp->peers->mutex); */ |
|
1284
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
508 |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
509 old_ssl_session = peer->ssl_session; |
| 884 | 510 peer->ssl_session = ssl_session; |
|
1284
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
511 |
| 884 | 512 /* ngx_unlock_mutex(rrp->peers->mutex); */ |
| 513 | |
|
1284
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
514 if (old_ssl_session) { |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
515 |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
516 ngx_log_debug2(NGX_LOG_DEBUG_HTTP, pc->log, 0, |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
517 "old session: %p:%d", |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
518 old_ssl_session, old_ssl_session->references); |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
519 |
| 884 | 520 /* TODO: may block */ |
|
1284
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
521 |
|
be2b895d31e0
fix segfault when session was freed twice
Igor Sysoev <igor@sysoev.ru>
parents:
1206
diff
changeset
|
522 ngx_ssl_free_session(old_ssl_session); |
| 884 | 523 } |
| 524 } | |
| 525 | |
| 526 #endif |