Akkariin/wrk
1
0
Fork 0
mirror of https://github.com/wg/wrk synced 2025-01-08 23:32:54 +08:00
Code Issues Releases Wiki Activity

stop after receiving SIGINT

Browse Source
This commit is contained in:
Will 2013-04-07 14:37:24 +09:00
parent 5496c588eb
commit 86a23c6beb
2 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Makefile
View File

@ -4,7 +4,7 @@ LIBS := -lpthread -lm
TARGET := $(shell uname -s | tr [A-Z] [a-z] 2>/dev/null || echo unknown)
ifeq ($(TARGET), sunos)
CFLAGS += -D_PTHREADS
CFLAGS += -D_PTHREADS -D_POSIX_C_SOURCE=200112L
LIBS += -lsocket
endif

16
src/wrk.c
View File

@ -51,6 +51,12 @@ static const struct http_parser_settings parser_settings = {
.on_message_complete = request_complete
};
static volatile sig_atomic_t stop = 0;
static void handler(int sig) {
stop = 1;
}
static void usage() {
printf("Usage: wrk <options> <url> \n"
" Options: \n"
@ -118,6 +124,7 @@ int main(int argc, char **argv) {
}
signal(SIGPIPE, SIG_IGN);
signal(SIGINT, SIG_IGN);
cfg.addr = *addr;
request.buf = format_request(host, port, path, headers);
request.size = strlen(request.buf);
@ -142,6 +149,13 @@ int main(int argc, char **argv) {
}
}
struct sigaction sa = {
.sa_handler = handler,
.sa_flags = 0,
};
sigfillset(&sa.sa_mask);
sigaction(SIGINT, &sa, NULL);
printf("Making %"PRIu64" requests to %s\n", cfg.requests, url);
printf(" %"PRIu64" threads and %"PRIu64" connections\n", cfg.threads, cfg.connections);
@ -323,7 +337,7 @@ static int check_timeouts(aeEventLoop *loop, long long id, void *data) {
errors += thread->errors.write;
errors += thread->errors.timeout;
if (errors >= cfg.errors) {
if (stop || errors >= cfg.errors) {
aeStop(loop);
}