files.h

Go to the documentation of this file.

/* SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
 *
 * Copyright (c) 2022, 郑昱笙，濮雯旭，张典典（牛校牛子队）
 * All rights reserved.
 */
 
#ifndef FILE_CMD_H
#define FILE_CMD_H
 
#include <iostream>
#include <mutex>
#include <string>
#include <thread>
 
#include "libbpf_print.h"
#include "model/tracker.h"
#include "prometheus/counter.h"
#include "prometheus_server.h"
extern "C"
{
#include <files/file_tracker.h>
}
 
 
struct files_tracker : public tracker_with_config<files_env, files_event>
{
  files_tracker(config_data config);
 
  static std::unique_ptr<files_tracker> create_tracker_with_default_env(tracker_event_handler handler);
  static std::unique_ptr<files_tracker> create_tracker_with_args(
      tracker_event_handler handler,
      const std::vector<std::string> &args)
  {
    return create_tracker_with_default_env(handler);
  }
 
  void start_tracker();
 
  struct prometheus_event_handler : public event_handler<files_event>
  {
    prometheus::Family<prometheus::Counter> &eunomia_files_read_counter;
    prometheus::Family<prometheus::Counter> &eunomia_files_write_counter;
    prometheus::Family<prometheus::Counter> &eunomia_files_write_bytes;
    prometheus::Family<prometheus::Counter> &eunomia_files_read_bytes;
    const container_manager &container_manager_ref;
    void report_prometheus_event(const struct files_event &e);
 
    prometheus_event_handler(prometheus_server &server);
    void handle(tracker_event<files_event> &e);
  };
 
  struct json_event_handler : public event_handler<files_event>
  {
    std::string to_json(const struct files_event &e);
  };
 
  struct json_event_printer : public json_event_handler
  {
    void handle(tracker_event<files_event> &e);
  };
 
  struct plain_text_event_printer : public event_handler<files_event>
  {
    void handle(tracker_event<files_event> &e);
  };
 
  struct csv_event_printer : public event_handler<files_event>
  {
    void handle(tracker_event<files_event> &e);
  };
};
 
#endif