3 /* Protoypes (static) */
4 static memcached_return
server_add(memcached_st
*ptr
, char *hostname
,
6 memcached_connection type
);
8 #define MEMCACHED_WHEEL_SIZE 1024
9 #define MEMCACHED_STRIDE 4
10 static void rebalance_wheel(memcached_st
*ptr
)
17 memset(ptr
->wheel
, 0, sizeof(unsigned int) * MEMCACHED_WHEEL_SIZE
);
19 for (latch
= y
= x
= 0; x
< MEMCACHED_WHEEL_SIZE
; x
++, latch
++)
21 if (latch
== MEMCACHED_STRIDE
)
24 if (y
== ptr
->number_of_hosts
)
33 static int compare_servers(const void *p1
, const void *p2
)
36 memcached_server_st
*a
= (memcached_server_st
*)p1
;
37 memcached_server_st
*b
= (memcached_server_st
*)p2
;
39 return_value
= strcmp(a
->hostname
, b
->hostname
);
41 if (return_value
== 0)
43 if (a
->port
> b
->port
)
52 void sort_hosts(memcached_st
*ptr
)
54 if (ptr
->number_of_hosts
)
56 qsort(ptr
->hosts
, ptr
->number_of_hosts
, sizeof(memcached_server_st
), compare_servers
);
57 ptr
->hosts
[0].count
= ptr
->number_of_hosts
;
61 static void host_reset(memcached_st
*ptr
, memcached_server_st
*host
,
62 char *hostname
, unsigned int port
,
63 memcached_connection type
)
65 memset(host
, 0, sizeof(memcached_server_st
));
66 strncpy(host
->hostname
, hostname
, MEMCACHED_MAX_HOST_LENGTH
- 1);
67 host
->root
= ptr
? ptr
: NULL
;
71 host
->read_ptr
= host
->read_buffer
;
73 host
->next_retry
= ptr
->retry_timeout
;
74 host
->sockaddr_inited
= MEMCACHED_NOT_ALLOCATED
;
77 void server_list_free(memcached_st
*ptr
, memcached_server_st
*servers
)
84 for (x
= 0; x
< servers
->count
; x
++)
85 if (servers
[x
].address_info
)
86 freeaddrinfo(servers
[x
].address_info
);
88 if (ptr
&& ptr
->call_free
)
89 ptr
->call_free(ptr
, servers
);
94 memcached_return
memcached_server_push(memcached_st
*ptr
, memcached_server_st
*list
)
98 memcached_server_st
*new_host_list
;
101 return MEMCACHED_SUCCESS
;
103 count
= list
[0].count
;
105 if (ptr
->call_realloc
)
107 (memcached_server_st
*)ptr
->call_realloc(ptr
, ptr
->hosts
,
108 sizeof(memcached_server_st
) * (count
+ ptr
->number_of_hosts
));
111 (memcached_server_st
*)realloc(ptr
->hosts
,
112 sizeof(memcached_server_st
) * (count
+ ptr
->number_of_hosts
));
115 return MEMCACHED_MEMORY_ALLOCATION_FAILURE
;
117 ptr
->hosts
= new_host_list
;
119 for (x
= 0; x
< count
; x
++)
121 WATCHPOINT_ASSERT(list
[x
].hostname
[0] != 0);
122 host_reset(ptr
, &ptr
->hosts
[ptr
->number_of_hosts
], list
[x
].hostname
,
123 list
[x
].port
, list
[x
].type
);
124 ptr
->number_of_hosts
++;
126 ptr
->hosts
[0].count
= ptr
->number_of_hosts
;
128 if (ptr
->flags
& MEM_USE_SORT_HOSTS
)
131 rebalance_wheel(ptr
);
133 return MEMCACHED_SUCCESS
;
136 memcached_return
memcached_server_add_unix_socket(memcached_st
*ptr
, char *filename
)
139 return MEMCACHED_FAILURE
;
141 return server_add(ptr
, filename
, 0, MEMCACHED_CONNECTION_UNIX_SOCKET
);
144 memcached_return
memcached_server_add_udp(memcached_st
*ptr
,
149 port
= MEMCACHED_DEFAULT_PORT
;
152 hostname
= "localhost";
154 return server_add(ptr
, hostname
, port
, MEMCACHED_CONNECTION_UDP
);
157 memcached_return
memcached_server_add(memcached_st
*ptr
,
162 port
= MEMCACHED_DEFAULT_PORT
;
165 hostname
= "localhost";
167 return server_add(ptr
, hostname
, port
, MEMCACHED_CONNECTION_TCP
);
170 static memcached_return
server_add(memcached_st
*ptr
, char *hostname
,
172 memcached_connection type
)
174 memcached_server_st
*new_host_list
;
175 LIBMEMCACHED_MEMCACHED_SERVER_ADD_START();
178 if (ptr
->call_realloc
)
179 new_host_list
= (memcached_server_st
*)ptr
->call_realloc(ptr
, ptr
->hosts
,
180 sizeof(memcached_server_st
) * (ptr
->number_of_hosts
+1));
182 new_host_list
= (memcached_server_st
*)realloc(ptr
->hosts
,
183 sizeof(memcached_server_st
) * (ptr
->number_of_hosts
+1));
184 if (new_host_list
== NULL
)
185 return MEMCACHED_MEMORY_ALLOCATION_FAILURE
;
187 ptr
->hosts
= new_host_list
;
189 host_reset(ptr
, &ptr
->hosts
[ptr
->number_of_hosts
], hostname
, port
, type
);
190 ptr
->number_of_hosts
++;
192 if (ptr
->flags
& MEM_USE_SORT_HOSTS
)
195 ptr
->hosts
[0].count
= ptr
->number_of_hosts
;
197 rebalance_wheel(ptr
);
199 LIBMEMCACHED_MEMCACHED_SERVER_ADD_END();
201 return MEMCACHED_SUCCESS
;
204 memcached_server_st
*memcached_server_list_append(memcached_server_st
*ptr
,
205 char *hostname
, unsigned int port
,
206 memcached_return
*error
)
209 memcached_server_st
*new_host_list
;
211 if (hostname
== NULL
|| error
== NULL
)
215 port
= MEMCACHED_DEFAULT_PORT
;
217 /* Increment count for hosts */
221 count
+= ptr
[0].count
;
224 new_host_list
= (memcached_server_st
*)realloc(ptr
, sizeof(memcached_server_st
) * count
);
227 *error
= MEMCACHED_MEMORY_ALLOCATION_FAILURE
;
231 host_reset(NULL
, &new_host_list
[count
-1], hostname
, port
, MEMCACHED_CONNECTION_TCP
);
233 /* Backwards compatibility hack */
234 new_host_list
[0].count
= count
;
236 *error
= MEMCACHED_SUCCESS
;
237 return new_host_list
;
240 unsigned int memcached_server_list_count(memcached_server_st
*ptr
)
248 void memcached_server_list_free(memcached_server_st
*ptr
)
250 server_list_free(NULL
, ptr
);