編寫健壯的C++鉤子以應對復雜環境

編寫健壯的C++鉤子（hook）需要考慮多個方面，包括環境適應性、錯誤處理、線程安全等。以下是一個示例，展示如何編寫一個健壯的C++鉤子，用于處理復雜環境中的日志記錄。

#include <iostream>
#include <fstream>
#include <string>
#include <mutex>
#include <thread>

// 定義一個日志級別枚舉
enum class LogLevel {
    DEBUG,
    INFO,
    WARNING,
    ERROR
};

// 定義一個日志鉤子接口
class ILogHook {
public:
    virtual ~ILogHook() = default;
    virtual void log(LogLevel level, const std::string& message) = 0;
};

// 實現一個文件日志鉤子
class FileLogHook : public ILogHook {
public:
    FileLogHook(const std::string& logFilePath) : logFilePath(logFilePath) {}

    void log(LogLevel level, const std::string& message) override {
        std::lock_guard<std::mutex> lock(mutex);
        std::ofstream logFile;
        logFile.open(logFilePath, std::ios::app);
        if (logFile.is_open()) {
            logFile << getCurrentTime() << " [" << logLevelToString(level) << "] " << message << std::endl;
            logFile.close();
        } else {
            std::cerr << "Failed to open log file: " << logFilePath << std::endl;
        }
    }

private:
    std::string logFilePath;
    std::mutex mutex;

    std::string getCurrentTime() {
        auto now = std::chrono::system_clock::now();
        std::time_t time = std::chrono::system_clock::to_time_t(now);
        std::string timeStr = std::ctime(&time);
        timeStr.pop_back(); // Remove the newline character
        return timeStr;
    }

    std::string logLevelToString(LogLevel level) {
        switch (level) {
            case LogLevel::DEBUG: return "DEBUG";
            case LogLevel::INFO: return "INFO";
            case LogLevel::WARNING: return "WARNING";
            case LogLevel::ERROR: return "ERROR";
            default: return "UNKNOWN";
        }
    }
};

// 示例使用
int main() {
    // 創建一個文件日志鉤子實例
    FileLogHook logHook("log.txt");

    // 使用多個線程記錄日志
    std::thread t1([&logHook]() {
        logHook.log(LogLevel::DEBUG, "This is a debug message from thread 1");
    });

    std::thread t2([&logHook]() {
        logHook.log(LogLevel::INFO, "This is an info message from thread 2");
    });

    std::thread t3([&logHook]() {
        logHook.log(LogLevel::WARNING, "This is a warning message from thread 3");
    });

    std::thread t4([&logHook]() {
        logHook.log(LogLevel::ERROR, "This is an error message from thread 4");
    });

    // 等待所有線程完成
    t1.join();
    t2.join();
    t3.join();
    t4.join();

    return 0;
}

關鍵點解釋

1. 接口定義：ILogHook接口定義了鉤子的基本功能，便于擴展和維護。
2. 文件日志鉤子實現：FileLogHook類實現了ILogHook接口，并將日志記錄到文件中。它使用互斥鎖確保線程安全。
3. 時間獲取和日志級別轉換：getCurrentTime和logLevelToString方法分別用于獲取當前時間和將日志級別轉換為字符串。
4. 多線程測試：示例中創建了多個線程來測試鉤子的多線程安全性。

通過這種方式，可以編寫一個健壯的C++鉤子，以應對復雜環境中的日志記錄需求。