osquery-1/osquery/tables/applications/posix/prometheus_metrics.h

/*
 *  Copyright (c) 2014-present, Facebook, Inc.
 *  All rights reserved.
 *
 *  This source code is licensed under the BSD-style license found in the
 *  LICENSE file in the root directory of this source tree. An additional grant
 *  of patent rights can be found in the PATENTS file in the same directory.
 *
 */
#pragma once

#include <chrono>
#include <map>
#include <string>
#include <vector>

#include <boost/network/protocol/http/client.hpp>

#include <osquery/tables.h>

namespace osquery {
namespace tables {

const std::string kColTargetName = "target_name";
const std::string kColMetric = "metric_name";
const std::string kColValue = "metric_value";
const std::string kColTimeStamp = "timestamp_ms";

struct PrometheusResponseData {
  std::string content;
  std::chrono::milliseconds timestampMS;
};

/**
 * @brief parse raw payload returned by scraped targets into QueryData.
 *
 * @param scrapeResults map where the key is the target url scraped and
 * value is the struct PrometheusResponseData of the corresponding target.
 *
 */
void parseScrapeResults(
    const std::map<std::string, PrometheusResponseData>& scrapeResults,
    QueryData& rows);

/**
 * @brief Scrapes the Prometheus targets and returns response payload and
 * timestamp.
 *
 * @param scrapeResults map where the key is the target url to be scraped and
 * value is the struct PrometheusResponseData where payload and timestamp are to
 * be written to.
 */
void scrapeTargets(
    std::map<std::string, PrometheusResponseData>& scrapeResults);
}
}