Clickhouse - add common

Author: sedmicha

extra_plugins/output/clickhouse/src/common.cpp

@@ -0,0 +1,74 @@
+/**
+ * @file
+ * @author Michal Sedlak <[email protected]>
+ * @brief Common functionality and general utilities
+ * @date 2024
+ *
+ * Copyright(c) 2024 CESNET z.s.p.o.
+ * SPDX-License-Identifier: BSD-3-Clause
+ */
+
+#include "common.h"
+
+#include <cctype>
+#include <iostream>
+
+std::string to_lower(std::string str)
+{
+    for (auto &c : str) {
+        c = std::tolower(c);
+    }
+    return str;
+}
+
+void debug_print_block(const clickhouse::Block& block) {
+    std::cout << "================================================================================\n";
+    for (size_t i = 0; i < block.GetColumnCount(); i++) {
+        std::cout << block.GetColumnName(i) << ":"
+                  << block[i]->GetType().GetName()
+                  << (i < block.GetColumnCount() - 1 ? '\t' : '\n');
+    }
+    std::cout << "--------------------------------------------------------------------------------\n";
+
+    for (size_t i = 0; i < block.GetRowCount(); i++) {
+        for (size_t j = 0; j < block.GetColumnCount(); j++) {
+            switch (block[j]->GetType().GetCode()) {
+            case clickhouse::Type::Int8:
+                std::cout << +block[j]->As<clickhouse::ColumnInt8>()->At(i);
+                break;
+            case clickhouse::Type::Int16:
+                std::cout << block[j]->As<clickhouse::ColumnInt16>()->At(i);
+                break;
+            case clickhouse::Type::Int32:
+                std::cout << block[j]->As<clickhouse::ColumnInt32>()->At(i);
+                break;
+            case clickhouse::Type::Int64:
+                std::cout << block[j]->As<clickhouse::ColumnInt64>()->At(i);
+                break;
+            case clickhouse::Type::Int128:
+                std::cout << block[j]->As<clickhouse::ColumnInt128>()->At(i);
+                break;
+            case clickhouse::Type::UInt8:
+                std::cout << +block[j]->As<clickhouse::ColumnUInt8>()->At(i);
+                break;
+            case clickhouse::Type::UInt16:
+                std::cout << block[j]->As<clickhouse::ColumnUInt16>()->At(i);
+                break;
+            case clickhouse::Type::UInt32:
+                std::cout << block[j]->As<clickhouse::ColumnUInt32>()->At(i);
+                break;
+            case clickhouse::Type::UInt64:
+                std::cout << block[j]->As<clickhouse::ColumnUInt64>()->At(i);
+                break;
+            case clickhouse::Type::String:
+                std::cout << block[j]->As<clickhouse::ColumnString>()->At(i);
+                break;
+            default:
+                std::cout << "-";
+                break;
+            }
+            std::cout << (j < block.GetColumnCount() - 1 ? '\t' : '\n');
+        }
+    }
+    std::cout << "================================================================================\n\n";
+};

extra_plugins/output/clickhouse/src/common.h

@@ -0,0 +1,76 @@
+/**
+ * @file
+ * @author Michal Sedlak <[email protected]>
+ * @brief Common functionality and general utilities
+ * @date 2024
+ *
+ * Copyright(c) 2024 CESNET z.s.p.o.
+ * SPDX-License-Identifier: BSD-3-Clause
+ */
+#pragma once
+
+#include "clickhouse.h"
+
+#include <fmt/core.h>
+#include <ipfixcol2.h>
+
+#include <stdexcept>
+
+/**
+ * @class Nonmoveable
+ * @brief Classes that will inherit from this class will be immovable
+ */
+struct Nonmoveable {
+    Nonmoveable() = default;
+    Nonmoveable(Nonmoveable&&) = delete;
+    Nonmoveable& operator=(Nonmoveable&&) = delete;
+};
+
+/**
+ * @class Noncopyable
+ * @brief Classes that will inherit from this class will not be able to be copied
+ */
+struct Noncopyable {
+    Noncopyable() = default;
+    Noncopyable(const Noncopyable&) = delete;
+    Noncopyable& operator=(const Noncopyable&) = delete;
+};
+
+/**
+ * @class Error
+ * @brief An exception type that will be thrown by the plugin
+ */
+class Error : public std::runtime_error {
+public:
+    template <typename ...Args>
+    Error(Args&& ...args) : std::runtime_error(fmt::format(args...)) {}
+};
+
+
+
+class Logger {
+public:
+    Logger(const ipx_ctx_t *ctx) : m_ctx(ctx) {}
+    template <typename ...Args> void info   (const char *fmt, Args... args) { IPX_CTX_INFO   (m_ctx, fmt, args..., 0); }
+    template <typename ...Args> void warning(const char *fmt, Args... args) { IPX_CTX_WARNING(m_ctx, fmt, args..., 0); }
+    template <typename ...Args> void error  (const char *fmt, Args... args) { IPX_CTX_ERROR  (m_ctx, fmt, args..., 0); }
+    template <typename ...Args> void debug  (const char *fmt, Args... args) { IPX_CTX_DEBUG  (m_ctx, fmt, args..., 0); }
+
+private:
+    const ipx_ctx_t *m_ctx;
+};
+
+
+/**
+ * @brief Convert string to lowercase
+ *
+ * @param str The string
+ */
+std::string to_lower(std::string s);
+
+/**
+ * @brief Debug function for printing contents of a block
+ *
+ * @param block The block
+ */
+void debug_print_block(const clickhouse::Block& block);