Skip to content

Commit

Permalink
first cut on client and server
Browse files Browse the repository at this point in the history
  • Loading branch information
RichardChambers committed Jan 1, 2018
1 parent 10e0d8f commit 60153a3
Show file tree
Hide file tree
Showing 6 changed files with 719 additions and 0 deletions.
180 changes: 180 additions & 0 deletions amqp/client.c
Original file line number Diff line number Diff line change
@@ -0,0 +1,180 @@

/*
* amqp_producer.c from
* https://github.com/alanxz/rabbitmq-c/blob/master/examples/amqp_producer.c
*
* This is an example of a AMQP client which requests a single URI, hello,
* which is provided by a sample CoAP server also in the same github folder.
*
* Starting with that source code, I am embellishing with more comments and
* some additional functionality to explore the AMQP protocol.
*
* To compile you must first have the RabbitMQ AMQP library installed using:
* sudo apt-get install librabbitmq-dev
*
* You will also need the FastJSON library for parsing JSON text libfastjson:
* sudo apt-get install libfastjson-dev
*
* You will need to specify the include path and you will need to specify
* the define -DWITH_POSIX on the compiler command line.
*
* cc -DWITH_POSIX -o client client.c -lrabbitmq -lfastjson
*/


#include <stdio.h>

// header for the Fast JSON library. should be in /usr/include/libfastjson
#include <libfastjson/json.h>

/*
* ***** BEGIN LICENSE BLOCK *****
* Version: MIT
*
* Portions created by Alan Antonuk are Copyright (c) 2012-2013
* Alan Antonuk. All Rights Reserved.
*
* Portions created by VMware are Copyright (c) 2007-2012 VMware, Inc.
* All Rights Reserved.
*
* Portions created by Tony Garnock-Jones are Copyright (c) 2009-2010
* VMware, Inc. and Tony Garnock-Jones. All Rights Reserved.
*
* Permission is hereby granted, free of charge, to any person
* obtaining a copy of this software and associated documentation
* files (the "Software"), to deal in the Software without
* restriction, including without limitation the rights to use, copy,
* modify, merge, publish, distribute, sublicense, and/or sell copies
* of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
* BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
* ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
* ***** END LICENSE BLOCK *****
*/

#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include <amqp.h>
#include <amqp_tcp_socket.h>

#include "utils.h"

#define SUMMARY_EVERY_US 1000000

static void send_batch(amqp_connection_state_t conn, char const *queue_name,
int rate_limit, int message_count) {
uint64_t start_time = now_microseconds();
int i;
int sent = 0;
int previous_sent = 0;
uint64_t previous_report_time = start_time;
uint64_t next_summary_time = start_time + SUMMARY_EVERY_US;

char message[256];
amqp_bytes_t message_bytes;

for (i = 0; i < (int)sizeof(message); i++) {
message[i] = i & 0xff;
}

message_bytes.len = sizeof(message);
message_bytes.bytes = message;

for (i = 0; i < message_count; i++) {
uint64_t now = now_microseconds();

die_on_error(amqp_basic_publish(conn, 1, amqp_cstring_bytes("amq.direct"),
amqp_cstring_bytes(queue_name), 0, 0, NULL,
message_bytes),
"Publishing");
sent++;
if (now > next_summary_time) {
int countOverInterval = sent - previous_sent;
double intervalRate =
countOverInterval / ((now - previous_report_time) / 1000000.0);
printf("%d ms: Sent %d - %d since last report (%d Hz)\n",
(int)(now - start_time) / 1000, sent, countOverInterval,
(int)intervalRate);

previous_sent = sent;
previous_report_time = now;
next_summary_time += SUMMARY_EVERY_US;
}

while (((i * 1000000.0) / (now - start_time)) > rate_limit) {
microsleep(2000);
now = now_microseconds();
}
}

{
uint64_t stop_time = now_microseconds();
int total_delta = (int)(stop_time - start_time);

printf("PRODUCER - Message count: %d\n", message_count);
printf("Total time, milliseconds: %d\n", total_delta / 1000);
printf("Overall messages-per-second: %g\n",
(message_count / (total_delta / 1000000.0)));
}
}

int main(int argc, char const *const *argv) {
char const *hostname;
int port, status;
int rate_limit;
int message_count;
amqp_socket_t *socket = NULL;
amqp_connection_state_t conn;

if (argc < 5) {
fprintf(stderr,
"Usage: amqp_producer host port rate_limit message_count\n");
return 1;
}

hostname = argv[1];
port = atoi(argv[2]);
rate_limit = atoi(argv[3]);
message_count = atoi(argv[4]);

conn = amqp_new_connection();

socket = amqp_tcp_socket_new(conn);
if (!socket) {
die("creating TCP socket");
}

status = amqp_socket_open(socket, hostname, port);
if (status) {
die("opening TCP socket");
}

die_on_amqp_error(amqp_login(conn, "/", 0, 131072, 0, AMQP_SASL_METHOD_PLAIN,
"pi", "raspberry"),
"Logging in");
amqp_channel_open(conn, 1);
die_on_amqp_error(amqp_get_rpc_reply(conn), "Opening channel");

send_batch(conn, "test queue", rate_limit, message_count);

die_on_amqp_error(amqp_channel_close(conn, 1, AMQP_REPLY_SUCCESS),
"Closing channel");
die_on_amqp_error(amqp_connection_close(conn, AMQP_REPLY_SUCCESS),
"Closing connection");
die_on_error(amqp_destroy_connection(conn), "Ending connection");
return 0;
}

16 changes: 16 additions & 0 deletions amqp/makefile
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
# this make file uses two different libraries for the build.
# libmosquitto is the mosquitto MQTT library
# use sudo apt-get install libmosquitto-dev
# use sudo apt-get install mosquitto mosquitto-clients

# libfastjson is the Fast JSON library for JSON text handling
# use sudo apt-get install libfastjson-dev

CFLAGS=-DWITH_POSIX

client: client.c
cc $(CFLAGS) -o client client.c utils.c platform_utils.c -lrabbitmq -lfastjson -lsqlite3

server: server.c
cc $(CFLAGS) -o server server.c utils.c platform_utils.c -lrabbitmq -lfastjson -lpthread -lwiringPi

53 changes: 53 additions & 0 deletions amqp/platform_utils.c
Original file line number Diff line number Diff line change
@@ -0,0 +1,53 @@
/*
* ***** BEGIN LICENSE BLOCK *****
* Version: MIT
*
* Portions created by Alan Antonuk are Copyright (c) 2012-2013
* Alan Antonuk. All Rights Reserved.
*
* Portions created by VMware are Copyright (c) 2007-2012 VMware, Inc.
* All Rights Reserved.
*
* Portions created by Tony Garnock-Jones are Copyright (c) 2009-2010
* VMware, Inc. and Tony Garnock-Jones. All Rights Reserved.
*
* Permission is hereby granted, free of charge, to any person
* obtaining a copy of this software and associated documentation
* files (the "Software"), to deal in the Software without
* restriction, including without limitation the rights to use, copy,
* modify, merge, publish, distribute, sublicense, and/or sell copies
* of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
* BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
* ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
* ***** END LICENSE BLOCK *****
*/

#include <stdint.h>
#include <sys/time.h>
#include <time.h>
#include <unistd.h>

uint64_t now_microseconds(void) {
struct timeval tv;
gettimeofday(&tv, NULL);
return (uint64_t)tv.tv_sec * 1000000 + (uint64_t)tv.tv_usec;
}

void microsleep(int usec) {
struct timespec req;
req.tv_sec = 0;
req.tv_nsec = 1000 * usec;
nanosleep(&req, NULL);
}

Loading

0 comments on commit 60153a3

Please sign in to comment.