ngx_http_upstream_keepalive: ngx_http_upstream_keepalive

annotate ngx_http_upstream_keepalive_module.c @ 29:c7c4b2d68fdf

Keepalive: set_session and save_session callbacks.

author	Maxim Dounin <mdounin@mdounin.ru>
date	Tue, 30 Aug 2011 14:30:58 +0400
parents	baf671881a8d
children	c5b498c5aa3c

rev	line source
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	1
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	2 /*
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	3 * Copyright (C) Maxim Dounin
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	4 */
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	5
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	6
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	7 #include <ngx_config.h>
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	8 #include <ngx_core.h>
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	9 #include <ngx_http.h>
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	10
5 e7d1b49e0611 Keepalive: style. Maxim Dounin <mdounin@mdounin.ru> parents: 4 diff changeset	11
2 8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	12 typedef struct {
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	13 ngx_uint_t max_cached;
6 bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	14 ngx_uint_t single; /* unsigned:1 */
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	15
2 8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	16 ngx_queue_t cache;
8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	17 ngx_queue_t free;
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	18
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	19 ngx_http_upstream_init_pt original_init_upstream;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	20 ngx_http_upstream_init_peer_pt original_init_peer;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	21
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	22 } ngx_http_upstream_keepalive_srv_conf_t;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	23
5 e7d1b49e0611 Keepalive: style. Maxim Dounin <mdounin@mdounin.ru> parents: 4 diff changeset	24
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	25 typedef struct {
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	26 ngx_http_upstream_keepalive_srv_conf_t *conf;
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	27
11 15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	28 ngx_http_upstream_t *upstream;
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	29
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	30 void *data;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	31
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	32 ngx_event_get_peer_pt original_get_peer;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	33 ngx_event_free_peer_pt original_free_peer;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	34
29 c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	35 #if (NGX_HTTP_SSL)
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	36 ngx_event_set_peer_session_pt original_set_session;
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	37 ngx_event_save_peer_session_pt original_save_session;
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	38 #endif
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	39
11 15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	40 ngx_uint_t failed; /* unsigned:1 */
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	41
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	42 } ngx_http_upstream_keepalive_peer_data_t;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	43
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	44
8 565939797f5c Keepalive: fix connection closing and avoid for loop there. Maxim Dounin <mdounin@mdounin.ru> parents: 6 diff changeset	45 typedef struct {
565939797f5c Keepalive: fix connection closing and avoid for loop there. Maxim Dounin <mdounin@mdounin.ru> parents: 6 diff changeset	46 ngx_http_upstream_keepalive_srv_conf_t *conf;
565939797f5c Keepalive: fix connection closing and avoid for loop there. Maxim Dounin <mdounin@mdounin.ru> parents: 6 diff changeset	47
565939797f5c Keepalive: fix connection closing and avoid for loop there. Maxim Dounin <mdounin@mdounin.ru> parents: 6 diff changeset	48 ngx_queue_t queue;
565939797f5c Keepalive: fix connection closing and avoid for loop there. Maxim Dounin <mdounin@mdounin.ru> parents: 6 diff changeset	49 ngx_connection_t *connection;
565939797f5c Keepalive: fix connection closing and avoid for loop there. Maxim Dounin <mdounin@mdounin.ru> parents: 6 diff changeset	50
565939797f5c Keepalive: fix connection closing and avoid for loop there. Maxim Dounin <mdounin@mdounin.ru> parents: 6 diff changeset	51 socklen_t socklen;
565939797f5c Keepalive: fix connection closing and avoid for loop there. Maxim Dounin <mdounin@mdounin.ru> parents: 6 diff changeset	52 struct sockaddr_storage sockaddr;
565939797f5c Keepalive: fix connection closing and avoid for loop there. Maxim Dounin <mdounin@mdounin.ru> parents: 6 diff changeset	53
565939797f5c Keepalive: fix connection closing and avoid for loop there. Maxim Dounin <mdounin@mdounin.ru> parents: 6 diff changeset	54 } ngx_http_upstream_keepalive_cache_t;
565939797f5c Keepalive: fix connection closing and avoid for loop there. Maxim Dounin <mdounin@mdounin.ru> parents: 6 diff changeset	55
565939797f5c Keepalive: fix connection closing and avoid for loop there. Maxim Dounin <mdounin@mdounin.ru> parents: 6 diff changeset	56
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	57 static ngx_int_t ngx_http_upstream_init_keepalive_peer(ngx_http_request_t *r,
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	58 ngx_http_upstream_srv_conf_t *us);
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	59 static ngx_int_t ngx_http_upstream_get_keepalive_peer(ngx_peer_connection_t *pc,
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	60 void *data);
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	61 static void ngx_http_upstream_free_keepalive_peer(ngx_peer_connection_t *pc,
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	62 void *data, ngx_uint_t state);
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	63
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	64 static void ngx_http_upstream_keepalive_dummy_handler(ngx_event_t *ev);
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	65 static void ngx_http_upstream_keepalive_close_handler(ngx_event_t *ev);
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	66
29 c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	67 #if (NGX_HTTP_SSL)
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	68 static ngx_int_t ngx_http_upstream_keepalive_set_session(
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	69 ngx_peer_connection_t pc, void data);
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	70 static void ngx_http_upstream_keepalive_save_session(ngx_peer_connection_t *pc,
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	71 void *data);
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	72 #endif
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	73
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	74 static void ngx_http_upstream_keepalive_create_conf(ngx_conf_t cf);
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	75 static char ngx_http_upstream_keepalive(ngx_conf_t cf, ngx_command_t *cmd,
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	76 void *conf);
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	77
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	78
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	79 static ngx_command_t ngx_http_upstream_keepalive_commands[] = {
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	80
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	81 { ngx_string("keepalive"),
6 bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	82 NGX_HTTP_UPS_CONF\|NGX_CONF_TAKE12,
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	83 ngx_http_upstream_keepalive,
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	84 0,
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	85 0,
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	86 NULL },
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	87
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	88 ngx_null_command
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	89 };
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	90
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	91
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	92 static ngx_http_module_t ngx_http_upstream_keepalive_module_ctx = {
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	93 NULL, /* preconfiguration */
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	94 NULL, /* postconfiguration */
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	95
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	96 NULL, /* create main configuration */
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	97 NULL, /* init main configuration */
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	98
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	99 ngx_http_upstream_keepalive_create_conf, /* create server configuration */
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	100 NULL, /* merge server configuration */
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	101
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	102 NULL, /* create location configuration */
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	103 NULL /* merge location configuration */
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	104 };
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	105
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	106
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	107 ngx_module_t ngx_http_upstream_keepalive_module = {
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	108 NGX_MODULE_V1,
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	109 &ngx_http_upstream_keepalive_module_ctx, /* module context */
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	110 ngx_http_upstream_keepalive_commands, /* module directives */
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	111 NGX_HTTP_MODULE, /* module type */
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	112 NULL, /* init master */
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	113 NULL, /* init module */
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	114 NULL, /* init process */
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	115 NULL, /* init thread */
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	116 NULL, /* exit thread */
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	117 NULL, /* exit process */
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	118 NULL, /* exit master */
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	119 NGX_MODULE_V1_PADDING
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	120 };
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	121
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	122
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	123 ngx_int_t
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	124 ngx_http_upstream_init_keepalive(ngx_conf_t *cf,
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	125 ngx_http_upstream_srv_conf_t *us)
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	126 {
2 8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	127 ngx_uint_t i;
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	128 ngx_http_upstream_keepalive_srv_conf_t *kcf;
2 8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	129 ngx_http_upstream_keepalive_cache_t *cached;
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	130
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	131 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, cf->log, 0,
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	132 "init keepalive");
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	133
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	134 kcf = ngx_http_conf_upstream_srv_conf(us,
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	135 ngx_http_upstream_keepalive_module);
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	136
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	137 if (kcf->original_init_upstream(cf, us) != NGX_OK) {
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	138 return NGX_ERROR;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	139 }
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	140
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	141 kcf->original_init_peer = us->peer.init;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	142
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	143 us->peer.init = ngx_http_upstream_init_keepalive_peer;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	144
2 8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	145 /* allocate cache items and add to free queue */
8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	146
8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	147 cached = ngx_pcalloc(cf->pool,
8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	148 sizeof(ngx_http_upstream_keepalive_cache_t) * kcf->max_cached);
8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	149 if (cached == NULL) {
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	150 return NGX_ERROR;
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	151 }
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	152
2 8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	153 ngx_queue_init(&kcf->cache);
8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	154 ngx_queue_init(&kcf->free);
8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	155
8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	156 for (i = 0; i < kcf->max_cached; i++) {
8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	157 ngx_queue_insert_head(&kcf->free, &cached[i].queue);
8 565939797f5c Keepalive: fix connection closing and avoid for loop there. Maxim Dounin <mdounin@mdounin.ru> parents: 6 diff changeset	158 cached[i].conf = kcf;
2 8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	159 }
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	160
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	161 return NGX_OK;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	162 }
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	163
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	164
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	165 static ngx_int_t
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	166 ngx_http_upstream_init_keepalive_peer(ngx_http_request_t *r,
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	167 ngx_http_upstream_srv_conf_t *us)
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	168 {
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	169 ngx_http_upstream_keepalive_peer_data_t *kp;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	170 ngx_http_upstream_keepalive_srv_conf_t *kcf;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	171
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	172 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	173 "init keepalive peer");
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	174
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	175 kcf = ngx_http_conf_upstream_srv_conf(us,
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	176 ngx_http_upstream_keepalive_module);
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	177
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	178 kp = ngx_palloc(r->pool, sizeof(ngx_http_upstream_keepalive_peer_data_t));
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	179 if (kp == NULL) {
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	180 return NGX_ERROR;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	181 }
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	182
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	183 if (kcf->original_init_peer(r, us) != NGX_OK) {
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	184 return NGX_ERROR;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	185 }
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	186
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	187 kp->conf = kcf;
11 15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	188 kp->upstream = r->upstream;
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	189 kp->data = r->upstream->peer.data;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	190 kp->original_get_peer = r->upstream->peer.get;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	191 kp->original_free_peer = r->upstream->peer.free;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	192
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	193 r->upstream->peer.data = kp;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	194 r->upstream->peer.get = ngx_http_upstream_get_keepalive_peer;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	195 r->upstream->peer.free = ngx_http_upstream_free_keepalive_peer;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	196
29 c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	197 #if (NGX_HTTP_SSL)
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	198 kp->original_set_session = r->upstream->peer.set_session;
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	199 kp->original_save_session = r->upstream->peer.save_session;
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	200 r->upstream->peer.set_session = ngx_http_upstream_keepalive_set_session;
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	201 r->upstream->peer.save_session = ngx_http_upstream_keepalive_save_session;
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	202 #endif
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	203
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	204 return NGX_OK;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	205 }
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	206
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	207
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	208 static ngx_int_t
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	209 ngx_http_upstream_get_keepalive_peer(ngx_peer_connection_t pc, void data)
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	210 {
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	211 ngx_http_upstream_keepalive_peer_data_t *kp = data;
2 8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	212 ngx_http_upstream_keepalive_cache_t *item;
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	213
6 bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	214 ngx_int_t rc;
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	215 ngx_queue_t q, cache;
5 e7d1b49e0611 Keepalive: style. Maxim Dounin <mdounin@mdounin.ru> parents: 4 diff changeset	216 ngx_connection_t *c;
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	217
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	218 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, pc->log, 0,
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	219 "get keepalive peer");
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	220
11 15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	221 kp->failed = 0;
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	222
6 bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	223 /* single pool of cached connections */
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	224
6 bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	225 if (kp->conf->single && !ngx_queue_empty(&kp->conf->cache)) {
2 8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	226
8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	227 q = ngx_queue_head(&kp->conf->cache);
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	228
2 8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	229 item = ngx_queue_data(q, ngx_http_upstream_keepalive_cache_t, queue);
8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	230 c = item->connection;
8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	231
28 baf671881a8d Keepalive: log connection used. Maxim Dounin <mdounin@mdounin.ru> parents: 26 diff changeset	232 ngx_queue_remove(q);
2 8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	233 ngx_queue_insert_head(&kp->conf->free, q);
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	234
28 baf671881a8d Keepalive: log connection used. Maxim Dounin <mdounin@mdounin.ru> parents: 26 diff changeset	235 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, pc->log, 0,
baf671881a8d Keepalive: log connection used. Maxim Dounin <mdounin@mdounin.ru> parents: 26 diff changeset	236 "get keepalive peer: using connection %p", c);
baf671881a8d Keepalive: log connection used. Maxim Dounin <mdounin@mdounin.ru> parents: 26 diff changeset	237
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	238 c->idle = 0;
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	239 c->log = pc->log;
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	240 c->read->log = pc->log;
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	241 c->write->log = pc->log;
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	242
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	243 pc->connection = c;
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	244 pc->cached = 1;
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	245
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	246 return NGX_DONE;
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	247 }
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	248
6 bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	249 rc = kp->original_get_peer(pc, kp->data);
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	250
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	251 if (kp->conf->single \|\| rc != NGX_OK) {
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	252 return rc;
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	253 }
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	254
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	255 /* search cache for suitable connection */
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	256
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	257 cache = &kp->conf->cache;
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	258
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	259 for (q = ngx_queue_head(cache);
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	260 q != ngx_queue_sentinel(cache);
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	261 q = ngx_queue_next(q))
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	262 {
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	263 item = ngx_queue_data(q, ngx_http_upstream_keepalive_cache_t, queue);
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	264 c = item->connection;
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	265
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	266 if (ngx_memn2cmp((u_char ) &item->sockaddr, (u_char ) pc->sockaddr,
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	267 item->socklen, pc->socklen)
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	268 == 0)
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	269 {
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	270 ngx_queue_remove(q);
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	271 ngx_queue_insert_head(&kp->conf->free, q);
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	272
28 baf671881a8d Keepalive: log connection used. Maxim Dounin <mdounin@mdounin.ru> parents: 26 diff changeset	273 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, pc->log, 0,
baf671881a8d Keepalive: log connection used. Maxim Dounin <mdounin@mdounin.ru> parents: 26 diff changeset	274 "get keepalive peer: using connection %p", c);
baf671881a8d Keepalive: log connection used. Maxim Dounin <mdounin@mdounin.ru> parents: 26 diff changeset	275
6 bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	276 c->idle = 0;
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	277 c->log = pc->log;
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	278 c->read->log = pc->log;
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	279 c->write->log = pc->log;
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	280
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	281 pc->connection = c;
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	282 pc->cached = 1;
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	283
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	284 return NGX_DONE;
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	285 }
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	286 }
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	287
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	288 return NGX_OK;
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	289 }
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	290
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	291
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	292 static void
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	293 ngx_http_upstream_free_keepalive_peer(ngx_peer_connection_t pc, void data,
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	294 ngx_uint_t state)
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	295 {
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	296 ngx_http_upstream_keepalive_peer_data_t *kp = data;
2 8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	297 ngx_http_upstream_keepalive_cache_t *item;
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	298
11 15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	299 ngx_uint_t status;
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	300 ngx_queue_t *q;
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	301 ngx_connection_t *c;
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	302 ngx_http_upstream_t *u;
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	303
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	304 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, pc->log, 0,
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	305 "free keepalive peer");
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	306
11 15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	307 /* remember failed state - peer.free() may be called more than once */
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	308
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	309 if (state & NGX_PEER_FAILED) {
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	310 kp->failed = 1;
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	311 }
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	312
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	313 /*
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	314 * cache valid connections
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	315 *
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	316 * For memcached this means status either 404 or 200. For status 200 we
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	317 * should also check if all response body was read (u->length == 0) and
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	318 * make sure that u->length is valid (we use u->header_sent flag to test
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	319 * this). Memcached is the only supported protocol for now.
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	320 *
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	321 * Some notes on other possibilities (incomplete):
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	322 *
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	323 * fastcgi: u->pipe->upstream_done should be sufficient
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	324 *
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	325 * proxy buffered: u->pipe->upstream_done, 304 replies, replies to head
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	326 * requests (see RFC 2616, 4.4 Message Length)
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	327 *
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	328 * proxy unbuffered: 200 as for memcached (with u->length == 0 and
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	329 * header_sent), 304, replies to head requests
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	330 *
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	331 * subrequest_in_memory: won't work as of now
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	332 *
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	333 * TODO: move this logic to protocol modules (NGX_PEER_KEEPALIVE?)
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	334 */
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	335
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	336 u = kp->upstream;
26 cb15f6f4d820 Keepalive: don't cache timed out connections. Maxim Dounin <mdounin@mdounin.ru> parents: 24 diff changeset	337 c = pc->connection;
11 15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	338 status = u->headers_in.status_n;
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	339
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	340 if (!kp->failed
26 cb15f6f4d820 Keepalive: don't cache timed out connections. Maxim Dounin <mdounin@mdounin.ru> parents: 24 diff changeset	341 && c != NULL
cb15f6f4d820 Keepalive: don't cache timed out connections. Maxim Dounin <mdounin@mdounin.ru> parents: 24 diff changeset	342 && !c->read->timedout
cb15f6f4d820 Keepalive: don't cache timed out connections. Maxim Dounin <mdounin@mdounin.ru> parents: 24 diff changeset	343 && !c->write->timedout
14 28af4b0b32c1 Keepalive: experimental fastcgi support. Maxim Dounin <mdounin@mdounin.ru> parents: 11 diff changeset	344 #if (NGX_UPSTREAM_KEEPALIVE_PATCHED)
28af4b0b32c1 Keepalive: experimental fastcgi support. Maxim Dounin <mdounin@mdounin.ru> parents: 11 diff changeset	345 && u->keepalive)
28af4b0b32c1 Keepalive: experimental fastcgi support. Maxim Dounin <mdounin@mdounin.ru> parents: 11 diff changeset	346 #else
11 15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	347 && (status == NGX_HTTP_NOT_FOUND
15530a464dba Keepalive: don't cache invalid connections. Maxim Dounin <mdounin@mdounin.ru> parents: 10 diff changeset	348 \|\| (status == NGX_HTTP_OK && u->header_sent && u->length == 0)))
14 28af4b0b32c1 Keepalive: experimental fastcgi support. Maxim Dounin <mdounin@mdounin.ru> parents: 11 diff changeset	349 #endif
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	350 {
10 06bd0e50e696 Keepalive: log saving connection. Maxim Dounin <mdounin@mdounin.ru> parents: 8 diff changeset	351 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, pc->log, 0,
06bd0e50e696 Keepalive: log saving connection. Maxim Dounin <mdounin@mdounin.ru> parents: 8 diff changeset	352 "free keepalive peer: saving connection %p", c);
06bd0e50e696 Keepalive: log saving connection. Maxim Dounin <mdounin@mdounin.ru> parents: 8 diff changeset	353
3 42e64900483b Keepalive: always save connections to cache. Maxim Dounin <mdounin@mdounin.ru> parents: 2 diff changeset	354 if (ngx_queue_empty(&kp->conf->free)) {
42e64900483b Keepalive: always save connections to cache. Maxim Dounin <mdounin@mdounin.ru> parents: 2 diff changeset	355
42e64900483b Keepalive: always save connections to cache. Maxim Dounin <mdounin@mdounin.ru> parents: 2 diff changeset	356 q = ngx_queue_last(&kp->conf->cache);
42e64900483b Keepalive: always save connections to cache. Maxim Dounin <mdounin@mdounin.ru> parents: 2 diff changeset	357 ngx_queue_remove(q);
42e64900483b Keepalive: always save connections to cache. Maxim Dounin <mdounin@mdounin.ru> parents: 2 diff changeset	358
42e64900483b Keepalive: always save connections to cache. Maxim Dounin <mdounin@mdounin.ru> parents: 2 diff changeset	359 item = ngx_queue_data(q, ngx_http_upstream_keepalive_cache_t,
42e64900483b Keepalive: always save connections to cache. Maxim Dounin <mdounin@mdounin.ru> parents: 2 diff changeset	360 queue);
42e64900483b Keepalive: always save connections to cache. Maxim Dounin <mdounin@mdounin.ru> parents: 2 diff changeset	361
42e64900483b Keepalive: always save connections to cache. Maxim Dounin <mdounin@mdounin.ru> parents: 2 diff changeset	362 ngx_close_connection(item->connection);
2 8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	363
3 42e64900483b Keepalive: always save connections to cache. Maxim Dounin <mdounin@mdounin.ru> parents: 2 diff changeset	364 } else {
42e64900483b Keepalive: always save connections to cache. Maxim Dounin <mdounin@mdounin.ru> parents: 2 diff changeset	365 q = ngx_queue_head(&kp->conf->free);
42e64900483b Keepalive: always save connections to cache. Maxim Dounin <mdounin@mdounin.ru> parents: 2 diff changeset	366 ngx_queue_remove(q);
42e64900483b Keepalive: always save connections to cache. Maxim Dounin <mdounin@mdounin.ru> parents: 2 diff changeset	367
42e64900483b Keepalive: always save connections to cache. Maxim Dounin <mdounin@mdounin.ru> parents: 2 diff changeset	368 item = ngx_queue_data(q, ngx_http_upstream_keepalive_cache_t,
42e64900483b Keepalive: always save connections to cache. Maxim Dounin <mdounin@mdounin.ru> parents: 2 diff changeset	369 queue);
42e64900483b Keepalive: always save connections to cache. Maxim Dounin <mdounin@mdounin.ru> parents: 2 diff changeset	370 }
42e64900483b Keepalive: always save connections to cache. Maxim Dounin <mdounin@mdounin.ru> parents: 2 diff changeset	371
2 8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	372 item->connection = c;
8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	373 ngx_queue_insert_head(&kp->conf->cache, q);
8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	374
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	375 pc->connection = NULL;
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	376
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	377 if (c->read->timer_set) {
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	378 ngx_del_timer(c->read);
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	379 }
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	380 if (c->write->timer_set) {
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	381 ngx_del_timer(c->write);
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	382 }
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	383
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	384 c->write->handler = ngx_http_upstream_keepalive_dummy_handler;
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	385 c->read->handler = ngx_http_upstream_keepalive_close_handler;
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	386
8 565939797f5c Keepalive: fix connection closing and avoid for loop there. Maxim Dounin <mdounin@mdounin.ru> parents: 6 diff changeset	387 c->data = item;
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	388 c->idle = 1;
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	389 c->log = ngx_cycle->log;
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	390 c->read->log = ngx_cycle->log;
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	391 c->write->log = ngx_cycle->log;
6 bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	392
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	393 item->socklen = pc->socklen;
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	394 ngx_memcpy(&item->sockaddr, pc->sockaddr, pc->socklen);
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	395 }
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	396
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	397 return kp->original_free_peer(pc, kp->data, state);
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	398 }
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	399
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	400
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	401 static void
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	402 ngx_http_upstream_keepalive_dummy_handler(ngx_event_t *ev)
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	403 {
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	404 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, ev->log, 0,
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	405 "keepalive dummy handler");
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	406 }
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	407
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	408
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	409 static void
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	410 ngx_http_upstream_keepalive_close_handler(ngx_event_t *ev)
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	411 {
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	412 ngx_http_upstream_keepalive_srv_conf_t *conf;
2 8545bbda9e4b Keepalive: reimplement using queues. Maxim Dounin <mdounin@mdounin.ru> parents: 1 diff changeset	413 ngx_http_upstream_keepalive_cache_t *item;
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	414
21 9a4ee6fe1c6d Keepalive: avoid closing upstream connection on stale events. Maxim Dounin <mdounin@mdounin.ru> parents: 18 diff changeset	415 int n;
9a4ee6fe1c6d Keepalive: avoid closing upstream connection on stale events. Maxim Dounin <mdounin@mdounin.ru> parents: 18 diff changeset	416 char buf[1];
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	417 ngx_connection_t *c;
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	418
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	419 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, ev->log, 0,
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	420 "keepalive close handler");
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	421
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	422 c = ev->data;
21 9a4ee6fe1c6d Keepalive: avoid closing upstream connection on stale events. Maxim Dounin <mdounin@mdounin.ru> parents: 18 diff changeset	423
24 2ee28064a04a Keepalive: correctly close connections on gracefull shutdown. Maxim Dounin <mdounin@mdounin.ru> parents: 21 diff changeset	424 if (c->close) {
2ee28064a04a Keepalive: correctly close connections on gracefull shutdown. Maxim Dounin <mdounin@mdounin.ru> parents: 21 diff changeset	425 goto close;
2ee28064a04a Keepalive: correctly close connections on gracefull shutdown. Maxim Dounin <mdounin@mdounin.ru> parents: 21 diff changeset	426 }
2ee28064a04a Keepalive: correctly close connections on gracefull shutdown. Maxim Dounin <mdounin@mdounin.ru> parents: 21 diff changeset	427
21 9a4ee6fe1c6d Keepalive: avoid closing upstream connection on stale events. Maxim Dounin <mdounin@mdounin.ru> parents: 18 diff changeset	428 n = recv(c->fd, buf, 1, MSG_PEEK);
9a4ee6fe1c6d Keepalive: avoid closing upstream connection on stale events. Maxim Dounin <mdounin@mdounin.ru> parents: 18 diff changeset	429
9a4ee6fe1c6d Keepalive: avoid closing upstream connection on stale events. Maxim Dounin <mdounin@mdounin.ru> parents: 18 diff changeset	430 if (n == -1 && ngx_socket_errno == NGX_EAGAIN) {
9a4ee6fe1c6d Keepalive: avoid closing upstream connection on stale events. Maxim Dounin <mdounin@mdounin.ru> parents: 18 diff changeset	431 /* stale event */
9a4ee6fe1c6d Keepalive: avoid closing upstream connection on stale events. Maxim Dounin <mdounin@mdounin.ru> parents: 18 diff changeset	432
9a4ee6fe1c6d Keepalive: avoid closing upstream connection on stale events. Maxim Dounin <mdounin@mdounin.ru> parents: 18 diff changeset	433 if (ngx_handle_read_event(c->read, 0) != NGX_OK) {
9a4ee6fe1c6d Keepalive: avoid closing upstream connection on stale events. Maxim Dounin <mdounin@mdounin.ru> parents: 18 diff changeset	434 goto close;
9a4ee6fe1c6d Keepalive: avoid closing upstream connection on stale events. Maxim Dounin <mdounin@mdounin.ru> parents: 18 diff changeset	435 }
9a4ee6fe1c6d Keepalive: avoid closing upstream connection on stale events. Maxim Dounin <mdounin@mdounin.ru> parents: 18 diff changeset	436
9a4ee6fe1c6d Keepalive: avoid closing upstream connection on stale events. Maxim Dounin <mdounin@mdounin.ru> parents: 18 diff changeset	437 return;
9a4ee6fe1c6d Keepalive: avoid closing upstream connection on stale events. Maxim Dounin <mdounin@mdounin.ru> parents: 18 diff changeset	438 }
9a4ee6fe1c6d Keepalive: avoid closing upstream connection on stale events. Maxim Dounin <mdounin@mdounin.ru> parents: 18 diff changeset	439
9a4ee6fe1c6d Keepalive: avoid closing upstream connection on stale events. Maxim Dounin <mdounin@mdounin.ru> parents: 18 diff changeset	440 close:
9a4ee6fe1c6d Keepalive: avoid closing upstream connection on stale events. Maxim Dounin <mdounin@mdounin.ru> parents: 18 diff changeset	441
8 565939797f5c Keepalive: fix connection closing and avoid for loop there. Maxim Dounin <mdounin@mdounin.ru> parents: 6 diff changeset	442 item = c->data;
565939797f5c Keepalive: fix connection closing and avoid for loop there. Maxim Dounin <mdounin@mdounin.ru> parents: 6 diff changeset	443 conf = item->conf;
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	444
8 565939797f5c Keepalive: fix connection closing and avoid for loop there. Maxim Dounin <mdounin@mdounin.ru> parents: 6 diff changeset	445 ngx_queue_remove(&item->queue);
565939797f5c Keepalive: fix connection closing and avoid for loop there. Maxim Dounin <mdounin@mdounin.ru> parents: 6 diff changeset	446 ngx_close_connection(item->connection);
565939797f5c Keepalive: fix connection closing and avoid for loop there. Maxim Dounin <mdounin@mdounin.ru> parents: 6 diff changeset	447 ngx_queue_insert_head(&conf->free, &item->queue);
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	448 }
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	449
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	450
29 c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	451 #if (NGX_HTTP_SSL)
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	452
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	453 static ngx_int_t
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	454 ngx_http_upstream_keepalive_set_session(ngx_peer_connection_t pc, void data)
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	455 {
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	456 ngx_http_upstream_keepalive_peer_data_t *kp = data;
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	457
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	458 return kp->original_set_session(pc, kp->data);
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	459 }
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	460
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	461
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	462 static void
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	463 ngx_http_upstream_keepalive_save_session(ngx_peer_connection_t pc, void data)
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	464 {
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	465 ngx_http_upstream_keepalive_peer_data_t *kp = data;
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	466
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	467 kp->original_save_session(pc, kp->data);
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	468 return;
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	469 }
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	470
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	471 #endif
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	472
c7c4b2d68fdf Keepalive: set_session and save_session callbacks. Maxim Dounin <mdounin@mdounin.ru> parents: 28 diff changeset	473
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	474 static void *
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	475 ngx_http_upstream_keepalive_create_conf(ngx_conf_t *cf)
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	476 {
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	477 ngx_http_upstream_keepalive_srv_conf_t *conf;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	478
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	479 conf = ngx_pcalloc(cf->pool,
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	480 sizeof(ngx_http_upstream_keepalive_srv_conf_t));
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	481 if (conf == NULL) {
18 2054159546d0 Keepalive: fix create_conf to return NULL on errors. Maxim Dounin <mdounin@mdounin.ru> parents: 14 diff changeset	482 return NULL;
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	483 }
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	484
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	485 /*
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	486 * set by ngx_pcalloc():
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	487 *
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	488 * conf->original_init_upstream = NULL;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	489 * conf->original_init_peer = NULL;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	490 */
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	491
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	492 conf->max_cached = 1;
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	493
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	494 return conf;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	495 }
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	496
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	497
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	498 static char *
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	499 ngx_http_upstream_keepalive(ngx_conf_t cf, ngx_command_t cmd, void *conf)
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	500 {
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	501 ngx_http_upstream_srv_conf_t *uscf;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	502 ngx_http_upstream_keepalive_srv_conf_t *kcf;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	503
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	504 ngx_int_t n;
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	505 ngx_str_t *value;
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	506 ngx_uint_t i;
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	507
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	508 uscf = ngx_http_conf_get_module_srv_conf(cf, ngx_http_upstream_module);
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	509
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	510 kcf = ngx_http_conf_upstream_srv_conf(uscf,
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	511 ngx_http_upstream_keepalive_module);
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	512
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	513 kcf->original_init_upstream = uscf->peer.init_upstream
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	514 ? uscf->peer.init_upstream
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	515 : ngx_http_upstream_init_round_robin;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	516
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	517 uscf->peer.init_upstream = ngx_http_upstream_init_keepalive;
725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	518
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	519 /* read options */
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	520
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	521 value = cf->args->elts;
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	522
6 bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	523 n = ngx_atoi(value[1].data, value[1].len);
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	524
6 bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	525 if (n == NGX_ERROR \|\| n == 0) {
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	526 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	527 "invalid value \"%V\" in \"%V\" directive",
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	528 &value[1], &cmd->name);
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	529 return NGX_CONF_ERROR;
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	530 }
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	531
6 bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	532 kcf->max_cached = n;
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	533
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	534 for (i = 2; i < cf->args->nelts; i++) {
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	535
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	536 if (ngx_strcmp(value[i].data, "single") == 0) {
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	537
bef88ba0b378 Keepalive: distinguish between upstream servers by default. Maxim Dounin <mdounin@mdounin.ru> parents: 5 diff changeset	538 kcf->single = 1;
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	539
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	540 continue;
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	541 }
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	542
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	543 goto invalid;
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	544 }
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	545
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	546 return NGX_CONF_OK;
1 ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	547
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	548 invalid:
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	549
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	550 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	551 "invalid parameter \"%V\"", &value[i]);
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	552
ca955a7f651b Keepalive: common cache for all backends. Maxim Dounin <mdounin@mdounin.ru> parents: 0 diff changeset	553 return NGX_CONF_ERROR;
0 725ee11164f3 Keepalive upstream balancer: transparent skeleton. Maxim Dounin <mdounin@mdounin.ru> parents: diff changeset	554 }

Mercurial > hg > ngx_http_upstream_keepalive

annotate ngx_http_upstream_keepalive_module.c @ 29:c7c4b2d68fdf