109 lines
2.9 KiB
C++
109 lines
2.9 KiB
C++
#ifndef ASSISTANT_H
|
|
#define ASSISTANT_H
|
|
|
|
#include <fstream>
|
|
#include <functional>
|
|
#include <random>
|
|
#include <string>
|
|
#include <thread>
|
|
|
|
#ifdef USE_BOOST
|
|
#include <boost/filesystem.hpp>
|
|
namespace fs = boost::filesystem;
|
|
#else
|
|
#include <filesystem>
|
|
namespace fs = std::filesystem;
|
|
#endif
|
|
|
|
/**
|
|
* @brief 比较两个文件的内容是否完全相同
|
|
*
|
|
* @param filea 第一个文件路径
|
|
* @param fileb 第二个文件路径
|
|
* @return true 如果文件内容完全相同
|
|
* @return false 如果文件内容不同或无法读取文件
|
|
*/
|
|
bool is_equal_filecontent(const std::string& filea, const std::string& fileb);
|
|
bool random_file(const std::string& file, size_t size);
|
|
|
|
/**
|
|
* @brief 等待变量的值达到预期值
|
|
*
|
|
* @tparam T 变量类型
|
|
* @tparam Compare 比较器类型
|
|
* @param value 变量的引用
|
|
* @param expected 预期值
|
|
* @param comparator 比较函数 (默认为 std::equal_to)
|
|
* @param timeout_ms 超时时间(毫秒)，0表示无限等待
|
|
* @param interval_ms 检查间隔(毫秒)
|
|
* @return true 变量值符合预期
|
|
* @return false 超时或条件不满足
|
|
*/
|
|
template <typename T, typename Compare = std::equal_to<T>>
|
|
bool value_wait(const std::function<T()>& value_ref, const T& expected, Compare comparator = Compare(),
|
|
unsigned long timeout_ms = 0, unsigned long interval_ms = 100)
|
|
{
|
|
auto start = std::chrono::steady_clock::now();
|
|
|
|
while (true) {
|
|
|
|
T value = value_ref();
|
|
if (comparator(value, expected)) {
|
|
return true;
|
|
}
|
|
|
|
if (timeout_ms > 0) {
|
|
auto elapsed = std::chrono::duration_cast<std::chrono::milliseconds>(
|
|
std::chrono::steady_clock::now() - start)
|
|
.count();
|
|
|
|
if (elapsed >= timeout_ms) {
|
|
return false;
|
|
}
|
|
}
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(interval_ms));
|
|
}
|
|
}
|
|
|
|
class ScopeExit
|
|
{
|
|
public:
|
|
ScopeExit() = default;
|
|
template <typename F> ScopeExit(F&& f) : func_(std::forward<F>(f))
|
|
{
|
|
}
|
|
ScopeExit(const ScopeExit&) = delete;
|
|
ScopeExit& operator=(const ScopeExit&) = delete;
|
|
ScopeExit(ScopeExit&& other) noexcept : func_(std::move(other.func_))
|
|
{
|
|
other.func_ = nullptr;
|
|
}
|
|
ScopeExit& operator=(ScopeExit&& other) noexcept
|
|
{
|
|
if (this != &other) {
|
|
if (func_)
|
|
func_();
|
|
func_ = std::move(other.func_);
|
|
other.func_ = nullptr;
|
|
}
|
|
return *this;
|
|
}
|
|
~ScopeExit()
|
|
{
|
|
if (func_)
|
|
func_();
|
|
}
|
|
void dismiss() noexcept
|
|
{
|
|
func_ = nullptr;
|
|
}
|
|
|
|
private:
|
|
std::function<void()> func_;
|
|
};
|
|
|
|
#define _SCOPE_EXIT_CONCAT(a, b) a##b
|
|
#define _MAKE_ON_SCOPE_EXIT(line) ScopeExit _SCOPE_EXIT_CONCAT(exit_defer_, line) = [&]()
|
|
#define ON_SCOPE_EXIT _MAKE_ON_SCOPE_EXIT(__LINE__)
|
|
|
|
#endif // ASSISTANT_H
|