Grant Limberg
7 years ago
9 changed files with 249 additions and 17 deletions
@ -0,0 +1,91 @@
|
||||
#include "RabbitMQ.hpp" |
||||
|
||||
#include <amqp.h> |
||||
#include <amqp_tcp_socket.h> |
||||
#include <stdexcept> |
||||
#include <cstring> |
||||
|
||||
namespace ZeroTier |
||||
{ |
||||
|
||||
RabbitMQ::RabbitMQ(MQConfig *cfg, const char *queueName) |
||||
: _mqc(cfg) |
||||
, _qName(queueName) |
||||
, _socket(NULL) |
||||
, _status(0) |
||||
{
|
||||
} |
||||
|
||||
RabbitMQ::~RabbitMQ() |
||||
{ |
||||
amqp_channel_close(_conn, _channel, AMQP_REPLY_SUCCESS); |
||||
amqp_connection_close(_conn, AMQP_REPLY_SUCCESS); |
||||
amqp_destroy_connection(_conn); |
||||
} |
||||
|
||||
void RabbitMQ::init() |
||||
{ |
||||
struct timeval tval; |
||||
memset(&tval, 0, sizeof(struct timeval)); |
||||
tval.tv_sec = 5; |
||||
|
||||
fprintf(stderr, "Initializing RabbitMQ %s\n", _qName); |
||||
_conn = amqp_new_connection(); |
||||
_socket = amqp_tcp_socket_new(_conn); |
||||
if (!_socket) { |
||||
throw std::runtime_error("Can't create socket for RabbitMQ"); |
||||
} |
||||
|
||||
_status = amqp_socket_open_noblock(_socket, _mqc->host, _mqc->port, &tval); |
||||
if (_status) { |
||||
throw std::runtime_error("Can't connect to RabbitMQ"); |
||||
} |
||||
|
||||
amqp_rpc_reply_t r = amqp_login(_conn, "/", 0, 131072, 0, AMQP_SASL_METHOD_PLAIN, |
||||
_mqc->username, _mqc->password); |
||||
if (r.reply_type != AMQP_RESPONSE_NORMAL) { |
||||
throw std::runtime_error("RabbitMQ Login Error"); |
||||
} |
||||
|
||||
static int chan = 0; |
||||
amqp_channel_open(_conn, ++chan); |
||||
r = amqp_get_rpc_reply(_conn); |
||||
if(r.reply_type != AMQP_RESPONSE_NORMAL) { |
||||
throw std::runtime_error("Error opening communication channel"); |
||||
} |
||||
_channel = chan; |
||||
|
||||
_q = amqp_queue_declare(_conn, _channel, amqp_cstring_bytes(_qName), 0, 0, 0, 0, amqp_empty_table); |
||||
r = amqp_get_rpc_reply(_conn); |
||||
if (r.reply_type != AMQP_RESPONSE_NORMAL) { |
||||
throw std::runtime_error("Error declaring queue"); |
||||
} |
||||
|
||||
amqp_basic_consume(_conn, _channel, amqp_cstring_bytes(_qName), amqp_empty_bytes, 0, 1, 0, amqp_empty_table); |
||||
r = amqp_get_rpc_reply(_conn); |
||||
if (r.reply_type != AMQP_RESPONSE_NORMAL) { |
||||
throw std::runtime_error("Error conuming"); |
||||
} |
||||
fprintf(stderr, "RabbitMQ Init OK %s\n", _qName); |
||||
} |
||||
|
||||
std::string RabbitMQ::consume() |
||||
{ |
||||
amqp_rpc_reply_t res; |
||||
amqp_envelope_t envelope; |
||||
amqp_maybe_release_buffers(_conn); |
||||
|
||||
res = amqp_consume_message(_conn, &envelope, NULL, 0); |
||||
if (res.reply_type != AMQP_RESPONSE_NORMAL) { |
||||
throw std::runtime_error("Error getting message"); |
||||
} |
||||
|
||||
std::string msg( |
||||
(const char*)envelope.message.body.bytes, |
||||
envelope.message.body.len |
||||
); |
||||
amqp_destroy_envelope(&envelope); |
||||
return msg; |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,75 @@
|
||||
/*
|
||||
* ZeroTier One - Network Virtualization Everywhere |
||||
* Copyright (C) 2011-2019 ZeroTier, Inc. https://www.zerotier.com/
|
||||
* |
||||
* This program is free software: you can redistribute it and/or modify |
||||
* it under the terms of the GNU General Public License as published by |
||||
* the Free Software Foundation, either version 3 of the License, or |
||||
* (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
* |
||||
* -- |
||||
* |
||||
* You can be released from the requirements of the license by purchasing |
||||
* a commercial license. Buying such a license is mandatory as soon as you |
||||
* develop commercial closed-source software that incorporates or links |
||||
* directly against ZeroTier software without disclosing the source code |
||||
* of your own application. |
||||
*/ |
||||
#ifndef ZT_CONTROLLER_RABBITMQ_HPP |
||||
#define ZT_CONTROLLER_RABBITMQ_HPP |
||||
|
||||
namespace ZeroTier |
||||
{ |
||||
struct MQConfig { |
||||
const char *host; |
||||
int port; |
||||
const char *username; |
||||
const char *password; |
||||
}; |
||||
} |
||||
|
||||
|
||||
#ifdef ZT_CONTROLLER_USE_LIBPQ |
||||
|
||||
#include <amqp.h> |
||||
#include <amqp_tcp_socket.h> |
||||
#include <string> |
||||
|
||||
namespace ZeroTier |
||||
{ |
||||
|
||||
class RabbitMQ { |
||||
public: |
||||
RabbitMQ(MQConfig *cfg, const char *queueName); |
||||
~RabbitMQ(); |
||||
|
||||
void init(); |
||||
|
||||
std::string consume(); |
||||
|
||||
private: |
||||
MQConfig *_mqc; |
||||
const char *_qName; |
||||
|
||||
amqp_socket_t *_socket; |
||||
amqp_connection_state_t _conn; |
||||
amqp_queue_declare_ok_t *_q; |
||||
int _status; |
||||
|
||||
int _channel; |
||||
}; |
||||
|
||||
} |
||||
|
||||
#endif // ZT_CONTROLLER_USE_LIBPQ
|
||||
|
||||
#endif // ZT_CONTROLLER_RABBITMQ_HPP
|
||||
|
||||
Binary file not shown.
Loading…
Reference in new issue