src/MQTTCommand.cpp
branchv_0
changeset 2 0799eaf338b9
parent 1 4993a084b8ba
child 3 610783d70ae9
--- a/src/MQTTCommand.cpp	Fri May 06 23:09:56 2022 +0200
+++ b/src/MQTTCommand.cpp	Sun May 08 21:42:14 2022 +0200
@@ -23,6 +23,7 @@
 #include <unistd.h>
 #include <sstream>
 #include <iomanip>
+#include <random>
 
 #include <mosquittopp.h>
 
@@ -37,7 +38,6 @@
 #include "MQTTCommand.h"
 #include "Hex.h"
 
-using namespace std;
 using namespace relpipe::cli;
 using namespace relpipe::writer;
 
@@ -45,31 +45,79 @@
 namespace in {
 namespace mqtt {
 
-void MQTTCommand::process(std::shared_ptr<writer::RelationalWriter> writer, Configuration& configuration) {
-	vector<AttributeMetadata> metadata;
+class MQTTClient : public mosqpp::mosquittopp {
+private:
+	std::shared_ptr<writer::RelationalWriter> writer;
+	Configuration& configuration;
+	int messageCount = 0;
+	std::string clientId;
 
-	{
-		// TODO: remove
-		int major, minor, patch;
-		mosqpp::lib_version(&major, &minor, &patch);
-		std::cerr << "mosquitto version: " << major << "." << minor << "." << patch << std::endl;
+	/**
+	 * @return unique (random) client ID for MQTT to allow multiple simultaneous connections
+	 */
+	static std::string generateClientID() {
+		std::stringstream result;
+		// result << "relpipe-in-";
+		std::string symbols("0123456789abcdef");
+
+		std::random_device dev;
+		std::mt19937 rng(dev());
+		std::uniform_int_distribution<std::mt19937::result_type> dist6(0, symbols.size());
+
+		for (int i = 0; i < 8; i++) result << symbols[dist6(rng)];
+
+		// std::cerr << "generated clien ID = " << result.str() << std::endl;
+		return result.str();
+	}
+public:
+
+	MQTTClient(std::shared_ptr<writer::RelationalWriter> writer, Configuration& configuration) : mosqpp::mosquittopp((generateClientID()).c_str()), writer(writer), configuration(configuration) {
 	}
 
+	void on_message(const mosquitto_message* message) override {
+		// std::cerr << "got MQTT message: length=" << message->payloadlen << std::endl;
+		std::string payload = std::string((const char*) message->payload, message->payloadlen);
+		writer->writeAttribute(configuration.stream);
+		writer->writeAttribute(Hex::toTxt(payload));
+		writer->writeAttribute(Hex::toHex(payload));
+		messageCount++;
+	}
+
+	int popMessageCount() {
+		int count = messageCount;
+		messageCount = 0;
+		return count;
+	}
+
+};
+
+void MQTTCommand::process(std::shared_ptr<writer::RelationalWriter> writer, Configuration& configuration) {
+	std::shared_ptr<MQTTClient> mq = std::make_shared<MQTTClient>(writer, configuration);
+
 	writer->startRelation(configuration.relation,{
-		{L"queue", TypeId::STRING},
+		{L"stream", TypeId::STRING},
 		{L"text", TypeId::STRING},
 		{L"data", TypeId::STRING}
 	}, true);
 
-	for (int i = configuration.messageCount; continueProcessing && i > 0; i--) {
-		// TODO: maybe rather call mq_timedreceive() inside and check continueProcessing (to be able to stop even when no messages are comming)
-		std::string message = "TODO"; // FIXME: receive message
+	mq->max_inflight_messages_set(1);
+	mq->connect("localhost", 1883);
+	int mid;
+	mq->subscribe(&mid, convertor.to_bytes(configuration.stream).c_str());
 
-		writer->writeAttribute(configuration.queue);
-		writer->writeAttribute(Hex::toTxt(message));
-		writer->writeAttribute(Hex::toHex(message));
+	//for (int i = configuration.messageCount; continueProcessing && i > 0; i--) {
+	for (int i = configuration.messageCount; continueProcessing && i > 0; i = i - mq->popMessageCount()) {
+		// std::cerr << "loop(): i=" << i << std::endl;
+
+		//mq->loop();
+		mq->loop(1000, 1);
+		//mq->loop(1000, -1);
+		//mq->loop_forever();
+		//mq->loop_write();
 	}
 
+	// FIXME: move do destructor
+	mq->disconnect();
 }
 
 MQTTCommand::~MQTTCommand() {