WIF - Reporters - Introduce unirec reporter

TG-35

Author: plnyrich

include/wif/reporters/unirecReporter.hpp

@@ -0,0 +1,64 @@
+/**
+ * @file
+ * @author Richard Plny <[email protected]>
+ * @brief Unirec reporter interface
+ *
+ * SPDX-License-Identifier: BSD-3-Clause
+ */
+
+#pragma once
+
+#include "wif/reporters/reporter.hpp"
+
+#include <unirec++/unirec.hpp>
+
+namespace WIF {
+
+/**
+ * @brief Unirec Reporter class
+ *
+ * Reporter specialization which sends data in UniRec format to Trap interface
+ */
+class UnirecReporter : public Reporter {
+public:
+	/**
+	 * @brief Construct a new Unirec Reporter object
+	 *
+	 * @param outputInterface output interface where data will be sent
+	 */
+	UnirecReporter(Nemea::UnirecOutputInterface& outputInterface);
+
+	/**
+	 * @brief Update UniRec field IDs
+	 * Should be called after UniRec template was changed
+	 *
+	 * @param unirecFieldIDs
+	 */
+	void updateUnirecFieldIDs(const std::vector<ur_field_id_t>& unirecFieldIDs);
+
+	/*! @copydoc Reporter::onRecordStart()
+	 */
+	virtual void onRecordStart() override;
+
+	/*! @copydoc Reporter::report(const DataVariant& data)
+	 * Data is stored to the UniRec record to the next ID
+	 */
+	void report(const DataVariant& data) override;
+
+	/*! @copydoc Reporter::onRecordEnd()
+	 * Current Unirec Record is sent to Unirec Output Interface
+	 */
+	virtual void onRecordEnd() override;
+
+	/*! @copydoc Reporter::flush()
+	 * Internally calls Nemea::UnirecOutputInterface::sendFlush()
+	 */
+	virtual void flush() override;
+
+private:
+	unsigned m_currentUnirecID = 0;
+	std::vector<ur_field_id_t> m_unirecFieldIDs;
+	Nemea::UnirecOutputInterface& m_outputInterface;
+};
+
+} // namespace WIF

src/wif/CMakeLists.txt

@@ -14,6 +14,7 @@ set(LIBWIF_SOURCES
 
 if(BUILD_WITH_UNIREC)
 	list(APPEND LIBWIF_SOURCES
+		reporters/unirecReporter.cpp
 	)
 endif()
 

src/wif/reporters/unirecReporter.cpp

@@ -0,0 +1,71 @@
+/**
+ * @file
+ * @author Richard Plny <[email protected]>
+ * @brief Unirec reporter implementation
+ *
+ * SPDX-License-Identifier: BSD-3-Clause
+ */
+
+#include "wif/reporters/unirecReporter.hpp"
+
+namespace WIF {
+
+UnirecReporter::UnirecReporter(Nemea::UnirecOutputInterface& outputInterface)
+	: m_outputInterface(outputInterface)
+{
+}
+
+void UnirecReporter::updateUnirecFieldIDs(const std::vector<ur_field_id_t>& unirecFieldIDs)
+{
+	m_unirecFieldIDs = unirecFieldIDs;
+}
+
+void UnirecReporter::onRecordStart()
+{
+	m_currentUnirecID = 0;
+}
+
+void UnirecReporter::report(const DataVariant& data)
+{
+	ur_field_id_t fieldID = m_unirecFieldIDs[m_currentUnirecID++];
+	auto& record = m_outputInterface.getUnirecRecord();
+
+	if (std::holds_alternative<uint8_t>(data)) {
+		auto value = std::get<uint8_t>(data);
+		record.setFieldFromType<uint8_t>(value, fieldID);
+	} else if (std::holds_alternative<uint16_t>(data)) {
+		auto value = std::get<uint16_t>(data);
+		record.setFieldFromType<uint16_t>(value, fieldID);
+	} else if (std::holds_alternative<uint32_t>(data)) {
+		auto value = std::get<uint32_t>(data);
+		record.setFieldFromType<uint32_t>(value, fieldID);
+	} else if (std::holds_alternative<uint64_t>(data)) {
+		auto value = std::get<uint64_t>(data);
+		record.setFieldFromType<uint64_t>(value, fieldID);
+	} else if (std::holds_alternative<double>(data)) {
+		auto value = std::get<double>(data);
+		record.setFieldFromType<double>(value, fieldID);
+	} else if (std::holds_alternative<std::string>(data)) {
+		auto value = std::get<std::string>(data);
+		record.setFieldFromType<std::string>(value, fieldID);
+	} else if (std::holds_alternative<IpAddress>(data)) {
+		auto value = std::get<IpAddress>(data);
+		Nemea::IpAddress nemeaIp(value.toString());
+		record.setFieldFromType<Nemea::IpAddress>(nemeaIp, fieldID);
+	} else if (std::holds_alternative<std::vector<double>>(data)) {
+		auto value = std::get<std::vector<double>>(data);
+		record.setFieldFromVector<double>(value, fieldID);
+	}
+}
+
+void UnirecReporter::onRecordEnd()
+{
+	m_outputInterface.send(m_outputInterface.getUnirecRecord());
+}
+
+void UnirecReporter::flush()
+{
+	m_outputInterface.sendFlush();
+}
+
+} // namespace WIF