3 /* Protoypes (static) */
4 static memcached_return
server_add(memcached_st
*ptr
, char *hostname
,
6 memcached_connection type
);
7 memcached_return
update_continuum(memcached_st
*ptr
);
9 #define MEMCACHED_WHEEL_SIZE 1024
10 #define MEMCACHED_STRIDE 4
11 static memcached_return
rebalance_wheel(memcached_st
*ptr
)
17 if (ptr
->number_of_hosts
> ptr
->wheel_count
)
21 if (ptr
->call_realloc
)
22 new_ptr
= (uint32_t *)ptr
->call_realloc(ptr
, ptr
->wheel
, sizeof(uint32_t) * (ptr
->number_of_hosts
+ MEMCACHED_CONTINUUM_ADDITION
) * MEMCACHED_STRIDE
);
24 new_ptr
= (uint32_t *)realloc(ptr
->wheel
, sizeof(uint32_t) * (ptr
->number_of_hosts
+ MEMCACHED_CONTINUUM_ADDITION
) * MEMCACHED_STRIDE
);
27 return MEMCACHED_MEMORY_ALLOCATION_FAILURE
;
30 ptr
->wheel_count
= ptr
->number_of_hosts
+ MEMCACHED_CONTINUUM_ADDITION
;
34 memset(ptr
->wheel
, 0, sizeof(uint32_t) * MEMCACHED_STRIDE
* ptr
->wheel_count
);
36 for (latch
= y
= x
= 0; x
< MEMCACHED_STRIDE
* ptr
->wheel_count
; x
++, latch
++)
38 if (latch
== MEMCACHED_STRIDE
)
41 if (y
== ptr
->number_of_hosts
)
49 return MEMCACHED_SUCCESS
;
52 static int compare_servers(const void *p1
, const void *p2
)
55 memcached_server_st
*a
= (memcached_server_st
*)p1
;
56 memcached_server_st
*b
= (memcached_server_st
*)p2
;
58 return_value
= strcmp(a
->hostname
, b
->hostname
);
60 if (return_value
== 0)
62 return_value
= (int) (a
->port
- b
->port
);
68 void sort_hosts(memcached_st
*ptr
)
70 if (ptr
->number_of_hosts
)
72 qsort(ptr
->hosts
, ptr
->number_of_hosts
, sizeof(memcached_server_st
), compare_servers
);
73 ptr
->hosts
[0].count
= ptr
->number_of_hosts
;
78 memcached_return
run_distribution(memcached_st
*ptr
)
80 switch (ptr
->distribution
)
82 case MEMCACHED_DISTRIBUTION_CONSISTENT
:
83 case MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA
:
84 return update_continuum(ptr
);
85 case MEMCACHED_DISTRIBUTION_CONSISTENT_WHEEL
:
86 return rebalance_wheel(ptr
);
88 case MEMCACHED_DISTRIBUTION_MODULA
:
89 if (ptr
->flags
& MEM_USE_SORT_HOSTS
)
93 WATCHPOINT_ASSERT(0); /* We have added a distribution without extending the logic */
96 return MEMCACHED_SUCCESS
;
99 static void host_reset(memcached_st
*ptr
, memcached_server_st
*host
,
100 char *hostname
, unsigned int port
,
101 memcached_connection type
)
103 memset(host
, 0, sizeof(memcached_server_st
));
104 strncpy(host
->hostname
, hostname
, MEMCACHED_MAX_HOST_LENGTH
- 1);
105 host
->root
= ptr
? ptr
: NULL
;
109 host
->read_ptr
= host
->read_buffer
;
111 host
->next_retry
= ptr
->retry_timeout
;
112 host
->sockaddr_inited
= MEMCACHED_NOT_ALLOCATED
;
115 void server_list_free(memcached_st
*ptr
, memcached_server_st
*servers
)
122 for (x
= 0; x
< servers
->count
; x
++)
123 if (servers
[x
].address_info
)
125 freeaddrinfo(servers
[x
].address_info
);
126 servers
[x
].address_info
= NULL
;
129 if (ptr
&& ptr
->call_free
)
130 ptr
->call_free(ptr
, servers
);
135 static int continuum_item_cmp(const void *t1
, const void *t2
)
137 memcached_continuum_item_st
*ct1
= (memcached_continuum_item_st
*)t1
;
138 memcached_continuum_item_st
*ct2
= (memcached_continuum_item_st
*)t2
;
140 WATCHPOINT_ASSERT(ct1
->value
!= 153);
141 if (ct1
->value
== ct2
->value
)
143 else if (ct1
->value
> ct2
->value
)
149 memcached_return
update_continuum(memcached_st
*ptr
)
153 uint32_t continuum_index
= 0;
155 memcached_server_st
*list
;
157 if (ptr
->number_of_hosts
> ptr
->continuum_count
)
159 memcached_continuum_item_st
*new_ptr
;
161 if (ptr
->call_realloc
)
162 new_ptr
= (memcached_continuum_item_st
*)ptr
->call_realloc(ptr
, ptr
->continuum
, sizeof(memcached_continuum_item_st
) * (ptr
->number_of_hosts
+ MEMCACHED_CONTINUUM_ADDITION
) * MEMCACHED_POINTS_PER_SERVER
);
164 new_ptr
= (memcached_continuum_item_st
*)realloc(ptr
->continuum
, sizeof(memcached_continuum_item_st
) * (ptr
->number_of_hosts
+ MEMCACHED_CONTINUUM_ADDITION
) * MEMCACHED_POINTS_PER_SERVER
);
167 return MEMCACHED_MEMORY_ALLOCATION_FAILURE
;
169 ptr
->continuum
= new_ptr
;
170 ptr
->continuum_count
= ptr
->number_of_hosts
+ MEMCACHED_CONTINUUM_ADDITION
;
174 for (host_index
= 0; host_index
< ptr
->number_of_hosts
; ++host_index
)
176 for(index
= 1; index
<= MEMCACHED_POINTS_PER_SERVER
; ++index
)
178 char sort_host
[MEMCACHED_MAX_HOST_SORT_LENGTH
]= "";
179 size_t sort_host_length
;
181 sort_host_length
= snprintf(sort_host
, MEMCACHED_MAX_HOST_SORT_LENGTH
, "%s:%d-%d",
182 list
[host_index
].hostname
, list
[host_index
].port
, index
);
183 WATCHPOINT_ASSERT(sort_host_length
);
184 value
= generate_hash(ptr
, sort_host
, sort_host_length
);
185 ptr
->continuum
[continuum_index
].index
= host_index
;
186 ptr
->continuum
[continuum_index
++].value
= value
;
190 WATCHPOINT_ASSERT(ptr
->number_of_hosts
* MEMCACHED_POINTS_PER_SERVER
<= MEMCACHED_CONTINUUM_SIZE
);
191 qsort(ptr
->continuum
, ptr
->number_of_hosts
* MEMCACHED_POINTS_PER_SERVER
, sizeof(memcached_continuum_item_st
), continuum_item_cmp
);
193 for (index
= 0; index
< ((ptr
->number_of_hosts
* MEMCACHED_POINTS_PER_SERVER
) - 1); index
++)
195 WATCHPOINT_ASSERT(ptr
->continuum
[index
].value
<= ptr
->continuum
[index
+ 1].value
);
199 return MEMCACHED_SUCCESS
;
203 memcached_return
memcached_server_push(memcached_st
*ptr
, memcached_server_st
*list
)
207 memcached_server_st
*new_host_list
;
210 return MEMCACHED_SUCCESS
;
212 count
= list
[0].count
;
214 if (ptr
->call_realloc
)
216 (memcached_server_st
*)ptr
->call_realloc(ptr
, ptr
->hosts
,
217 sizeof(memcached_server_st
) * (count
+ ptr
->number_of_hosts
));
220 (memcached_server_st
*)realloc(ptr
->hosts
,
221 sizeof(memcached_server_st
) * (count
+ ptr
->number_of_hosts
));
224 return MEMCACHED_MEMORY_ALLOCATION_FAILURE
;
226 ptr
->hosts
= new_host_list
;
228 for (x
= 0; x
< count
; x
++)
230 WATCHPOINT_ASSERT(list
[x
].hostname
[0] != 0);
231 host_reset(ptr
, &ptr
->hosts
[ptr
->number_of_hosts
], list
[x
].hostname
,
232 list
[x
].port
, list
[x
].type
);
233 ptr
->number_of_hosts
++;
235 ptr
->hosts
[0].count
= ptr
->number_of_hosts
;
237 return run_distribution(ptr
);
240 memcached_return
memcached_server_add_unix_socket(memcached_st
*ptr
, char *filename
)
243 return MEMCACHED_FAILURE
;
245 return server_add(ptr
, filename
, 0, MEMCACHED_CONNECTION_UNIX_SOCKET
);
248 memcached_return
memcached_server_add_udp(memcached_st
*ptr
,
253 port
= MEMCACHED_DEFAULT_PORT
;
256 hostname
= "localhost";
258 return server_add(ptr
, hostname
, port
, MEMCACHED_CONNECTION_UDP
);
261 memcached_return
memcached_server_add(memcached_st
*ptr
,
266 port
= MEMCACHED_DEFAULT_PORT
;
269 hostname
= "localhost";
271 return server_add(ptr
, hostname
, port
, MEMCACHED_CONNECTION_TCP
);
274 static memcached_return
server_add(memcached_st
*ptr
, char *hostname
,
276 memcached_connection type
)
278 memcached_server_st
*new_host_list
;
280 if (ptr
->call_realloc
)
281 new_host_list
= (memcached_server_st
*)ptr
->call_realloc(ptr
, ptr
->hosts
,
282 sizeof(memcached_server_st
) * (ptr
->number_of_hosts
+1));
284 new_host_list
= (memcached_server_st
*)realloc(ptr
->hosts
,
285 sizeof(memcached_server_st
) * (ptr
->number_of_hosts
+1));
286 if (new_host_list
== NULL
)
287 return MEMCACHED_MEMORY_ALLOCATION_FAILURE
;
289 ptr
->hosts
= new_host_list
;
291 host_reset(ptr
, &ptr
->hosts
[ptr
->number_of_hosts
], hostname
, port
, type
);
292 ptr
->number_of_hosts
++;
293 ptr
->hosts
[0].count
= ptr
->number_of_hosts
;
295 return run_distribution(ptr
);
298 memcached_server_st
*memcached_server_list_append(memcached_server_st
*ptr
,
299 char *hostname
, unsigned int port
,
300 memcached_return
*error
)
303 memcached_server_st
*new_host_list
;
305 if (hostname
== NULL
|| error
== NULL
)
309 port
= MEMCACHED_DEFAULT_PORT
;
311 /* Increment count for hosts */
315 count
+= ptr
[0].count
;
318 new_host_list
= (memcached_server_st
*)realloc(ptr
, sizeof(memcached_server_st
) * count
);
321 *error
= MEMCACHED_MEMORY_ALLOCATION_FAILURE
;
325 host_reset(NULL
, &new_host_list
[count
-1], hostname
, port
, MEMCACHED_CONNECTION_TCP
);
327 /* Backwards compatibility hack */
328 new_host_list
[0].count
= count
;
330 *error
= MEMCACHED_SUCCESS
;
331 return new_host_list
;
334 unsigned int memcached_server_list_count(memcached_server_st
*ptr
)
342 void memcached_server_list_free(memcached_server_st
*ptr
)
344 server_list_free(NULL
, ptr
);