Mirror/zeek
1
0
Fork 0
mirror of https://github.com/zeek/zeek.git synced 2025-10-02 22:58:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Add SQLite page_count and file_size metrics

Browse source
This commit is contained in:
Tim Wojtulewicz 2025-07-03 09:49:54 -07:00
parent f73ac7089f
commit d0a6d84237
6 changed files with 47 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

36
src/storage/backend/sqlite/SQLite.cc
View file

@ -2,6 +2,7 @@
#include "zeek/storage/backend/sqlite/SQLite.h"
#include <sys/stat.h>
#include <filesystem>
#include <thread>
@ -12,7 +13,7 @@
#include "zeek/Func.h"
#include "zeek/Val.h"
#include "zeek/storage/ReturnCode.h"
#include "zeek/telemetry/Counter.h"
#include "zeek/telemetry/Manager.h"
#include "const.bif.netvar_h"
@ -260,6 +261,33 @@ OperationResult SQLite::DoOpen(OpenResultCallback* cb, RecordValPtr options) {
get_expiry_last_run_stmt = std::move(stmt_ptrs[6]);
update_expiry_last_run_stmt = std::move(stmt_ptrs[7]);
page_count_metric =
telemetry_mgr->GaugeInstance("zeek", "storage_sqlite_database_size", {{"config", GetConfigMetricsLabel()}},
"Storage sqlite backend value of page_count pragma", "pages", [this]() {
static auto double_parser = [](sqlite3_stmt* stmt) -> OperationResult {
double val = sqlite3_column_double(stmt, 0);
return {ReturnCode::SUCCESS, "", make_intrusive<DoubleVal>(val)};
};
auto res = RunPragma("page_count", std::nullopt, double_parser);
if ( res.code == ReturnCode::SUCCESS ) {
last_page_count_value = cast_intrusive<DoubleVal>(res.value)->Get();
}
return last_page_count_value;
});
file_size_metric =
telemetry_mgr->GaugeInstance("zeek", "storage_sqlite_database_size", {{"config", GetConfigMetricsLabel()}},
"Storage sqlite backend database file size on disk", "bytes", [this]() {
struct stat s{0};
if ( int ret = stat(full_path.c_str(), &s); ret == 0 ) {
last_file_size_value = static_cast<double>(s.st_size);
}
return last_file_size_value;
});
return {ReturnCode::SUCCESS};
}
@ -306,6 +334,12 @@ OperationResult SQLite::DoClose(ResultCallback* cb) {
expire_db = nullptr;
}
if ( page_count_metric )
page_count_metric->RemoveCallback();
if ( file_size_metric )
file_size_metric->RemoveCallback();
return op_res;
}