127 lines
2.9 KiB
C
127 lines
2.9 KiB
C
|
#pragma once
|
|||
|
|
/**
|
|||
|
|
* @file public.h
|
|||
|
|
* @brief data save handle
|
|||
|
|
* 保存正常数据的处理类
|
|||
|
|
* @author your name (you@domain.com)
|
|||
|
|
* @version 0.1
|
|||
|
|
* @date 2026-01-29
|
|||
|
|
*
|
|||
|
|
* Copyright: Baosight Co. Ltd.
|
|||
|
|
* DO NOT COPY/USE WITHOUT PERMISSION
|
|||
|
|
*
|
|||
|
|
*/
|
|||
|
|
#include "mix_cc/json.h"
|
|||
|
|
#include <chrono>
|
|||
|
|
#include <map>
|
|||
|
|
#include <memory>
|
|||
|
|
#include <string>
|
|||
|
|
#include <vector>
|
|||
|
|
#include <zlib/MemMap.hpp>
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief dsm用MemMap的配置
|
|||
|
|
*/
|
|||
|
|
struct DsmMapConfig {
|
|||
|
|
static inline constexpr char TableName[] = "ZONE10"; ///<表名
|
|||
|
|
static constexpr int MapMaxSize = 5000; ///< map 最大容量
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
typedef char TagName[50]; ///<数据项(tag点英文item，长度限制50字节以内)
|
|||
|
|
typedef TagName Tags[30]; ///<单个规则tag点，个数限制30个以内
|
|||
|
|
using std::vector;
|
|||
|
|
/**
|
|||
|
|
* @brief 保存正常数据，数据信息
|
|||
|
|
*/
|
|||
|
|
struct DataInfo {
|
|||
|
|
int64_t start;
|
|||
|
|
int64_t end;
|
|||
|
|
Tags tags;
|
|||
|
|
int flag = 0; ///< 0--RuleNormData；1----RuleAlertData
|
|||
|
|
int tag_size = 0;
|
|||
|
|
|
|||
|
|
void init(vector<std::string> tag_s) {
|
|||
|
|
tag_size = 0;
|
|||
|
|
int size_n = tag_s.size();
|
|||
|
|
if (size_n == 0) {
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
for (int i = 0; i < size_n; i++) {
|
|||
|
|
std::strcpy(tags[i], tag_s[i].c_str());
|
|||
|
|
tag_size++;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
/**
|
|||
|
|
* @brief
|
|||
|
|
* @param _start 开始时间 时间戳
|
|||
|
|
* @param _end 结束时间 时间戳
|
|||
|
|
* @param _flag 数据类型 0--正常数据；1--报警数据，
|
|||
|
|
* default 0 正常数据
|
|||
|
|
*/
|
|||
|
|
void update(int64_t _start, int64_t _end, int _flag = 0) {
|
|||
|
|
start = _start;
|
|||
|
|
end = _end;
|
|||
|
|
flag = _flag;
|
|||
|
|
}
|
|||
|
|
DataInfo() {
|
|||
|
|
tag_size = 0;
|
|||
|
|
start = 0;
|
|||
|
|
end = 0;
|
|||
|
|
flag = 0;
|
|||
|
|
}
|
|||
|
|
DataInfo(nlohmann::json js1) {
|
|||
|
|
tag_size = 0;
|
|||
|
|
start = js1["start"];
|
|||
|
|
end = js1["end"];
|
|||
|
|
vector<std::string> tag_s = js1["tags"];
|
|||
|
|
for (int i = 0; i < tag_s.size(); i++) {
|
|||
|
|
std::strcpy(tags[i], tag_s[i].c_str());
|
|||
|
|
tag_size++;
|
|||
|
|
}
|
|||
|
|
if (js1.contains("data_type")) {
|
|||
|
|
std::string data_type = js1["data_type"].get<std::string>();
|
|||
|
|
flag = 0;
|
|||
|
|
if (data_type == "RuleAlertData") {
|
|||
|
|
flag = 1;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
nlohmann::json invert2json() {
|
|||
|
|
nlohmann::json js1;
|
|||
|
|
js1["start"] = start;
|
|||
|
|
js1["end"] = end;
|
|||
|
|
vector<std::string> tag_s;
|
|||
|
|
for (int i = 0; i < tag_size; i++) {
|
|||
|
|
tag_s.push_back(std::string(tags[i]));
|
|||
|
|
}
|
|||
|
|
js1["tags"] = tag_s;
|
|||
|
|
js1["data_type"] = "RuleNormData";
|
|||
|
|
if (flag == 1) {
|
|||
|
|
js1["data_type"] = "RuleAlertData";
|
|||
|
|
}
|
|||
|
|
return js1;
|
|||
|
|
}
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
class DsmHandle {
|
|||
|
|
private:
|
|||
|
|
std::unique_ptr<CMemMap<std::string, DataInfo>> ruleid_CMemMap_;
|
|||
|
|
|
|||
|
|
public:
|
|||
|
|
DsmHandle() {
|
|||
|
|
if (ruleid_CMemMap_ == nullptr) {
|
|||
|
|
ruleid_CMemMap_ = std::make_unique<CMemMap<std::string, DataInfo>>(
|
|||
|
|
DsmMapConfig::TableName, DsmMapConfig::MapMaxSize);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
~DsmHandle() {}
|
|||
|
|
bool insert(const std::string& ruleid, DataInfo *data_info) {
|
|||
|
|
try {
|
|||
|
|
ruleid_CMemMap_->insert(ruleid, data_info);
|
|||
|
|
} catch (const std::exception &e) {
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
};
|