A number of fixes/etc.
[m6w6/libmemcached] / libmemcached / server_list.c
1 /* LibMemcached
2 * Copyright (C) 2006-2010 Brian Aker
3 * All rights reserved.
4 *
5 * Use and distribution licensed under the BSD license. See
6 * the COPYING file in the parent directory for full text.
7 *
8 * Summary:
9 *
10 */
11
12
13 #include "common.h"
14
15 memcached_server_list_st
16 memcached_server_list_append_with_weight(memcached_server_list_st ptr,
17 const char *hostname, in_port_t port,
18 uint32_t weight,
19 memcached_return_t *error)
20 {
21 uint32_t count;
22 memcached_server_list_st new_host_list;
23
24 if (hostname == NULL || error == NULL)
25 return NULL;
26
27 if (! port)
28 port= MEMCACHED_DEFAULT_PORT;
29
30 /* Increment count for hosts */
31 count= 1;
32 if (ptr != NULL)
33 {
34 count+= memcached_server_list_count(ptr);
35 }
36
37 new_host_list= (memcached_server_write_instance_st)realloc(ptr, sizeof(memcached_server_st) * count);
38 if (!new_host_list)
39 {
40 *error= MEMCACHED_MEMORY_ALLOCATION_FAILURE;
41 return NULL;
42 }
43
44 /* TODO: Check return type */
45 memcached_server_create_with(NULL, &new_host_list[count-1], hostname, port, weight, MEMCACHED_CONNECTION_TCP);
46
47 /* Backwards compatibility hack */
48 memcached_servers_set_count(new_host_list, count);
49
50 *error= MEMCACHED_SUCCESS;
51 return new_host_list;
52 }
53
54 memcached_server_list_st
55 memcached_server_list_append(memcached_server_list_st ptr,
56 const char *hostname, in_port_t port,
57 memcached_return_t *error)
58 {
59 return memcached_server_list_append_with_weight(ptr, hostname, port, 0, error);
60 }
61
62 inline uint32_t memcached_server_list_count(const memcached_server_list_st self)
63 {
64 return (self == NULL)
65 ? 0
66 : self->number_of_hosts;
67 }
68
69 inline memcached_server_st *memcached_server_list(const memcached_st *self)
70 {
71 return self->servers;
72 }
73
74 inline void memcached_server_list_set(memcached_st *self, memcached_server_st *list)
75 {
76 self->servers= list;
77 }