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