git.m6w6.name Git - awesomized/libmemcached/blob - 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 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 memcached_server_st *memcached_server_list(const memcached_st *self)
  70 {
  71   return self->servers;
  72 }
  73
  74 void memcached_server_list_set(memcached_st *self, memcached_server_st *list)
  75 {
  76   self->servers= list;
  77 }