eis/eqpalg/threads/manager.h

#pragma once
/**
 * @file manager.h
 * @brief 线程管理
 * @author your name (you@domain.com)
 * @version 0.1
 * @date 2024-03-28
 *
 * Copyright: Baosight Co. Ltd.
 * DO NOT COPY/USE WITHOUT PERMISSION
 *
 */
#include <eqpalg/build_algorithm.h>
#include <eqpalg/threads/handler_exec.h>
#include <eqpalg/utility/alarm_handler.hpp>
#include <memory>
#include <shared_mutex>
#include <string>
#include <thread>
#include <utility>
namespace threads {
/**
 * @brief 线程句柄管理器
 */
class Manager {
public:
  /**
   * @brief Construct a new Manager object
   */
  Manager();

  /**
   * @brief Destroy the Manager object
   * 并销毁其所管理的每一个线程实例
   */
  ~Manager();

protected:
  std::map<std::string, std::tuple<int, std::string, mix_cc::json, bool, double,
                                   double, int>>
      stored_cfg_data_; ///< 存储的配置信息，用于初始化单次执行信息和重新载入算法时组织信息

  std::vector<std::tuple<std::thread::id, std::string>>
      tid_2_thread_name_; ///< 进程id到进程名的关系

  bool is_running_;                         ///< 是否正在运行
  bool is_start_;                           ///< 是否调用了start
  std::unique_ptr<LOG> logger_;             ///< 本地logger
  std::unique_ptr<LOG> alarm_logger_;       ///< 本地alarm_logger_
  map<string, string> rule_alg_id_mapping_; ///< id和规则的对应关系
  std::shared_mutex handles_mutex;          // 针对handles_的互斥锁
  map<std::string, std::unique_ptr<HandlerExec>>
      handles_; ///< 监控程序的handles，每个alg_id_datasource_seq一个handle

  std::unique_ptr<std::thread> r_thread_; ///< 检测工厂运行状态的线程

  std::unique_ptr<AlarmHandler> alarm_handler_ptr_;

  int thread_num_ = 0; ///<规则 线程数

public:
  /**
   * @brief 保持参数信息到数据集合中
   * @param  ruleId           规则id
   * @param  alg_id           算法id
   * @param  name             规则名
   * @param  rule_json        规则配置参数
   * @param  usable           规则启停标记
   * @param  padding_low      区间下 弃用
   * @param  padding_up       区间上 弃用
   * @param  task_seq         线程序号
   * @return int              0-正常；其它-异常
   */
  int storage(const string &ruleId, int alg_id, const string &name,
              const mix_cc::json &rule_json, bool usable, double padding_low,
              double padding_up, int task_seq);

  /**
   * @brief 程序启动时，载入数据所使用的方法
   * @param  ruleId           算法实例id
   * @return int
   */
  int load(const string &ruleId);

  /**
   * @brief 启动载入的算法所对应的线程 mon
   * @return int
   */
  int start();

  /**
   * @brief 在算法运行过程中，把对应的数据添加到具体线程中
   * @param  ruleId           算法实例id
   * @return int
   */
  int attach(const string &ruleId);
  /**
   * @brief 重置统计
   *    主要用于 6-累计时间
   *  7-出现次数
   * @param  ruleId           My Param doc
   * @return int
   */
  int reset(const string &ruleId);

  /**
   * @brief 取消对应规则
   * @param  ruleId           要取消的算法实例id
   * @return int
   */
  int detach(const string &ruleId);

  /**
   * @brief 单次执行
   * @param  ruleId           算法实例（规则）id
   * @param  time_start       单次执行的开始时间
   * @param  time_end         单次执行的结束时间
   * @return int
   */
  int exec_task(std::string ruleId, TimePoint time_start, TimePoint time_end);

  /**
   * @brief 启用算法/停用算法
   * 应该考虑两个方面：1.当前运行线程中的算法启停用；2.stored_cfg_data_的维护，以便停机开机后，保持算法的启停用状态
   * @param  ruleId           算法id
   * @param  usable           是否停用/启用
   * @return int
   */
  int enable(std::string ruleId, bool usable);

  /**
   * @brief 删除存储的算法实例信息
   * @param  ruleId           My Param doc
   * @return int
   */
  int delete_instance(std::string ruleId);

protected:
  /**
   * @brief  把数据实例化成算法指针，并完善对应关系
   * @param  ruleId           算法实例id
   * @return int
   */
  int build_alg_to_handle(const string &ruleId);

  /**
   * @brief 完全抹除某个算法线程
   * @param  alg_thread_name  算法线程名
   * @return int
   */
  int detach_full_alg(std::string alg_thread_name);

public:
  /**
   * @brief
   * @param  ruleid      规则信息
   * @param  lb          下限
   * @param  ub          上限
   * @param  va          当前值
   * @param stime        开始时间
   * @param etime        结束时间
   */
  void update_limit_alarm(std::string ruleid, double lb, double ub, double va,
                          int64_t stime, int64_t etime);

  /**
   * @brief Get the thread size object
   * @return int
   */
  int get_thread_size();
};

} // namespace threads