调整类重载，修复部分BUG，增加请求头检查

2026-08-05 14:15:47 +08:00
parent 557cc8b995
commit 0d50588ea7
30 changed files with 1816 additions and 157 deletions
@@ -0,0 +1,66 @@
+#pragma once
+#include <map>
+#include <string>
+#include <thread>
+#include <chrono>
+#include "Export.h"
+#include <shared_mutex>
+#include "WebFileInfo.h"
+#include <unordered_map>
+
+using std::chrono::operator""s;
+using std::chrono::operator""min;
+
+class UNSWSC_DLL_EXPORT TempFileManager
+{
+private:
+	struct FileControlBlock
+	{
+		WebFileInfo info;
+		std::chrono::steady_clock::time_point expire_time;
+		std::chrono::steady_clock::time_point active_start_time;
+		std::chrono::seconds remaining_timeout;
+		std::chrono::seconds max_processing_timeout;
+		bool is_active = false;
+		bool is_permanent = false;                               // 是否永久保留
+		bool is_invalid = false;
+	};
+
+	std::string base_dir;
+	std::unordered_map<std::string, FileControlBlock> file_map;
+	mutable std::shared_mutex rw_mutex;
+
+	// C++20 jthread
+	std::jthread cleanup_thread;
+
+	void CleanupLoop(std::stop_token st) noexcept;
+
+public:
+	TempFileManager() noexcept;
+	~TempFileManager() noexcept;
+
+	// 禁止拷贝与移动
+	TempFileManager(const TempFileManager&) = delete;
+	TempFileManager& operator=(const TempFileManager&) = delete;
+	TempFileManager(TempFileManager&&) = delete;
+	TempFileManager& operator=(TempFileManager&&) = delete;
+
+	// 基础业务接口（保持不变）
+	bool SetBaseDirectory(std::string dir) noexcept;
+	bool RegisterFile(const WebFileInfo& info, std::chrono::seconds timeout, std::chrono::seconds max_proc_timeout = std::chrono::seconds(0)) noexcept;
+	bool InvalidateFile(const WebFileInfo& info);
+	//For SFR_FileMap<OriginFileName, StorageFileName>
+	bool InvalidateFile(const std::map<std::string, std::string>& info);
+	bool CopyFileTo(const std::string& storage_name, const std::string& dest_absolute_path) noexcept;
+	bool CutFileTo(const std::string& storage_name, const std::string& dest_absolute_path) noexcept;
+	bool RenameFile(const std::string& old_storage_name, const std::string& new_storage_name) noexcept;
+	bool DeleteFile(const std::string& storage_name) noexcept;
+	bool ActiveFile(const std::string& storage_name) noexcept;
+	bool DeactiveFile(const std::string& storage_name) noexcept;
+	bool SetPermanent(const std::string& storage_name, bool permanent) noexcept;
+	bool FileExists(const std::string& storage_name) const noexcept;
+	bool GetFileInfo(const std::string& storage_name, WebFileInfo& out_info) const noexcept;
+	size_t GetFileCount() const noexcept;
+	// 获取当前所有管理中的文件信息快照（线程安全，绝不抛出异常）
+	std::vector<WebFileInfo> GetAllFileInfos() const noexcept;
+};