89 lines
2.8 KiB
C++
89 lines
2.8 KiB
C++
#ifndef DEBUG_DATA_SAVER_H
|
|
#define DEBUG_DATA_SAVER_H
|
|
|
|
#include <atomic>
|
|
#include <condition_variable>
|
|
#include <cstdint>
|
|
#include <map>
|
|
#include <mutex>
|
|
#include <queue>
|
|
#include <string>
|
|
#include <thread>
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
#include "VZNL_Types.h"
|
|
|
|
class DebugDataSaver
|
|
{
|
|
public:
|
|
explicit DebugDataSaver(int maxThreads = 4);
|
|
~DebugDataSaver();
|
|
|
|
void SaveAsync(const std::string& basePath,
|
|
const std::string& timestamp,
|
|
const std::vector<std::pair<EVzResultDataType, SVzLaserLineData>>& data,
|
|
const std::string& deviceName = "");
|
|
|
|
void Stop();
|
|
|
|
void SetMaxStorageSize(int64_t bytes) { m_maxStorageSize = bytes; }
|
|
void SetMaxQueueSize(int maxTasks);
|
|
void PrepareStorageCache(const std::string& basePath);
|
|
|
|
private:
|
|
struct SaveTask {
|
|
std::string basePath;
|
|
std::string filePath;
|
|
std::string deviceName;
|
|
std::vector<std::pair<EVzResultDataType, SVzLaserLineData>> data;
|
|
};
|
|
|
|
struct StorageFileInfo {
|
|
std::string path;
|
|
int64_t size = 0;
|
|
int64_t lastModifiedMSecs = 0;
|
|
};
|
|
|
|
struct StorageCache {
|
|
bool initialized = false;
|
|
int64_t totalSize = 0;
|
|
std::vector<StorageFileInfo> files;
|
|
};
|
|
|
|
void WorkerThread();
|
|
|
|
static std::vector<std::pair<EVzResultDataType, SVzLaserLineData>> DeepCopyData(
|
|
const std::vector<std::pair<EVzResultDataType, SVzLaserLineData>>& src);
|
|
static void FreeData(std::vector<std::pair<EVzResultDataType, SVzLaserLineData>>& data);
|
|
|
|
std::string GenerateSavePath(const std::string& basePath,
|
|
const std::string& timestamp,
|
|
const std::string& deviceName = "");
|
|
|
|
void EnsureStorageLimit(const std::string& basePath);
|
|
void EnsureStorageLimitLocked(const std::string& basePath, StorageCache& cache);
|
|
void BuildStorageCacheLocked(const std::string& basePath, StorageCache& cache);
|
|
void AddFileToCacheLocked(StorageCache& cache, const StorageFileInfo& fileInfo);
|
|
void RecordSavedFile(const std::string& basePath, const std::string& filePath);
|
|
|
|
static bool RemoveFileFromCache(StorageCache& cache, const std::string& filePath);
|
|
static bool IsOlderFile(const StorageFileInfo& lhs, const StorageFileInfo& rhs);
|
|
static int64_t ComputeDirSize(const std::string& rootPath);
|
|
static std::string FindOldestFile(const std::string& rootPath);
|
|
|
|
int m_maxThreads;
|
|
std::vector<std::thread> m_workers;
|
|
std::queue<SaveTask> m_taskQueue;
|
|
std::mutex m_queueMutex;
|
|
std::condition_variable m_queueCondition;
|
|
std::atomic<bool> m_running;
|
|
std::atomic<int> m_maxQueueSize{10};
|
|
|
|
std::mutex m_diskSpaceMutex;
|
|
std::atomic<int64_t> m_maxStorageSize{10LL * 1024 * 1024 * 1024};
|
|
std::map<std::string, StorageCache> m_storageCaches;
|
|
};
|
|
|
|
#endif // DEBUG_DATA_SAVER_H
|