mirror of
https://gitee.com/Zhaoxin59/my-chat_-server.git
synced 2026-02-13 17:11:48 +08:00
86 lines
2.9 KiB
C++
86 lines
2.9 KiB
C++
//实现项目中一些公共的工具类接口
|
||
//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;
|
||
}
|
||
|
||
} |