nginx: src/http/modules/ngx_http_upstream_hash

annotate src/http/modules/ngx_http_upstream_hash_module.c @ 6099:6ff0ebd6fbf4

Upstream: track the number of active connections to upstreams. This also simplifies the implementation of the least_conn module.

author	Ruslan Ermilov <ru@nginx.com>
date	Fri, 10 Apr 2015 13:16:23 +0300
parents	435ee290c2e1
children	c44459611d91

rev	line source
5717 efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	1
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	2 /*
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	3 * Copyright (C) Roman Arutyunyan
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	4 * Copyright (C) Nginx, Inc.
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	5 */
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	6
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	7
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	8 #include <ngx_config.h>
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	9 #include <ngx_core.h>
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	10 #include <ngx_http.h>
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	11
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	12
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	13 typedef struct {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	14 uint32_t hash;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	15 ngx_str_t *server;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	16 } ngx_http_upstream_chash_point_t;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	17
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	18
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	19 typedef struct {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	20 ngx_uint_t number;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	21 ngx_http_upstream_chash_point_t point[1];
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	22 } ngx_http_upstream_chash_points_t;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	23
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	24
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	25 typedef struct {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	26 ngx_http_complex_value_t key;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	27 ngx_http_upstream_chash_points_t *points;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	28 } ngx_http_upstream_hash_srv_conf_t;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	29
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	30
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	31 typedef struct {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	32 /* the round robin data must be first */
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	33 ngx_http_upstream_rr_peer_data_t rrp;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	34 ngx_http_upstream_hash_srv_conf_t *conf;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	35 ngx_str_t key;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	36 ngx_uint_t tries;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	37 ngx_uint_t rehash;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	38 uint32_t hash;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	39 ngx_event_get_peer_pt get_rr_peer;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	40 } ngx_http_upstream_hash_peer_data_t;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	41
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	42
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	43 static ngx_int_t ngx_http_upstream_init_hash(ngx_conf_t *cf,
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	44 ngx_http_upstream_srv_conf_t *us);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	45 static ngx_int_t ngx_http_upstream_init_hash_peer(ngx_http_request_t *r,
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	46 ngx_http_upstream_srv_conf_t *us);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	47 static ngx_int_t ngx_http_upstream_get_hash_peer(ngx_peer_connection_t *pc,
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	48 void *data);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	49
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	50 static ngx_int_t ngx_http_upstream_init_chash(ngx_conf_t *cf,
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	51 ngx_http_upstream_srv_conf_t *us);
5991 435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	52 static int ngx_libc_cdecl
435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	53 ngx_http_upstream_chash_cmp_points(const void one, const void two);
5717 efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	54 static ngx_uint_t ngx_http_upstream_find_chash_point(
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	55 ngx_http_upstream_chash_points_t *points, uint32_t hash);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	56 static ngx_int_t ngx_http_upstream_init_chash_peer(ngx_http_request_t *r,
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	57 ngx_http_upstream_srv_conf_t *us);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	58 static ngx_int_t ngx_http_upstream_get_chash_peer(ngx_peer_connection_t *pc,
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	59 void *data);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	60
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	61 static void ngx_http_upstream_hash_create_conf(ngx_conf_t cf);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	62 static char ngx_http_upstream_hash(ngx_conf_t cf, ngx_command_t *cmd,
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	63 void *conf);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	64
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	65
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	66 static ngx_command_t ngx_http_upstream_hash_commands[] = {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	67
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	68 { ngx_string("hash"),
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	69 NGX_HTTP_UPS_CONF\|NGX_CONF_TAKE12,
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	70 ngx_http_upstream_hash,
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	71 NGX_HTTP_SRV_CONF_OFFSET,
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	72 0,
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	73 NULL },
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	74
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	75 ngx_null_command
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	76 };
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	77
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	78
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	79 static ngx_http_module_t ngx_http_upstream_hash_module_ctx = {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	80 NULL, /* preconfiguration */
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	81 NULL, /* postconfiguration */
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	82
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	83 NULL, /* create main configuration */
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	84 NULL, /* init main configuration */
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	85
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	86 ngx_http_upstream_hash_create_conf, /* create server configuration */
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	87 NULL, /* merge server configuration */
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	88
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	89 NULL, /* create location configuration */
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	90 NULL /* merge location configuration */
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	91 };
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	92
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	93
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	94 ngx_module_t ngx_http_upstream_hash_module = {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	95 NGX_MODULE_V1,
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	96 &ngx_http_upstream_hash_module_ctx, /* module context */
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	97 ngx_http_upstream_hash_commands, /* module directives */
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	98 NGX_HTTP_MODULE, /* module type */
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	99 NULL, /* init master */
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	100 NULL, /* init module */
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	101 NULL, /* init process */
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	102 NULL, /* init thread */
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	103 NULL, /* exit thread */
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	104 NULL, /* exit process */
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	105 NULL, /* exit master */
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	106 NGX_MODULE_V1_PADDING
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	107 };
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	108
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	109
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	110 static ngx_int_t
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	111 ngx_http_upstream_init_hash(ngx_conf_t cf, ngx_http_upstream_srv_conf_t us)
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	112 {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	113 if (ngx_http_upstream_init_round_robin(cf, us) != NGX_OK) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	114 return NGX_ERROR;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	115 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	116
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	117 us->peer.init = ngx_http_upstream_init_hash_peer;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	118
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	119 return NGX_OK;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	120 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	121
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	122
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	123 static ngx_int_t
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	124 ngx_http_upstream_init_hash_peer(ngx_http_request_t *r,
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	125 ngx_http_upstream_srv_conf_t *us)
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	126 {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	127 ngx_http_upstream_hash_srv_conf_t *hcf;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	128 ngx_http_upstream_hash_peer_data_t *hp;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	129
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	130 hp = ngx_palloc(r->pool, sizeof(ngx_http_upstream_hash_peer_data_t));
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	131 if (hp == NULL) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	132 return NGX_ERROR;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	133 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	134
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	135 r->upstream->peer.data = &hp->rrp;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	136
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	137 if (ngx_http_upstream_init_round_robin_peer(r, us) != NGX_OK) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	138 return NGX_ERROR;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	139 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	140
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	141 r->upstream->peer.get = ngx_http_upstream_get_hash_peer;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	142
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	143 hcf = ngx_http_conf_upstream_srv_conf(us, ngx_http_upstream_hash_module);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	144
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	145 if (ngx_http_complex_value(r, &hcf->key, &hp->key) != NGX_OK) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	146 return NGX_ERROR;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	147 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	148
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	149 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	150 "upstream hash key:\"%V\"", &hp->key);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	151
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	152 hp->conf = hcf;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	153 hp->tries = 0;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	154 hp->rehash = 0;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	155 hp->hash = 0;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	156 hp->get_rr_peer = ngx_http_upstream_get_round_robin_peer;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	157
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	158 return NGX_OK;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	159 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	160
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	161
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	162 static ngx_int_t
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	163 ngx_http_upstream_get_hash_peer(ngx_peer_connection_t pc, void data)
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	164 {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	165 ngx_http_upstream_hash_peer_data_t *hp = data;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	166
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	167 time_t now;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	168 u_char buf[NGX_INT_T_LEN];
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	169 size_t size;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	170 uint32_t hash;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	171 ngx_int_t w;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	172 uintptr_t m;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	173 ngx_uint_t i, n, p;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	174 ngx_http_upstream_rr_peer_t *peer;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	175
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	176 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, pc->log, 0,
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	177 "get hash peer, try: %ui", pc->tries);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	178
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	179 if (hp->tries > 20 \|\| hp->rrp.peers->single) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	180 return hp->get_rr_peer(pc, &hp->rrp);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	181 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	182
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	183 now = ngx_time();
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	184
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	185 pc->cached = 0;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	186 pc->connection = NULL;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	187
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	188 for ( ;; ) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	189
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	190 /*
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	191 * Hash expression is compatible with Cache::Memcached:
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	192 * ((crc32([REHASH] KEY) >> 16) & 0x7fff) + PREV_HASH
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	193 * with REHASH omitted at the first iteration.
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	194 */
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	195
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	196 ngx_crc32_init(hash);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	197
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	198 if (hp->rehash > 0) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	199 size = ngx_sprintf(buf, "%ui", hp->rehash) - buf;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	200 ngx_crc32_update(&hash, buf, size);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	201 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	202
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	203 ngx_crc32_update(&hash, hp->key.data, hp->key.len);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	204 ngx_crc32_final(hash);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	205
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	206 hash = (hash >> 16) & 0x7fff;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	207
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	208 hp->hash += hash;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	209 hp->rehash++;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	210
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	211 if (!hp->rrp.peers->weighted) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	212 p = hp->hash % hp->rrp.peers->number;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	213
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	214 } else {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	215 w = hp->hash % hp->rrp.peers->total_weight;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	216
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	217 for (i = 0; i < hp->rrp.peers->number; i++) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	218 w -= hp->rrp.peers->peer[i].weight;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	219 if (w < 0) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	220 break;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	221 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	222 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	223
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	224 p = i;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	225 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	226
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	227 n = p / (8 * sizeof(uintptr_t));
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	228 m = (uintptr_t) 1 << p % (8 * sizeof(uintptr_t));
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	229
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	230 if (hp->rrp.tried[n] & m) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	231 goto next;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	232 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	233
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	234 ngx_log_debug2(NGX_LOG_DEBUG_HTTP, pc->log, 0,
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	235 "get hash peer, value:%uD, peer:%ui", hp->hash, p);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	236
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	237 peer = &hp->rrp.peers->peer[p];
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	238
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	239 if (peer->down) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	240 goto next;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	241 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	242
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	243 if (peer->max_fails
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	244 && peer->fails >= peer->max_fails
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	245 && now - peer->checked <= peer->fail_timeout)
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	246 {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	247 goto next;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	248 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	249
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	250 break;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	251
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	252 next:
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	253
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	254 if (++hp->tries > 20) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	255 return hp->get_rr_peer(pc, &hp->rrp);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	256 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	257 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	258
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	259 hp->rrp.current = p;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	260
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	261 pc->sockaddr = peer->sockaddr;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	262 pc->socklen = peer->socklen;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	263 pc->name = &peer->name;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	264
6099 6ff0ebd6fbf4 Upstream: track the number of active connections to upstreams. Ruslan Ermilov <ru@nginx.com> parents: 5991 diff changeset	265 peer->conns++;
6ff0ebd6fbf4 Upstream: track the number of active connections to upstreams. Ruslan Ermilov <ru@nginx.com> parents: 5991 diff changeset	266
5717 efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	267 if (now - peer->checked > peer->fail_timeout) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	268 peer->checked = now;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	269 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	270
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	271 hp->rrp.tried[n] \|= m;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	272
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	273 return NGX_OK;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	274 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	275
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	276
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	277 static ngx_int_t
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	278 ngx_http_upstream_init_chash(ngx_conf_t cf, ngx_http_upstream_srv_conf_t us)
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	279 {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	280 u_char host, port, c;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	281 size_t host_len, port_len, size;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	282 uint32_t hash, base_hash, prev_hash;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	283 ngx_str_t *server;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	284 ngx_uint_t npoints, i, j;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	285 ngx_http_upstream_rr_peer_t *peer;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	286 ngx_http_upstream_rr_peers_t *peers;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	287 ngx_http_upstream_chash_points_t *points;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	288 ngx_http_upstream_hash_srv_conf_t *hcf;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	289
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	290 if (ngx_http_upstream_init_round_robin(cf, us) != NGX_OK) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	291 return NGX_ERROR;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	292 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	293
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	294 us->peer.init = ngx_http_upstream_init_chash_peer;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	295
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	296 peers = us->peer.data;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	297 npoints = peers->total_weight * 160;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	298
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	299 size = sizeof(ngx_http_upstream_chash_points_t)
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	300 + sizeof(ngx_http_upstream_chash_point_t) * (npoints - 1);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	301
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	302 points = ngx_palloc(cf->pool, size);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	303 if (points == NULL) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	304 return NGX_ERROR;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	305 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	306
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	307 points->number = 0;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	308
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	309 for (i = 0; i < peers->number; i++) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	310 peer = &peers->peer[i];
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	311 server = &peer->server;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	312
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	313 /*
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	314 * Hash expression is compatible with Cache::Memcached::Fast:
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	315 * crc32(HOST \0 PORT PREV_HASH).
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	316 */
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	317
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	318 if (server->len >= 5
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	319 && ngx_strncasecmp(server->data, (u_char *) "unix:", 5) == 0)
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	320 {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	321 host = server->data + 5;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	322 host_len = server->len - 5;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	323 port = NULL;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	324 port_len = 0;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	325 goto done;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	326 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	327
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	328 for (j = 0; j < server->len; j++) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	329 c = server->data[server->len - j - 1];
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	330
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	331 if (c == ':') {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	332 host = server->data;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	333 host_len = server->len - j - 1;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	334 port = server->data + server->len - j;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	335 port_len = j;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	336 goto done;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	337 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	338
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	339 if (c < '0' \|\| c > '9') {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	340 break;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	341 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	342 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	343
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	344 host = server->data;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	345 host_len = server->len;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	346 port = NULL;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	347 port_len = 0;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	348
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	349 done:
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	350
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	351 ngx_crc32_init(base_hash);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	352 ngx_crc32_update(&base_hash, host, host_len);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	353 ngx_crc32_update(&base_hash, (u_char *) "", 1);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	354 ngx_crc32_update(&base_hash, port, port_len);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	355
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	356 prev_hash = 0;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	357 npoints = peer->weight * 160;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	358
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	359 for (j = 0; j < npoints; j++) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	360 hash = base_hash;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	361
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	362 ngx_crc32_update(&hash, (u_char *) &prev_hash, sizeof(uint32_t));
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	363 ngx_crc32_final(hash);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	364
5991 435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	365 points->point[points->number].hash = hash;
435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	366 points->point[points->number].server = server;
435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	367 points->number++;
5717 efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	368
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	369 prev_hash = hash;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	370 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	371 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	372
5991 435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	373 ngx_qsort(points->point,
435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	374 points->number,
435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	375 sizeof(ngx_http_upstream_chash_point_t),
435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	376 ngx_http_upstream_chash_cmp_points);
435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	377
435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	378 for (i = 0, j = 1; j < points->number; j++) {
435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	379 if (points->point[i].hash != points->point[j].hash) {
435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	380 points->point[++i] = points->point[j];
435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	381 }
435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	382 }
435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	383
435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	384 points->number = i + 1;
435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	385
5717 efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	386 hcf = ngx_http_conf_upstream_srv_conf(us, ngx_http_upstream_hash_module);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	387 hcf->points = points;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	388
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	389 return NGX_OK;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	390 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	391
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	392
5991 435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	393 static int ngx_libc_cdecl
435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	394 ngx_http_upstream_chash_cmp_points(const void one, const void two)
5717 efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	395 {
5991 435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	396 ngx_http_upstream_chash_point_t *first =
435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	397 (ngx_http_upstream_chash_point_t *) one;
435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	398 ngx_http_upstream_chash_point_t *second =
435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	399 (ngx_http_upstream_chash_point_t *) two;
5717 efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	400
5991 435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	401 if (first->hash < second->hash) {
435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	402 return -1;
5717 efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	403
5991 435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	404 } else if (first->hash > second->hash) {
435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	405 return 1;
5717 efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	406
5991 435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	407 } else {
435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	408 return 0;
435ee290c2e1 Upstream hash: speedup consistent hash init. Roman Arutyunyan <arut@nginx.com> parents: 5717 diff changeset	409 }
5717 efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	410 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	411
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	412
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	413 static ngx_uint_t
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	414 ngx_http_upstream_find_chash_point(ngx_http_upstream_chash_points_t *points,
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	415 uint32_t hash)
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	416 {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	417 ngx_uint_t i, j, k;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	418 ngx_http_upstream_chash_point_t *point;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	419
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	420 /* find first point >= hash */
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	421
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	422 point = &points->point[0];
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	423
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	424 i = 0;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	425 j = points->number;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	426
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	427 while (i < j) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	428 k = (i + j) / 2;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	429
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	430 if (hash > point[k].hash) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	431 i = k + 1;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	432
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	433 } else if (hash < point[k].hash) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	434 j = k;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	435
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	436 } else {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	437 return k;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	438 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	439 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	440
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	441 return i;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	442 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	443
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	444
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	445 static ngx_int_t
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	446 ngx_http_upstream_init_chash_peer(ngx_http_request_t *r,
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	447 ngx_http_upstream_srv_conf_t *us)
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	448 {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	449 uint32_t hash;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	450 ngx_http_upstream_hash_srv_conf_t *hcf;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	451 ngx_http_upstream_hash_peer_data_t *hp;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	452
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	453 if (ngx_http_upstream_init_hash_peer(r, us) != NGX_OK) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	454 return NGX_ERROR;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	455 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	456
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	457 r->upstream->peer.get = ngx_http_upstream_get_chash_peer;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	458
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	459 hp = r->upstream->peer.data;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	460 hcf = ngx_http_conf_upstream_srv_conf(us, ngx_http_upstream_hash_module);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	461
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	462 hash = ngx_crc32_long(hp->key.data, hp->key.len);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	463 hp->hash = ngx_http_upstream_find_chash_point(hcf->points, hash);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	464
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	465 return NGX_OK;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	466 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	467
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	468
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	469 static ngx_int_t
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	470 ngx_http_upstream_get_chash_peer(ngx_peer_connection_t pc, void data)
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	471 {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	472 ngx_http_upstream_hash_peer_data_t *hp = data;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	473
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	474 time_t now;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	475 intptr_t m;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	476 ngx_str_t *server;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	477 ngx_int_t total;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	478 ngx_uint_t i, n;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	479 ngx_http_upstream_rr_peer_t peer, best;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	480 ngx_http_upstream_chash_point_t *point;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	481 ngx_http_upstream_chash_points_t *points;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	482 ngx_http_upstream_hash_srv_conf_t *hcf;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	483
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	484 ngx_log_debug1(NGX_LOG_DEBUG_HTTP, pc->log, 0,
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	485 "get consistent hash peer, try: %ui", pc->tries);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	486
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	487 pc->cached = 0;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	488 pc->connection = NULL;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	489
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	490 now = ngx_time();
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	491 hcf = hp->conf;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	492
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	493 points = hcf->points;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	494 point = &points->point[0];
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	495
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	496 for ( ;; ) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	497 server = point[hp->hash % points->number].server;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	498
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	499 ngx_log_debug2(NGX_LOG_DEBUG_HTTP, pc->log, 0,
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	500 "consistent hash peer:%uD, server:\"%V\"",
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	501 hp->hash, server);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	502
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	503 best = NULL;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	504 total = 0;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	505
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	506 for (i = 0; i < hp->rrp.peers->number; i++) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	507
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	508 n = i / (8 * sizeof(uintptr_t));
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	509 m = (uintptr_t) 1 << i % (8 * sizeof(uintptr_t));
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	510
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	511 if (hp->rrp.tried[n] & m) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	512 continue;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	513 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	514
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	515 peer = &hp->rrp.peers->peer[i];
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	516
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	517 if (peer->down) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	518 continue;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	519 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	520
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	521 if (peer->server.len != server->len
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	522 \|\| ngx_strncmp(peer->server.data, server->data, server->len)
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	523 != 0)
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	524 {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	525 continue;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	526 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	527
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	528 if (peer->max_fails
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	529 && peer->fails >= peer->max_fails
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	530 && now - peer->checked <= peer->fail_timeout)
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	531 {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	532 continue;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	533 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	534
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	535 peer->current_weight += peer->effective_weight;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	536 total += peer->effective_weight;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	537
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	538 if (peer->effective_weight < peer->weight) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	539 peer->effective_weight++;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	540 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	541
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	542 if (best == NULL \|\| peer->current_weight > best->current_weight) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	543 best = peer;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	544 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	545 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	546
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	547 if (best) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	548 best->current_weight -= total;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	549
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	550 i = best - &hp->rrp.peers->peer[0];
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	551
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	552 hp->rrp.current = i;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	553
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	554 n = i / (8 * sizeof(uintptr_t));
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	555 m = (uintptr_t) 1 << i % (8 * sizeof(uintptr_t));
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	556
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	557 hp->rrp.tried[n] \|= m;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	558
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	559 if (now - best->checked > best->fail_timeout) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	560 best->checked = now;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	561 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	562
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	563 pc->sockaddr = best->sockaddr;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	564 pc->socklen = best->socklen;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	565 pc->name = &best->name;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	566
6099 6ff0ebd6fbf4 Upstream: track the number of active connections to upstreams. Ruslan Ermilov <ru@nginx.com> parents: 5991 diff changeset	567 best->conns++;
6ff0ebd6fbf4 Upstream: track the number of active connections to upstreams. Ruslan Ermilov <ru@nginx.com> parents: 5991 diff changeset	568
5717 efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	569 return NGX_OK;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	570 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	571
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	572 hp->hash++;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	573 hp->tries++;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	574
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	575 if (hp->tries >= points->number) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	576 return NGX_BUSY;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	577 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	578 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	579 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	580
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	581
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	582 static void *
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	583 ngx_http_upstream_hash_create_conf(ngx_conf_t *cf)
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	584 {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	585 ngx_http_upstream_hash_srv_conf_t *conf;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	586
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	587 conf = ngx_palloc(cf->pool, sizeof(ngx_http_upstream_hash_srv_conf_t));
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	588 if (conf == NULL) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	589 return NULL;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	590 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	591
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	592 conf->points = NULL;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	593
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	594 return conf;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	595 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	596
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	597
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	598 static char *
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	599 ngx_http_upstream_hash(ngx_conf_t cf, ngx_command_t cmd, void *conf)
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	600 {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	601 ngx_http_upstream_hash_srv_conf_t *hcf = conf;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	602
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	603 ngx_str_t *value;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	604 ngx_http_upstream_srv_conf_t *uscf;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	605 ngx_http_compile_complex_value_t ccv;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	606
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	607 value = cf->args->elts;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	608
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	609 ngx_memzero(&ccv, sizeof(ngx_http_compile_complex_value_t));
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	610
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	611 ccv.cf = cf;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	612 ccv.value = &value[1];
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	613 ccv.complex_value = &hcf->key;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	614
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	615 if (ngx_http_compile_complex_value(&ccv) != NGX_OK) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	616 return NGX_CONF_ERROR;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	617 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	618
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	619 uscf = ngx_http_conf_get_module_srv_conf(cf, ngx_http_upstream_module);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	620
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	621 if (uscf->peer.init_upstream) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	622 ngx_conf_log_error(NGX_LOG_WARN, cf, 0,
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	623 "load balancing method redefined");
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	624 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	625
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	626 uscf->flags = NGX_HTTP_UPSTREAM_CREATE
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	627 \|NGX_HTTP_UPSTREAM_WEIGHT
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	628 \|NGX_HTTP_UPSTREAM_MAX_FAILS
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	629 \|NGX_HTTP_UPSTREAM_FAIL_TIMEOUT
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	630 \|NGX_HTTP_UPSTREAM_DOWN;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	631
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	632 if (cf->args->nelts == 2) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	633 uscf->peer.init_upstream = ngx_http_upstream_init_hash;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	634
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	635 } else if (ngx_strcmp(value[2].data, "consistent") == 0) {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	636 uscf->peer.init_upstream = ngx_http_upstream_init_chash;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	637
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	638 } else {
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	639 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	640 "invalid parameter \"%V\"", &value[2]);
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	641 return NGX_CONF_ERROR;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	642 }
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	643
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	644 return NGX_CONF_OK;
efc84a5723b3 Upstream: generic hash module. Roman Arutyunyan <arut@nginx.com> parents: diff changeset	645 }

Mercurial > hg > nginx

annotate src/http/modules/ngx_http_upstream_hash_module.c @ 6099:6ff0ebd6fbf4