Merge branch 'mf/curl-select-fdset'

* mf/curl-select-fdset:
  http: drop "local" member from request struct
  http.c: Rely on select instead of tracking whether data was received
  http.c: Use timeout suggested by curl instead of fixed 50ms timeout
  http.c: Use curl_multi_fdset to select on curl fds instead of just sleeping
This commit is contained in:
Junio C Hamano 2011-12-05 15:10:28 -08:00
Родитель 62cdb6b23a 093c44a360
Коммит c4c9a63b54
2 изменённых файлов: 21 добавлений и 27 удалений

44
http.c
Просмотреть файл

@ -4,7 +4,6 @@
#include "run-command.h" #include "run-command.h"
#include "url.h" #include "url.h"
int data_received;
int active_requests; int active_requests;
int http_is_verbose; int http_is_verbose;
size_t http_post_buffer = 16 * LARGE_PACKET_MAX; size_t http_post_buffer = 16 * LARGE_PACKET_MAX;
@ -99,13 +98,11 @@ size_t fwrite_buffer(char *ptr, size_t eltsize, size_t nmemb, void *buffer_)
struct strbuf *buffer = buffer_; struct strbuf *buffer = buffer_;
strbuf_add(buffer, ptr, size); strbuf_add(buffer, ptr, size);
data_received++;
return size; return size;
} }
size_t fwrite_null(char *ptr, size_t eltsize, size_t nmemb, void *strbuf) size_t fwrite_null(char *ptr, size_t eltsize, size_t nmemb, void *strbuf)
{ {
data_received++;
return eltsize * nmemb; return eltsize * nmemb;
} }
@ -536,7 +533,6 @@ struct active_request_slot *get_active_slot(void)
active_requests++; active_requests++;
slot->in_use = 1; slot->in_use = 1;
slot->local = NULL;
slot->results = NULL; slot->results = NULL;
slot->finished = NULL; slot->finished = NULL;
slot->callback_data = NULL; slot->callback_data = NULL;
@ -640,8 +636,6 @@ void step_active_slots(void)
void run_active_slot(struct active_request_slot *slot) void run_active_slot(struct active_request_slot *slot)
{ {
#ifdef USE_CURL_MULTI #ifdef USE_CURL_MULTI
long last_pos = 0;
long current_pos;
fd_set readfds; fd_set readfds;
fd_set writefds; fd_set writefds;
fd_set excfds; fd_set excfds;
@ -651,25 +645,33 @@ void run_active_slot(struct active_request_slot *slot)
slot->finished = &finished; slot->finished = &finished;
while (!finished) { while (!finished) {
data_received = 0;
step_active_slots(); step_active_slots();
if (!data_received && slot->local != NULL) { if (slot->in_use) {
current_pos = ftell(slot->local); #if LIBCURL_VERSION_NUM >= 0x070f04
if (current_pos > last_pos) long curl_timeout;
data_received++; curl_multi_timeout(curlm, &curl_timeout);
last_pos = current_pos; if (curl_timeout == 0) {
continue;
} else if (curl_timeout == -1) {
select_timeout.tv_sec = 0;
select_timeout.tv_usec = 50000;
} else {
select_timeout.tv_sec = curl_timeout / 1000;
select_timeout.tv_usec = (curl_timeout % 1000) * 1000;
} }
#else
select_timeout.tv_sec = 0;
select_timeout.tv_usec = 50000;
#endif
if (slot->in_use && !data_received) { max_fd = -1;
max_fd = 0;
FD_ZERO(&readfds); FD_ZERO(&readfds);
FD_ZERO(&writefds); FD_ZERO(&writefds);
FD_ZERO(&excfds); FD_ZERO(&excfds);
select_timeout.tv_sec = 0; curl_multi_fdset(curlm, &readfds, &writefds, &excfds, &max_fd);
select_timeout.tv_usec = 50000;
select(max_fd, &readfds, &writefds, select(max_fd+1, &readfds, &writefds, &excfds, &select_timeout);
&excfds, &select_timeout);
} }
} }
#else #else
@ -814,7 +816,6 @@ static int http_request(const char *url, void *result, int target, int options)
headers = curl_slist_append(headers, buf.buf); headers = curl_slist_append(headers, buf.buf);
strbuf_reset(&buf); strbuf_reset(&buf);
} }
slot->local = result;
} else } else
curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION,
fwrite_buffer); fwrite_buffer);
@ -862,7 +863,6 @@ static int http_request(const char *url, void *result, int target, int options)
ret = HTTP_START_FAILED; ret = HTTP_START_FAILED;
} }
slot->local = NULL;
curl_slist_free_all(headers); curl_slist_free_all(headers);
strbuf_release(&buf); strbuf_release(&buf);
@ -1057,7 +1057,6 @@ void release_http_pack_request(struct http_pack_request *preq)
if (preq->packfile != NULL) { if (preq->packfile != NULL) {
fclose(preq->packfile); fclose(preq->packfile);
preq->packfile = NULL; preq->packfile = NULL;
preq->slot->local = NULL;
} }
if (preq->range_header != NULL) { if (preq->range_header != NULL) {
curl_slist_free_all(preq->range_header); curl_slist_free_all(preq->range_header);
@ -1079,7 +1078,6 @@ int finish_http_pack_request(struct http_pack_request *preq)
fclose(preq->packfile); fclose(preq->packfile);
preq->packfile = NULL; preq->packfile = NULL;
preq->slot->local = NULL;
lst = preq->lst; lst = preq->lst;
while (*lst != p) while (*lst != p)
@ -1148,7 +1146,6 @@ struct http_pack_request *new_http_pack_request(
} }
preq->slot = get_active_slot(); preq->slot = get_active_slot();
preq->slot->local = preq->packfile;
curl_easy_setopt(preq->slot->curl, CURLOPT_FILE, preq->packfile); curl_easy_setopt(preq->slot->curl, CURLOPT_FILE, preq->packfile);
curl_easy_setopt(preq->slot->curl, CURLOPT_WRITEFUNCTION, fwrite); curl_easy_setopt(preq->slot->curl, CURLOPT_WRITEFUNCTION, fwrite);
curl_easy_setopt(preq->slot->curl, CURLOPT_URL, preq->url); curl_easy_setopt(preq->slot->curl, CURLOPT_URL, preq->url);
@ -1205,7 +1202,6 @@ static size_t fwrite_sha1_file(char *ptr, size_t eltsize, size_t nmemb,
git_SHA1_Update(&freq->c, expn, git_SHA1_Update(&freq->c, expn,
sizeof(expn) - freq->stream.avail_out); sizeof(expn) - freq->stream.avail_out);
} while (freq->stream.avail_in && freq->zret == Z_OK); } while (freq->stream.avail_in && freq->zret == Z_OK);
data_received++;
return size; return size;
} }

2
http.h
Просмотреть файл

@ -49,7 +49,6 @@ struct slot_results {
struct active_request_slot { struct active_request_slot {
CURL *curl; CURL *curl;
FILE *local;
int in_use; int in_use;
CURLcode curl_result; CURLcode curl_result;
long http_code; long http_code;
@ -89,7 +88,6 @@ extern void step_active_slots(void);
extern void http_init(struct remote *remote, const char *url); extern void http_init(struct remote *remote, const char *url);
extern void http_cleanup(void); extern void http_cleanup(void);
extern int data_received;
extern int active_requests; extern int active_requests;
extern int http_is_verbose; extern int http_is_verbose;
extern size_t http_post_buffer; extern size_t http_post_buffer;