ZYM/BH-TicketSystem
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

avoid bug

Browse Source
This commit is contained in:
happyZYM
2024-04-26 13:15:00 +00:00
parent bb4aaebc5b
commit 39dffd88cd
7 changed files with 487 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
bpt/include/bpt/disk_manager.h
View File

@ -17,7 +17,7 @@ class DiskManager {
* for first_empty_page_id).
*/
public:
explicit DiskManager(const std::string &file_path_);
explicit DiskManager(const std::string &file_path_, bool renew = false);
~DiskManager();
char *RawDataMemory();
size_t RawDatMemorySize();

5
bpt/src/disk_manager.cpp
View File

@ -3,13 +3,14 @@
#include <exception>
#include <stdexcept>
const size_t kPageSize = 4096;
DiskManager::DiskManager(const std::string &file_path_)
DiskManager::DiskManager(const std::string &file_path_, bool renew)
: file_path(file_path_),
first_empty_page_id(0),
current_total_page_count(0),
current_none_empty_page_count(0),
raw_data_memory(nullptr),
fp(nullptr) {
if (renew) remove(file_path.c_str());
fp = fopen(file_path.c_str(), "r+b");
if (fp == nullptr) {
// File doesn't exist, create a new one
@ -46,7 +47,7 @@ char *DiskManager::RawDataMemory() { return raw_data_memory; }
size_t DiskManager::RawDatMemorySize() { return kPageSize - meta_data_size; }
void DiskManager::FullyFlush() {
if(fp==nullptr) return;
if (fp == nullptr) return;
fseek(fp, 0, SEEK_SET);
fwrite(&first_empty_page_id, sizeof(page_id_t), 1, fp);
fwrite(&current_total_page_count, sizeof(size_t), 1, fp);