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 |