update

common/utils.hpp

@@ -0,0 +1,86 @@
+//实现项目中一些公共的工具类接口
+//1. 生成一个唯一ID的接口
+//2. 文件的读写操作接口
+
+#include <iostream>
+#include <fstream>
+#include <sstream>
+#include <string>
+#include <atomic>
+#include <random>
+#include <iomanip>
+#include "logger.hpp"
+
+namespace bite_im {
+
+std::string uuid() {
+    //生成一个由16位随机字符组成的字符串作为唯一ID
+    // 1. 生成6个0~255之间的随机数字(1字节-转换为16进制字符)--生成12位16进制字符
+    std::random_device rd;//实例化设备随机数对象-用于生成设备随机数
+    std::mt19937 generator(rd());//以设备随机数为种子，实例化伪随机数对象
+    std::uniform_int_distribution<int> distribution(0,255); //限定数据范围
+
+    std::stringstream ss;
+    for (int i = 0; i < 6; i++) {
+        if (i == 2) ss << "-";
+        ss << std::setw(2) << std::setfill('0') << std::hex << distribution(generator);
+    }
+    ss << "-";
+    // 2. 通过一个静态变量生成一个2字节的编号数字--生成4位16进制数字字符
+    static std::atomic<short> idx(0);
+    short tmp = idx.fetch_add(1);
+    ss << std::setw(4) << std::setfill('0') << std::hex << tmp;
+    return ss.str();
+}
+
+std::string vcode() {
+    std::random_device rd;//实例化设备随机数对象-用于生成设备随机数
+    std::mt19937 generator(rd());//以设备随机数为种子，实例化伪随机数对象
+    std::uniform_int_distribution<int> distribution(0,9); //限定数据范围
+
+    std::stringstream ss;
+    for (int i = 0; i < 4; i++) {
+        ss << distribution(generator);
+    }
+    return ss.str();
+}
+
+bool readFile(const std::string &filename, std::string &body){
+    //实现读取一个文件的所有数据，放入body中
+    std::ifstream ifs(filename, std::ios::binary | std::ios::in);
+    if (ifs.is_open() == false) {
+        LOG_ERROR("打开文件 {} 失败！", filename);
+        return false;
+    }
+    ifs.seekg(0, std::ios::end);//跳转到文件末尾
+    size_t flen = ifs.tellg();  //获取当前偏移量-- 文件大小
+    ifs.seekg(0, std::ios::beg);//跳转到文件起始
+    body.resize(flen);
+    ifs.read(&body[0], flen);
+    if (ifs.good() == false) {
+        LOG_ERROR("读取文件 {} 数据失败！", filename);
+        ifs.close();
+        return false;
+    }
+    ifs.close();
+    return true;
+}
+
+bool writeFile(const std::string &filename, const std::string &body){
+    //实现将body中的数据，写入filename对应的文件中
+    std::ofstream ofs(filename, std::ios::out | std::ios::binary | std::ios::trunc);
+    if (ofs.is_open() == false) {
+        LOG_ERROR("打开文件 {} 失败！", filename);
+        return false;
+    }
+    ofs.write(body.c_str(), body.size());
+    if (ofs.good() == false) {
+        LOG_ERROR("读取文件 {} 数据失败！", filename);
+        ofs.close();
+        return false;
+    }
+    ofs.close();
+    return true;
+}
+
+}