1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
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;
}
|