DTrace support

[awesomized/libmemcached] / lib / memcached_connect.c

diff --git a/lib/memcached_connect.c b/lib/memcached_connect.c
index 899101a2f4199ddf7342aa08e8b692eebfb9a28d..516b5db346482e24500854deef4fbc0c4267aef5 100644 (file)
--- a/lib/memcached_connect.c
+++ b/lib/memcached_connect.c
@@ -1,9 +1,10 @@
-#include <memcached.h>
+#include "common.h"
 
 memcached_return memcached_server_add(memcached_st *ptr, char *hostname, unsigned int port)
 {
   memcached_server_st *new_host_list;
   char *new_hostname;
+  LIBMEMCACHED_MEMCACHED_SERVER_ADD_START();
 
   if (!port)
     port= MEMCACHED_DEFAULT_PORT; 
@@ -34,6 +35,8 @@ memcached_return memcached_server_add(memcached_st *ptr, char *hostname, unsigne
   ptr->hosts[ptr->number_of_hosts].fd= -1;
   ptr->number_of_hosts++;
 
+  LIBMEMCACHED_MEMCACHED_SERVER_ADD_END();
+
   return MEMCACHED_SUCCESS;
 }
 
@@ -43,6 +46,8 @@ memcached_return memcached_connect(memcached_st *ptr)
   struct sockaddr_in localAddr, servAddr;
   struct hostent *h;
 
+  LIBMEMCACHED_MEMCACHED_CONNECT_START();
+
   if (ptr->connected == ptr->number_of_hosts)
     return MEMCACHED_SUCCESS;
 
@@ -82,6 +87,7 @@ memcached_return memcached_connect(memcached_st *ptr)
       ptr->connected++;
     }
   }
+  LIBMEMCACHED_MEMCACHED_CONNECT_END();
 
   return MEMCACHED_SUCCESS;
 }