X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=lib%2Fmemcached_connect.c;h=2120db5c7b4e3d3a5f5aa8bde378490bbf6c765f;hb=3529e69b4a637b6875d82cb22a250d8abc6e8e85;hp=8a6acdd7e5d5430628d2a49c066ac5ece5ffe281;hpb=1f34771f7ba527e3afb675f1a448d6c5ab66596b;p=m6w6%2Flibmemcached diff --git a/lib/memcached_connect.c b/lib/memcached_connect.c index 8a6acdd7..2120db5c 100644 --- a/lib/memcached_connect.c +++ b/lib/memcached_connect.c @@ -1,53 +1,59 @@ -#include +#include "common.h" + +#include memcached_return memcached_connect(memcached_st *ptr) { - int rc; + unsigned int x; struct sockaddr_in localAddr, servAddr; struct hostent *h; - char *server_name= "localhost"; - - if (ptr->connected) - return MEMCACHED_SUCCESS; - - if ((h= gethostbyname(server_name)) == NULL) - { - fprintf(stderr, "unknown host '%s'\n", server_name); - return MEMCACHED_HOST_LOCKUP_FAILURE; - } - - servAddr.sin_family= h->h_addrtype; - memcpy((char *) &servAddr.sin_addr.s_addr, h->h_addr_list[0], h->h_length); - servAddr.sin_port = htons(MEMCACHED_DEFAULT_PORT); - - /* Create the socket */ - if ((ptr->fd= socket(AF_INET, SOCK_STREAM, 0)) < 0) - { - fprintf(stderr, "cannot open socket"); - return MEMCACHED_CONNECTION_SOCKET_CREATE_FAILURE; - } + LIBMEMCACHED_MEMCACHED_CONNECT_START(); + if (ptr->connected == ptr->number_of_hosts) + return MEMCACHED_SUCCESS; - /* bind any port number */ - localAddr.sin_family = AF_INET; - localAddr.sin_addr.s_addr = htonl(INADDR_ANY); - localAddr.sin_port = htons(0); + if (!ptr->hosts) + return MEMCACHED_NO_SERVERS; - if (bind(ptr->fd, (struct sockaddr *) &localAddr, sizeof(localAddr)) < 0) + for (x= 0; x < ptr->number_of_hosts; x++) { - fprintf(stderr, "cannot bind port TCP %u\n", MEMCACHED_DEFAULT_PORT); - return(MEMCACHED_CONNECTION_BIND_FAILURE); + if (ptr->hosts[x].fd == -1) + { + int flags; + + if ((h= gethostbyname(ptr->hosts[x].hostname)) == NULL) + return MEMCACHED_HOST_LOCKUP_FAILURE; + + servAddr.sin_family= h->h_addrtype; + memcpy((char *) &servAddr.sin_addr.s_addr, h->h_addr_list[0], h->h_length); + servAddr.sin_port = htons(ptr->hosts[x].port); + + /* Create the socket */ + if ((ptr->hosts[0].fd= socket(AF_INET, SOCK_STREAM, 0)) < 0) + return MEMCACHED_CONNECTION_SOCKET_CREATE_FAILURE; + + + /* bind any port number */ + localAddr.sin_family = AF_INET; + localAddr.sin_addr.s_addr = htonl(INADDR_ANY); + localAddr.sin_port = htons(0); + +#ifdef NOT_YET + /* For the moment, not getting a nonblocking mode will note be fatal */ + flags= fcntl(ptr->hosts[0].fd, F_GETFL, 0); + if (flags != -1) + (void)fcntl(ptr->hosts[0].fd, F_SETFL, flags | O_NONBLOCK); +#endif + + /* connect to server */ + if (connect(ptr->hosts[0].fd, (struct sockaddr *) &servAddr, sizeof(servAddr)) < 0) + return MEMCACHED_HOST_LOCKUP_FAILURE; + + ptr->connected++; + } } - - /* connect to server */ - if (connect(ptr->fd, (struct sockaddr *) &servAddr, sizeof(servAddr)) < 0) - { - fprintf(stderr, "cannot connect to host '%s'\n", server_name); - return MEMCACHED_HOST_LOCKUP_FAILURE; - } - - ptr->connected= 1; + LIBMEMCACHED_MEMCACHED_CONNECT_END(); return MEMCACHED_SUCCESS; }