Introduce -E argument (don't exit on error)

This introduce a new ERROR macro, similar to FATAL By default ERROR has the behaviour of FATAL If you use -E, it will not exit Signed-off-by: Etienne Champetier <[email protected]>
edenhill · Mar 13, 2017 · f416dec · f416dec
1 parent 9deecb5
commit f416dec
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 1 deletion.
diff --git a/kafkacat.c b/kafkacat.c
@@ -53,6 +53,7 @@
 struct conf conf = {
         .run = 1,
         .verbosity = 1,
+        .exitonerror = 1,
         .partition = RD_KAFKA_PARTITION_UA,
         .msg_size = 1024*1024,
         .null_str = "NULL",
@@ -94,6 +95,26 @@ void RD_NORETURN fatal0 (const char *func, int line,
         exit(1);
 }
 
+/**
+ * Print error and exit if needed
+ */
+void error0 (int exitonerror, const char *func, int line, const char *fmt, ...) {
+        va_list ap;
+        char buf[1024];
+
+        va_start(ap, fmt);
+        vsnprintf(buf, sizeof(buf), fmt, ap);
+        va_end(ap);
+
+        if (exitonerror)
+                INFO(2, "Error at %s:%i:\n", func, line);
+
+        fprintf(stderr, "%% ERROR: %s%s\n", buf, exitonerror ? " : terminating":"");
+
+        if (exitonerror)
+                exit(1);
+}
+
 
 
 /**
@@ -887,6 +908,7 @@ static void RD_NORETURN usage (const char *argv0, int exitcode,
                 "  -D <delim>         Message delimiter character:\n"
                 "                     a-z.. | \\r | \\n | \\t | \\xNN\n"
                 "                     Default: \\n\n"
+                "  -E                 Do not exit on non fatal error\n"
                 "  -K <delim>         Key delimiter (same format as -D)\n"
                 "  -c <cnt>           Limit message count\n"
                 "  -X list            List available librdkafka configuration "
@@ -1116,7 +1138,7 @@ static void argparse (int argc, char **argv,
         int do_conf_dump = 0;
 
         while ((opt = getopt(argc, argv,
-                             "PCG:LQt:p:b:z:o:eD:K:Od:qvX:c:Tuf:ZlVh"
+                             "PCG:LQt:p:b:z:o:eED:K:Od:qvX:c:Tuf:ZlVh"
 #if ENABLE_JSON
                              "J"
 #endif
@@ -1178,6 +1200,9 @@ static void argparse (int argc, char **argv,
                 case 'e':
                         conf.exit_eof = 1;
                         break;
+                case 'E':
+                        conf.exitonerror = 0;
+                        break;
                 case 'f':
                         fmt = optarg;
                         break;

diff --git a/kafkacat.h b/kafkacat.h
@@ -63,6 +63,7 @@ struct conf {
         int     run;
         int     verbosity;
         int     exitcode;
+        int     exitonerror;
         char    mode;
         int     flags;
 #define CONF_F_FMT_JSON   0x1 /* JSON formatting */
@@ -108,8 +109,13 @@ extern struct conf conf;
 void RD_NORETURN fatal0 (const char *func, int line,
                                        const char *fmt, ...);
 
+void error0 (int erroronexit, const char *func, int line,
+                                       const char *fmt, ...);
+
 #define FATAL(.../*fmt*/)  fatal0(__FUNCTION__, __LINE__, __VA_ARGS__)
 
+#define ERROR(.../*fmt*/)  error0(conf.exitonerror, __FUNCTION__, __LINE__, __VA_ARGS__)
+
 /* Info printout */
 #define INFO(VERBLVL,.../*fmt*/) do {                    \
                 if (conf.verbosity >= (VERBLVL))     \