54a4b8dcd16d6c154604349533c03c9a067942e0
2 Basic socket buffered IO
6 #include "memcached_io.h"
7 #include <sys/select.h>
9 ssize_t
memcached_io_read(memcached_st
*ptr
, unsigned int server_key
,
10 char *buffer
, size_t length
)
18 if (!ptr
->read_buffer_length
)
24 if (ptr
->flags
& MEM_NO_BLOCK
)
29 struct timeval local_tv
;
32 memset(&local_tv
, 0, sizeof(struct timeval
));
35 local_tv
.tv_usec
= 300;
38 FD_SET(ptr
->hosts
[server_key
].fd
, &set
);
40 select_return
= select(1, &set
, NULL
, NULL
, &local_tv
);
42 if (select_return
== -1)
47 else if (!select_return
)
52 data_read
= recv(ptr
->hosts
[server_key
].fd
,
54 MEMCACHED_MAX_BUFFER
, 0);
70 /* If zero, just keep looping */
73 ptr
->read_buffer_length
= data_read
;
74 ptr
->read_ptr
= ptr
->read_buffer
;
77 *buffer_ptr
= *ptr
->read_ptr
;
80 ptr
->read_buffer_length
--;
84 return (size_t)(buffer_ptr
- buffer
);
87 ssize_t
memcached_io_write(memcached_st
*ptr
, unsigned int server_key
,
88 char *buffer
, size_t length
, char with_flush
)
92 for (x
= 0; x
< length
; x
++)
94 ptr
->write_buffer
[ptr
->write_buffer_offset
]= buffer
[x
];
95 ptr
->write_buffer_offset
++;
97 if (ptr
->write_buffer_offset
== MEMCACHED_MAX_BUFFER
)
101 sent_length
= memcached_io_flush(ptr
, server_key
);
103 assert(sent_length
== MEMCACHED_MAX_BUFFER
);
104 ptr
->write_buffer_offset
= 0;
110 if (memcached_io_flush(ptr
, server_key
) == -1)
117 ssize_t
memcached_io_flush(memcached_st
*ptr
, unsigned int server_key
)
120 char *write_ptr
= ptr
->write_buffer
;
121 size_t write_length
= ptr
->write_buffer_offset
;
122 unsigned int loop
= 1;
124 if (ptr
->write_buffer_offset
== 0)
129 if (ptr
->flags
& MEM_NO_BLOCK
)
134 struct timeval local_tv
;
139 local_tv
.tv_usec
= 300 * loop
;
142 FD_SET(ptr
->hosts
[server_key
].fd
, &set
);
144 select_return
= select(1, NULL
, &set
, NULL
, &local_tv
);
146 if (select_return
== -1)
148 ptr
->my_errno
= errno
;
151 else if (!select_return
)
158 if ((sent_length
= send(ptr
->hosts
[server_key
].fd
, write_ptr
,
159 write_length
, 0)) == -1)
161 if ((sent_length
= write(ptr
->hosts
[server_key
].fd
, write_ptr
,
162 write_length
)) == -1)
173 /* Yes, we want to fall through */
175 ptr
->my_errno
= errno
;
181 write_ptr
+= sent_length
;
182 write_length
-= sent_length
;
186 ptr
->write_buffer_offset
= 0;
192 Eventually we will just kill off the server with the problem.
194 void memcached_io_reset(memcached_st
*ptr
, unsigned int server_key
)
196 ptr
->write_buffer_offset
= 0;