diff options
Diffstat (limited to 'tools/get.c')
| -rw-r--r-- | tools/get.c | 16 |
1 files changed, 11 insertions, 5 deletions
diff --git a/tools/get.c b/tools/get.c index b433e2b..f746fd1 100644 --- a/tools/get.c +++ b/tools/get.c @@ -55,12 +55,11 @@ #include <popt.h> #include "common.h" -#include "common_consume.h" -static int do_get(amqp_connection_state_t conn) +static int do_get(amqp_connection_state_t conn, char *queue) { amqp_rpc_reply_t r - = amqp_basic_get(conn, 1, setup_queue(conn), 1); + = amqp_basic_get(conn, 1, cstring_bytes(queue), 1); die_rpc(r, "basic.get"); if (r.reply.id == AMQP_BASIC_GET_EMPTY_METHOD) @@ -73,19 +72,26 @@ static int do_get(amqp_connection_state_t conn) int main(int argc, const char **argv) { amqp_connection_state_t conn; + char *queue = NULL; int got_something; struct poptOption options[] = { INCLUDE_OPTIONS(connect_options), - INCLUDE_OPTIONS(consume_queue_options), + {"queue", 'q', POPT_ARG_STRING, &queue, 0, + "the queue to consume from", "queue"}, POPT_AUTOHELP { NULL, 0, 0, NULL, 0 } }; process_all_options(argc, argv, options); + + if (!queue) { + fprintf(stderr, "queue not specified\n"); + return 1; + } conn = make_connection(); - got_something = do_get(conn); + got_something = do_get(conn, queue); close_connection(conn); return got_something ? 0 : 2; } |