Command line AMQP tools based on rabbitmq-c

1 files changed, 90 insertions, 0 deletions

diff --git a/tools/publish.c b/tools/publish.c
new file mode 100644
index 0000000..08a78b7
--- /dev/null
+++ b/tools/publish.c
@@ -0,0 +1,90 @@
+#include "config.h"
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include <popt.h>
+
+#include "common.h"
+
+static void do_publish(amqp_connection_state_t conn,
+                       char *exchange, char *routing_key,
+		       amqp_basic_properties_t *props, amqp_bytes_t body)
+{
+	int res = amqp_basic_publish(conn, 1,
+				     cstring_bytes(exchange),
+				     cstring_bytes(routing_key),
+				     0, 0, props, body);
+	if (res != 0)
+		die_errno(-res, "basic.publish");
+}
+
+int main(int argc, const char **argv)
+{
+	amqp_connection_state_t conn;
+	char *exchange = NULL;
+	char *routing_key = NULL;
+	char *content_type = NULL;
+	char *content_encoding = NULL;
+	char *body = NULL;
+	amqp_basic_properties_t props;
+	amqp_bytes_t body_bytes;
+	int delivery = 1; /* non-persistent by default */
+
+	struct poptOption options[] = {
+		INCLUDE_OPTIONS(connect_options),
+		{"exchange", 'e', POPT_ARG_STRING, &exchange, 0,
+		 "the exchange to publish to", "exchange"},
+		{"routing-key", 'r', POPT_ARG_STRING, &routing_key, 0,
+		 "the routing key to publish with", "routing key"},
+		{"persistent", 'p', POPT_ARG_VAL, &delivery, 2,
+		 "use the persistent delivery mode", NULL},
+		{"content-type", 'C', POPT_ARG_STRING, &content_type, 0,
+		 "the content-type for the message", "content type"},
+		{"content-encoding", 'E', POPT_ARG_STRING,
+		 &content_encoding, 0,
+		 "the content-encoding for the message", "content encoding"},
+		{"body", 'b', POPT_ARG_STRING, &body, 0,          
+                 "specify the message body", "body"}, 
+		POPT_AUTOHELP
+		{ NULL, 0, 0, NULL, 0 }
+	};
+
+	process_all_options(argc, argv, options);
+	
+	if (!exchange && !routing_key) {
+		fprintf(stderr,
+			"neither exchange nor routing key specified\n");
+		return 1;
+	}
+
+	memset(&props, 0, sizeof props);
+	props._flags = AMQP_BASIC_DELIVERY_MODE_FLAG;
+	props.delivery_mode = 2; // persistent delivery mode
+
+	if (content_type) {
+		props._flags |= AMQP_BASIC_CONTENT_TYPE_FLAG;
+		props.content_type = amqp_cstring_bytes(content_type);
+	}
+
+	if (content_encoding) {
+		props._flags |= AMQP_BASIC_CONTENT_ENCODING_FLAG;
+		props.content_encoding = amqp_cstring_bytes(content_encoding);
+	}
+
+	conn = make_connection();
+
+	if (body)
+		body_bytes = amqp_cstring_bytes(body);
+	else
+		body_bytes = read_all(0);
+	
+	do_publish(conn, exchange, routing_key, &props, body_bytes);
+
+	if (!body)
+		free(body_bytes.bytes);
+	
+	close_connection(conn);
+	return 0;
+}