代码更新

2025-08-11 11:17:32 +08:00
parent dd2e305f2a
commit a8633f5879
3 changed files with 244 additions and 1 deletions
--- a/README.md
+++ b/README.md
@@ -6,3 +6,12 @@ c/c++基本开发库

 # Linux 中屏蔽所有信号操作
 屏蔽所有信号,以防止意外退出
+
+
+# 写文件操作
+允许原文本进行覆盖写,追加写
+允许二进制进行覆盖写,追加写
+允许在特定位置进行写操作
+可以计算特定符号最后一个字节所在位置
+
+所有操作都添加mutex锁机制 ,保障线程安全
--- a/include/Netra.hpp
+++ b/include/Netra.hpp
@@ -79,7 +79,116 @@ namespace QCL
        std::vector<int> clientSockets_;         ///< 当前所有连接的客户端Socket集合
    };
    ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
+    /**
+     * @brief 文件写入工具类（线程安全）
+     *
+     * 该类支持多种文件写入方式：
+     *  - 覆盖写文本
+     *  - 追加写文本
+     *  - 按位置覆盖原文写入
+     *  - 二进制覆盖写
+     *  - 二进制追加写
+     *
+     * 特点：
+     *  - 文件不存在时自动创建
+     *  - 支持覆盖和追加两种模式
+     *  - 支持二进制模式，适合写入非文本数据
+     *  - 内部使用 std::mutex 实现线程安全
+     */
+    class WriteFile
+    {
+    public:
+        /**
+         * @brief 构造函数
+         * @param filePath 文件路径
+         */
+        explicit WriteFile(const std::string &filePath);

+        /**
+         * @brief 覆盖写文本文件（线程安全）
+         * @param content 要写入的文本内容
+         * @return true 写入成功
+         * @return false 写入失败
+         */
+        bool overwriteText(const std::string &content);
+
+        /**
+         * @brief 追加写文本文件（线程安全）
+         * @param content 要写入的文本内容
+         * @return true 写入成功
+         * @return false 写入失败
+         */
+        bool appendText(const std::string &content);
+
+        /**
+         * @brief 按指定位置写入文本（原文写，线程安全）
+         *
+         * 不清空文件内容，仅替换指定位置的内容。
+         * 若文件不存在，则会自动创建空文件。
+         *
+         * @param content 要写入的内容
+         * @param position 写入位置（默认为文件开头）
+         * @return true 写入成功
+         * @return false 写入失败
+         */
+        bool writeOriginal(const std::string &content, std::streampos position = 0);
+
+        /**
+         * @brief 覆盖写二进制文件（线程安全）
+         * @param data 要写入的二进制数据
+         * @return true 写入成功
+         * @return false 写入失败
+         */
+        bool overwriteBinary(const std::vector<char> &data);
+
+        /**
+         * @brief 追加写二进制文件（线程安全）
+         * @param data 要写入的二进制数据
+         * @return true 写入成功
+         * @return false 写入失败
+         */
+        bool appendBinary(const std::vector<char> &data);
+
+        /**
+         * @brief 计算第一个指定字节序列前的字节数（包含该字节序列本身）
+         *
+         * 例如：文件内容是 "ABC***--***XYZ"，模式是 "***--***"
+         * 返回值应为 11（"ABC"=3字节 + "***--***"=8字节）。
+         *
+         * @param pattern 要查找的字节模式（支持多字节）
+         * @return long 字节数（包含匹配模式本身），未找到返回 -1
+         */
+        long countBytesBeforePattern(const std::string &pattern);
+
+    private:
+        std::string filePath_;  ///< 文件路径
+        std::mutex writeMutex_; ///< 线程锁，保证多线程写入安全
+
+        /**
+         * @brief 通用文本写入接口（线程安全）
+         * @param content 要写入的内容
+         * @param mode 打开模式（追加/覆盖等）
+         * @return true 写入成功
+         * @return false 写入失败
+         */
+        bool writeToFile(const std::string &content, std::ios::openmode mode);
+
+        /**
+         * @brief 通用二进制写入接口（线程安全）
+         * @param data 要写入的二进制数据
+         * @param mode 打开模式（追加/覆盖等）
+         * @return true 写入成功
+         * @return false 写入失败
+         */
+        bool writeBinary(const std::vector<char> &data, std::ios::openmode mode);
+    };
+
+    // 读文件操作
+    class ReadFile
+    {
+    };
+
+    ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
    // 屏蔽所有信号
    void blockAllSignals();
 }
--- a/src/Netra.cpp
+++ b/src/Netra.cpp
@@ -2,6 +2,7 @@

 namespace QCL
 {
+    ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
    TcpServer::TcpServer(int port)
        : port_(port), running_(false), serverSock_(-1) {}

@@ -215,7 +216,130 @@ namespace QCL
        snprintf(result, INET_ADDRSTRLEN + 10, "%s:%d", ip, port);
        return result;
    }
+    ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
+    WriteFile::WriteFile(const std::string &filePath)
+        : filePath_(filePath) {}

+    /**
+     * @brief 覆盖写文本（线程安全）
+     */
+    bool WriteFile::overwriteText(const std::string &content)
+    {
+        std::lock_guard<std::mutex> lock(writeMutex_); // 加锁
+        return writeToFile(content, std::ios::out | std::ios::trunc);
+    }
+
+    /**
+     * @brief 追加写文本（线程安全）
+     */
+    bool WriteFile::appendText(const std::string &content)
+    {
+        std::lock_guard<std::mutex> lock(writeMutex_);
+        return writeToFile(content, std::ios::out | std::ios::app);
+    }
+
+    /**
+     * @brief 按位置写入（线程安全）
+     */
+    bool WriteFile::writeOriginal(const std::string &content, std::streampos position)
+    {
+        std::lock_guard<std::mutex> lock(writeMutex_);
+
+        std::ofstream file(filePath_, std::ios::in | std::ios::out);
+        if (!file.is_open())
+        {
+            // 文件不存在则创建
+            file.open(filePath_, std::ios::out);
+            file.close();
+            file.open(filePath_, std::ios::in | std::ios::out);
+        }
+        if (!file.is_open())
+            return false;
+
+        file.seekp(position);
+        file << content;
+        file.close();
+        return true;
+    }
+
+    /**
+     * @brief 覆盖写二进制（线程安全）
+     */
+    bool WriteFile::overwriteBinary(const std::vector<char> &data)
+    {
+        std::lock_guard<std::mutex> lock(writeMutex_);
+        return writeBinary(data, std::ios::out | std::ios::trunc | std::ios::binary);
+    }
+
+    /**
+     * @brief 追加写二进制（线程安全）
+     */
+    bool WriteFile::appendBinary(const std::vector<char> &data)
+    {
+        std::lock_guard<std::mutex> lock(writeMutex_);
+        return writeBinary(data, std::ios::out | std::ios::app | std::ios::binary);
+    }
+
+    /**
+     * @brief 通用文本写入（私有）
+     */
+    bool WriteFile::writeToFile(const std::string &content, std::ios::openmode mode)
+    {
+        std::ofstream file(filePath_, mode);
+        if (!file.is_open())
+            return false;
+        file << content;
+        file.close();
+        return true;
+    }
+
+    /**
+     * @brief 通用二进制写入（私有）
+     */
+    bool WriteFile::writeBinary(const std::vector<char> &data, std::ios::openmode mode)
+    {
+        std::ofstream file(filePath_, mode);
+        if (!file.is_open())
+            return false;
+        file.write(data.data(), data.size());
+        file.close();
+        return true;
+    }
+
+    /**
+     * @brief 计算第一个指定字节序列前的字节数（包含该字节序列本身）
+     */
+    long WriteFile::countBytesBeforePattern(const std::string &pattern)
+    {
+        std::lock_guard<std::mutex> lock(writeMutex_);
+
+        if (pattern.empty())
+            return -1;
+
+        std::ifstream file(filePath_, std::ios::binary); // 二进制模式防止编码干扰
+        if (!file.is_open())
+            return -1;
+
+        // 将整个文件读入内存
+        std::vector<char> buffer((std::istreambuf_iterator<char>(file)),
+                                 std::istreambuf_iterator<char>());
+        file.close();
+
+        // 在 buffer 中查找 pattern
+        auto it = std::search(buffer.begin(), buffer.end(),
+                              pattern.begin(), pattern.end());
+
+        if (it == buffer.end())
+        {
+            return -1; // 没找到
+        }
+
+        // 计算从开头到 pattern 结束的字节数
+        size_t pos = std::distance(buffer.begin(), it);
+        return static_cast<long>(pos + pattern.size());
+    }
+
+    ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
    // 屏蔽所有信号
    void blockAllSignals()
    {
@@ -223,4 +347,5 @@ namespace QCL
        for (int ii = 1; ii <= 64; ii++)
            signal(ii, SIG_IGN);
    }
+    ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
 }