81 lines
2.1 KiB
C
81 lines
2.1 KiB
C
|
#pragma once
|
|||
|
|
|
|||
|
|
#include <limits.h>
|
|||
|
|
#include <unistd.h>
|
|||
|
|
|
|||
|
|
#include <chrono>
|
|||
|
|
#include <ctime>
|
|||
|
|
#include <filesystem>
|
|||
|
|
#include <fstream>
|
|||
|
|
#include <iomanip>
|
|||
|
|
#include <sstream>
|
|||
|
|
#include <string>
|
|||
|
|
|
|||
|
|
namespace common
|
|||
|
|
{
|
|||
|
|
|
|||
|
|
// 获取当前北京时间字符串 (精确到毫秒)
|
|||
|
|
inline std::string get_current_time_string()
|
|||
|
|
{
|
|||
|
|
using namespace std::chrono;
|
|||
|
|
auto now = system_clock::now();
|
|||
|
|
auto ms = duration_cast<milliseconds>(now.time_since_epoch()) % 1000;
|
|||
|
|
|
|||
|
|
// 先转 UTC
|
|||
|
|
auto t = system_clock::to_time_t(now);
|
|||
|
|
std::tm utc_tm = *std::gmtime(&t);
|
|||
|
|
|
|||
|
|
utc_tm.tm_hour += 8;
|
|||
|
|
mktime(&utc_tm);
|
|||
|
|
|
|||
|
|
std::ostringstream oss;
|
|||
|
|
oss << std::put_time(&utc_tm, "%Y-%m-%d %H:%M:%S") << '.' << std::setw(3) << std::setfill('0') << ms.count();
|
|||
|
|
return oss.str();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取当前日期字符串 (仅日期，用于日志文件名)
|
|||
|
|
inline std::string get_current_date_string()
|
|||
|
|
{
|
|||
|
|
using namespace std::chrono;
|
|||
|
|
auto now = system_clock::now();
|
|||
|
|
auto t = system_clock::to_time_t(now);
|
|||
|
|
|
|||
|
|
// 转成 UTC
|
|||
|
|
std::tm utc_tm = *std::gmtime(&t);
|
|||
|
|
|
|||
|
|
// UTC + 8 小时
|
|||
|
|
utc_tm.tm_hour += 8;
|
|||
|
|
mktime(&utc_tm); // 自动修正日期
|
|||
|
|
|
|||
|
|
std::ostringstream oss;
|
|||
|
|
oss << std::put_time(&utc_tm, "%Y-%m-%d");
|
|||
|
|
return oss.str();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取文件大小
|
|||
|
|
inline size_t get_file_size(const std::string& filename)
|
|||
|
|
{
|
|||
|
|
std::ifstream in(filename, std::ifstream::ate | std::ifstream::binary);
|
|||
|
|
return in.is_open() ? static_cast<size_t>(in.tellg()) : 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取可执行文件目录下的文件完整路径
|
|||
|
|
inline std::string get_executable_file_path(const std::string& filename)
|
|||
|
|
{
|
|||
|
|
char result[PATH_MAX];
|
|||
|
|
ssize_t count = readlink("/proc/self/exe", result, PATH_MAX);
|
|||
|
|
std::string path(result, count);
|
|||
|
|
auto dir = path.substr(0, path.find_last_of('/'));
|
|||
|
|
if (dir.back() != '/') dir += '/';
|
|||
|
|
return dir + filename;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取当前毫秒时间戳（steady_clock，不受系统时间影响）
|
|||
|
|
inline uint64_t get_timestamp_ms()
|
|||
|
|
{
|
|||
|
|
using namespace std::chrono;
|
|||
|
|
return duration_cast<milliseconds>(steady_clock::now().time_since_epoch()).count();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
} // namespace common
|