sync code

core/include/services_sdk/resources/metric/metric_scraper.h

@@ -0,0 +1,45 @@
+// Copyright (C) 2022 Check Point Software Technologies Ltd. All rights reserved.
+
+// Licensed under the Apache License, Version 2.0 (the "License");
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#ifndef __METRIC_SCRAPER_H__
+#define __METRIC_SCRAPER_H__
+
+#include <string>
+#include <fstream>
+#include <vector>
+#include <streambuf>
+
+#include "singleton.h"
+#include "debug.h"
+#include "component.h"
+#include "event.h"
+#include "i_rest_api.h"
+#include "generic_metric.h"
+
+class MetricScraper
+        :
+    public Component,
+    Singleton::Consume<I_RestApi>
+{
+public:
+    MetricScraper();
+    ~MetricScraper();
+
+    void init();
+
+private:
+    class Impl;
+    std::unique_ptr<Impl> pimpl;
+};
+
+#endif // __METRIC_SCRAPER_H__

core/metric/metric_scraper.cc

@@ -0,0 +1,50 @@
+#include "metric/metric_scraper.h"
+
+using namespace std;
+
+USE_DEBUG_FLAG(D_METRICS);
+
+class MetricScraper::Impl
+{
+public:
+    void
+    init()
+    {
+        Singleton::Consume<I_RestApi>::by<MetricScraper>()->addGetCall(
+            "service-metrics",
+            [&] () { return getAllPrometheusMetrics(); }
+        );
+    }
+
+    string
+    getAllPrometheusMetrics()
+    {
+        auto all_metrics_events_res = MetricScrapeEvent().query();
+        for (auto metric_vec : all_metrics_events_res) {
+            for (PrometheusData metric : metric_vec) {
+                metric.label = "{" + metric.label + "}";
+                all_metrics.emplace_back(metric);
+            }
+        }
+        stringstream ss;
+        {
+            cereal::JSONOutputArchive archive(ss);
+            archive(cereal::make_nvp("metrics", all_metrics));
+        }
+        all_metrics.clear();
+        return ss.str();
+    }
+
+private:
+    vector<PrometheusData> all_metrics;
+};
+
+MetricScraper::MetricScraper() : Component("MetricScraper"), pimpl(make_unique<MetricScraper::Impl>()) {}
+
+MetricScraper::~MetricScraper() {}
+
+void
+MetricScraper::init()
+{
+    pimpl->init();
+}