osb/source/utility/btree_repacker.cpp

59 lines
1.8 KiB
C++
Raw Normal View History

2023-10-18 18:44:32 +00:00
#include "StarBTreeDatabase.hpp"
#include "StarTime.hpp"
#include "StarFile.hpp"
#include "StarVersionOptionParser.hpp"
using namespace Star;
int main(int argc, char** argv) {
try {
double startTime = Time::monotonicTime();
VersionOptionParser optParse;
optParse.setSummary("Repacks a Starbound BTree file to shrink its file size");
optParse.addArgument("input file path", OptionParser::Required, "Path to the BTree to be repacked");
optParse.addArgument("output filename", OptionParser::Optional, "Output BTree file");
auto opts = optParse.commandParseOrDie(argc, argv);
String bTreePath = opts.arguments.at(0);
String outputFilename = opts.arguments.get(1, bTreePath + ".repack");
outputFilename = File::relativeTo(File::fullPath(File::dirName(outputFilename)), File::baseName(outputFilename));
//open the old db
BTreeDatabase db;
db.setIODevice(std::move(File::open(bTreePath, IOMode::Read)));
db.open();
//make a new db
BTreeDatabase newDb;
newDb.setBlockSize(db.blockSize());
newDb.setContentIdentifier(db.contentIdentifier());
newDb.setKeySize(db.keySize());
newDb.setAutoCommit(false);
newDb.setIODevice(std::move(File::open(outputFilename, IOMode::ReadWrite | IOMode::Truncate)));
newDb.open();
2023-10-22 17:39:15 +00:00
coutf("Repacking {}...\n", bTreePath);
2023-10-18 18:44:32 +00:00
//copy the data over
unsigned count = 0;
db.forAll([&count, &newDb](ByteArray key, ByteArray data) {
newDb.insert(key, data);
++count;
});
//close the old db
db.close();
//commit and close the new db
newDb.commit();
newDb.close();
2023-10-22 17:39:15 +00:00
coutf("Repacked BTree to {} in {}s\n", outputFilename, Time::monotonicTime() - startTime);
2023-10-18 18:44:32 +00:00
return 0;
} catch (std::exception const& e) {
2023-10-22 17:39:15 +00:00
cerrf("Exception caught: {}\n", outputException(e, true));
2023-10-18 18:44:32 +00:00
return 1;
}
}