#include <dsm/dsm_alg.h>
#include <dsm/utility.h>
DsmAlg::DsmAlg() {
  logger_ = std::make_unique<LOG>("DsmAlg");
  ruleid_norminfo_ =
      std::make_unique<CMemMap<std::string, DataInfo>>(DsmMapConfig::TableName);
}
DsmAlg::~DsmAlg() {}
void DsmAlg::dsm_handle(std::string ruleid) {
  auto value = ruleid_norminfo_->operator[](ruleid);
  logger_->Debug() << "map size:" << ruleid_norminfo_->size() << std::endl;
  if (value != nullptr) {
    try {
      json js1 = (*value).invert2json();
      int64_t start = js1.at("start").get<int64_t>();
      int64_t end = js1.at("end").get<int64_t>();
      std::string data_type = "RuleNormData";
      if (js1.contains("data_type")) {
        data_type = js1.at("data_type").get<std::string>();
      }
      std::string root_dir = get_root_dir(data_type);
      string file_name =
          mix_cc::mix_time_t(UtilityTools::millsecond_zero2time_point(start))
              .to_formatted_time("%Y-%m-%d");
      std::vector<std::string> tags =
          js1.at("tags").get<std::vector<std::string>>();
      logger_->Debug() << "value:" << js1.dump() << std::endl;
      std::string full_path =
          root_dir + "/" + ruleid + "/" + file_name + ".json";
      if (UtilityTools::is_file_exists(full_path)) {
        logger_->Debug() << full_path << " already exists!" << endl;
        return;
      }
      auto js2 = data_query_(
          tags,
          mix_cc::time_range_t(UtilityTools::millsecond_zero2time_point(start),
                               UtilityTools::millsecond_zero2time_point(end)));
      if (js2.empty()) {
        logger_->Error() << ruleid << "|no data to save!" << std::endl;
        return;
      }
      auto res = UtilityTools::save_json_file(ruleid, file_name, js2, root_dir);
      if (res == 0) {
        logger_->Debug() << ruleid << "|save file:" << file_name
                         << " successfully!" << endl;
        return;
      } else if (res == 1) {
        logger_->Debug() << ruleid << "|" << file_name << " already exists!"
                         << endl;
        return;
      }
      logger_->Error() << ruleid << "|Failed to save file!" << endl;

    } catch (const std::exception &e) {
      logger_->Error() << e.what() << std::endl;
      return;
    }
  }
}

std::string DsmAlg::get_root_dir(std::string data_type) {
  if (data_type == "RuleNormData") {
    return this->RuleNormDataRootDir;
  }
  return this->RuleAlertDataRootDir;
}

void DsmAlg::timed_task() {
  for (auto ruleid : need_ruleid_) {
    this->dsm_handle(ruleid);
  }
}

void DsmAlg::dispose(int event_no, const ::Ice::ByteSeq &seq) {
  string str(seq.begin(), seq.end());
  logger_->Debug() << "seq:" << str << std::endl;
  // str -> json
  mix_cc::json json_value = mix_cc::json::parse(str);

  if (json_value.contains("ruleid")) {
    need_ruleid_ = json_value["ruleid"].get<std::vector<std::string>>();
  }
}