2 Basic socket buffered IO
6 #include "memcached_io.h"
7 #include <sys/select.h>
10 int io_wait(memcached_st
*ptr
, unsigned int server_key
, unsigned read_or_write
)
16 flags
= POLLOUT
| POLLERR
;
18 flags
= POLLIN
| POLLERR
;
20 memset(&fds
, 0, sizeof(struct pollfd
));
21 fds
[0].fd
= ptr
->hosts
[server_key
].fd
;
24 if (poll(fds
, 1, -1) < 0)
25 return MEMCACHED_FAILURE
;
27 return MEMCACHED_SUCCESS
;
32 struct timeval local_tv
;
35 memset(&local_tv
, 0, sizeof(struct timeval
));
38 local_tv
.tv_usec
= 300;
41 FD_SET(ptr
->hosts
[server_key
].fd
, &set
);
44 select_return
= select(1, &set
, NULL
, NULL
, &local_tv
);
46 select_return
= select(1, NULL
, &set
, NULL
, &local_tv
);
48 if (select_return
== -1)
51 return MEMCACHED_FAILURE
;
53 else if (!select_return
)
57 return MEMCACHED_SUCCESS
;
61 ssize_t
memcached_io_read(memcached_st
*ptr
, unsigned int server_key
,
62 char *buffer
, size_t length
)
70 if (!ptr
->read_buffer_length
)
76 if (ptr
->flags
& MEM_NO_BLOCK
)
80 rc
= io_wait(ptr
, server_key
, 0);
81 if (rc
!= MEMCACHED_SUCCESS
)
85 data_read
= recv(ptr
->hosts
[server_key
].fd
,
87 MEMCACHED_MAX_BUFFER
, 0);
103 /* If zero, just keep looping */
106 ptr
->read_buffer_length
= data_read
;
107 ptr
->read_ptr
= ptr
->read_buffer
;
110 *buffer_ptr
= *ptr
->read_ptr
;
113 ptr
->read_buffer_length
--;
117 return (size_t)(buffer_ptr
- buffer
);
120 ssize_t
memcached_io_write(memcached_st
*ptr
, unsigned int server_key
,
121 char *buffer
, size_t length
, char with_flush
)
123 unsigned long long x
;
125 for (x
= 0; x
< length
; x
++)
127 ptr
->write_buffer
[ptr
->write_buffer_offset
]= buffer
[x
];
128 ptr
->write_buffer_offset
++;
130 if (ptr
->write_buffer_offset
== MEMCACHED_MAX_BUFFER
)
134 sent_length
= memcached_io_flush(ptr
, server_key
);
136 assert(sent_length
== MEMCACHED_MAX_BUFFER
);
137 ptr
->write_buffer_offset
= 0;
143 if (memcached_io_flush(ptr
, server_key
) == -1)
150 ssize_t
memcached_io_flush(memcached_st
*ptr
, unsigned int server_key
)
153 char *write_ptr
= ptr
->write_buffer
;
154 size_t write_length
= ptr
->write_buffer_offset
;
155 unsigned int loop
= 1;
157 if (ptr
->write_buffer_offset
== 0)
162 if (ptr
->flags
& MEM_NO_BLOCK
)
166 rc
= io_wait(ptr
, server_key
, 1);
167 if (rc
!= MEMCACHED_SUCCESS
)
173 if ((sent_length
= send(ptr
->hosts
[server_key
].fd
, write_ptr
,
174 write_length
, 0)) == -1)
176 if ((sent_length
= write(ptr
->hosts
[server_key
].fd
, write_ptr
,
177 write_length
)) == -1)
188 /* Yes, we want to fall through */
190 ptr
->my_errno
= errno
;
196 write_ptr
+= sent_length
;
197 write_length
-= sent_length
;
201 ptr
->write_buffer_offset
= 0;
207 Eventually we will just kill off the server with the problem.
209 void memcached_io_reset(memcached_st
*ptr
, unsigned int server_key
)
211 ptr
->write_buffer_offset
= 0;