diff options
author | Keith Packard <keithp@keithp.com> | 2009-11-13 09:05:13 -0800 |
---|---|---|
committer | Carl Worth <cworth@cworth.org> | 2009-11-13 09:05:13 -0800 |
commit | f4245aec94815176a52baae57a6c260b2bbae9c2 (patch) | |
tree | e7429a51e9d6b0937652f5082d4ef8cfeb3fc0ed /notmuch-tag.c | |
parent | 8561c7463a5de3bd0990d8244abb1c67ca4f4a7d (diff) |
notmuch new/tag: Flush all changes to database when interrupted.
By installing a signal handler for SIGINT we can ensure that no work
that is already complete will be lost if the user interrupts a
"notmuch new" run with Control-C.
Diffstat (limited to 'notmuch-tag.c')
-rw-r--r-- | notmuch-tag.c | 22 |
1 files changed, 20 insertions, 2 deletions
diff --git a/notmuch-tag.c b/notmuch-tag.c index 80582acf..7d92ec48 100644 --- a/notmuch-tag.c +++ b/notmuch-tag.c @@ -20,6 +20,16 @@ #include "notmuch-client.h" +static volatile sig_atomic_t interrupted; + +static void +handle_sigint (unused (int sig)) +{ + static char msg[] = "Stopping... \n"; + write(2, msg, sizeof(msg)-1); + interrupted = 1; +} + int notmuch_tag_command (void *ctx, unused (int argc), unused (char *argv[])) { @@ -32,8 +42,16 @@ notmuch_tag_command (void *ctx, unused (int argc), unused (char *argv[])) notmuch_query_t *query; notmuch_messages_t *messages; notmuch_message_t *message; + struct sigaction action; int i; + /* Setup our handler for SIGINT */ + memset (&action, 0, sizeof (struct sigaction)); + action.sa_handler = handle_sigint; + sigemptyset (&action.sa_mask); + action.sa_flags = SA_RESTART; + sigaction (SIGINT, &action, NULL); + add_tags = talloc_size (ctx, argc * sizeof (int)); if (add_tags == NULL) { fprintf (stderr, "Out of memory.\n"); @@ -87,7 +105,7 @@ notmuch_tag_command (void *ctx, unused (int argc), unused (char *argv[])) } for (messages = notmuch_query_search_messages (query, 0, -1); - notmuch_messages_has_more (messages); + notmuch_messages_has_more (messages) && !interrupted; notmuch_messages_advance (messages)) { message = notmuch_messages_get (messages); @@ -109,5 +127,5 @@ notmuch_tag_command (void *ctx, unused (int argc), unused (char *argv[])) notmuch_query_destroy (query); notmuch_database_close (notmuch); - return 0; + return interrupted; } |