From f5a7f355c2125f45f2be23caca072f50a1f0b321 Mon Sep 17 00:00:00 2001 From: xxy <3237539022@qq.com> Date: Sun, 1 Dec 2024 10:06:45 +0800 Subject: [PATCH] benchmark --- test/benchmark_4leveldb.cpp | 175 +++++++++++++++++++++++++------------------- 1 file changed, 98 insertions(+), 77 deletions(-) diff --git a/test/benchmark_4leveldb.cpp b/test/benchmark_4leveldb.cpp index da83ab5..b14944d 100644 --- a/test/benchmark_4leveldb.cpp +++ b/test/benchmark_4leveldb.cpp @@ -1,108 +1,129 @@ -#include -#include +#include +#include #include +#include #include - -#include "leveldb/db.h" - -#define TEST_FREQUENCY (10000) - -char* randomstr() -{ - static char buf[1024]; - int len = rand() % 768 + 255; +#include "leveldb/db.h" + +// 配置 +const int TEST_EXPONENT = 4; +const int TEST_FREQUENCY = static_cast(std::pow(10, TEST_EXPONENT)); +const int MIN_STR_LEN = 255; +const int MAX_STR_LEN = 1024; +const std::string DB_PATH = "./db_benchmark"; +const std::string CHARSET = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"; + +// 多语言 +const std::string BASE_VALUE = "こんにちは世界！Hello World! Привет, мир! ¡Hola Mundo! 你好，世界！Bonjour le monde! Hallo Welt!"; +// 莎士比亚 +// const std::string BASE_VALUE = "To be, or not to be, that is the question: Whether 'tis nobler in the mind to suffer the slings and arrows of outrageous fortune, or to take arms against a sea of troubles and by opposing end them."; +// 超长字符 +// const std::string BASE_VALUE = []() { +// std::string base = "壹贰叁肆伍陆柒捌玖拾"; +// std::string long_text; +// for (int i = 0; i < 100; ++i) { // 重复 100 次 +// long_text += base; +// } +// return long_text; +// }(); + + +// 随机字符串生成 +std::string randomStr() { + int len = rand() % (MAX_STR_LEN - MIN_STR_LEN + 1) + MIN_STR_LEN; + std::string str(len, '\0'); for (int i = 0; i < len; ++i) { - buf[i] = 'A' + rand() % 26; + str[i] = CHARSET[rand() % CHARSET.size()]; } - buf[len] = '\0'; - return buf; + return str; } - -int main() -{ + +// 计算并输出耗时的模板函数 +template +void measureTime(const std::string& operation, Func func) { + auto start = std::chrono::system_clock::now(); + func(); + auto end = std::chrono::system_clock::now(); + auto duration = std::chrono::duration_cast(end - start); + std::cout << TEST_FREQUENCY << "次" << operation << "耗时: " + << double(duration.count()) * std::chrono::microseconds::period::num / std::chrono::microseconds::period::den + << "秒" << std::endl; +} + +int main() { leveldb::DB* db; leveldb::Options options; options.create_if_missing = true; - + // 打开数据库 - leveldb::Status status = leveldb::DB::Open(options, "./testdb", &db); + leveldb::Status status = leveldb::DB::Open(options, DB_PATH, &db); assert(status.ok()); - + std::cout << "数据库已打开: " << DB_PATH << std::endl; + srand(2017); - std::string k[TEST_FREQUENCY]; + + // 生成测试数据 + std::string keys[TEST_FREQUENCY]; for (int i = 0; i < TEST_FREQUENCY; ++i) { - k[i] = (randomstr()); + keys[i] = randomStr(); + } + std::string value = BASE_VALUE; + for (int i = 0; i < 4; ++i) { + value += value; // 扩展 base value } - std::string v("壹贰叁肆伍陆柒捌玖拾"); - v.append(v).append(v).append(v).append(v).append(v); - + // 测试添加 - { - auto start = std::chrono::system_clock::now(); + measureTime("添加", [&]() { for (int i = 0; i < TEST_FREQUENCY; ++i) { - status = db->Put(leveldb::WriteOptions(), k[i], v); + status = db->Put(leveldb::WriteOptions(), keys[i], value); assert(status.ok()); } - auto end = std::chrono::system_clock::now(); - auto duration = std::chrono::duration_cast(end - start); - - std::cout << TEST_FREQUENCY <<"次添加耗时: " - << double(duration.count()) * std::chrono::microseconds::period::num / std::chrono::microseconds::period::den - << "秒" << std::endl; - } + }); + // 测试获取 - { - auto start = std::chrono::system_clock::now(); - std::string v2[TEST_FREQUENCY]; + measureTime("获取", [&]() { + std::string retrievedValues[TEST_FREQUENCY]; for (int i = 0; i < TEST_FREQUENCY; ++i) { - status = db->Get(leveldb::ReadOptions(), k[i], &v2[i]); + status = db->Get(leveldb::ReadOptions(), keys[i], &retrievedValues[i]); assert(status.ok()); + assert(retrievedValues[i] == value); // 验证获取结果 } - auto end = std::chrono::system_clock::now(); - auto duration = std::chrono::duration_cast(end - start); - - std::cout << TEST_FREQUENCY <<"次获取耗时: " - << double(duration.count()) * std::chrono::microseconds::period::num / std::chrono::microseconds::period::den - << "秒" << std::endl; - // 验证获取结果是否正确 - std::string ss; - for (int i = 0; i < TEST_FREQUENCY; ++i) { - if (v2[i] != v) { - std::cout << "第 " << i << " 个结果不正确" << std::endl; - std::cout << v2[i] << std::endl; - } - } - } + }); + // 测试修改 - { - auto start = std::chrono::system_clock::now(); - v.append(v); + measureTime("修改", [&]() { + std::string newValue = value + value; for (int i = 0; i < TEST_FREQUENCY; ++i) { - status = db->Put(leveldb::WriteOptions(), k[i], v); + status = db->Put(leveldb::WriteOptions(), keys[i], newValue); assert(status.ok()); } - auto end = std::chrono::system_clock::now(); - auto duration = std::chrono::duration_cast(end - start); - - std::cout << TEST_FREQUENCY <<"次修改耗时: " - << double(duration.count()) * std::chrono::microseconds::period::num / std::chrono::microseconds::period::den - << "秒" << std::endl; - } - + }); + // 测试删除 - { - auto start = std::chrono::system_clock::now(); + measureTime("删除", [&]() { for (int i = 0; i < TEST_FREQUENCY; ++i) { - status = db->Delete(leveldb::WriteOptions(), k[i]); + status = db->Delete(leveldb::WriteOptions(), keys[i]); assert(status.ok()); } - auto end = std::chrono::system_clock::now(); - auto duration = std::chrono::duration_cast(end - start); - - std::cout << TEST_FREQUENCY <<"次删除耗时: " - << double(duration.count()) * std::chrono::microseconds::period::num / std::chrono::microseconds::period::den - << "秒" << std::endl; - } + }); + + if (db) { delete db; + db = nullptr; + } + std::cout << "测试完成，数据库已关闭。" << std::endl; + + // 删除数据库目录 + #ifdef _WIN32 + std::string command = "rd /s /q \"" + DB_PATH + "\""; // Windows 删除目录 + #else + std::string command = "rm -rf \"" + DB_PATH + "\""; // Linux/macOS 删除目录 + #endif + if (std::system(command.c_str()) == 0) { + std::cout << "数据库目录已删除: " << DB_PATH << std::endl; + } else { + std::cerr << "警告: 数据库目录删除失败，请手动检查!" << std::endl; + } + return 0; -} \ No newline at end of file +}