Mirror/zeek
1
0
Fork 0
mirror of https://github.com/zeek/zeek.git synced 2025-10-08 17:48:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Always register backend for expiration, check for open during loop

Browse source
This commit is contained in:
Tim Wojtulewicz 2025-02-06 12:50:41 -07:00
parent 28951dccf1
commit 64f3969434
3 changed files with 9 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/storage/Backend.cc
View file

@ -62,13 +62,10 @@ OpenResultCallback::OpenResultCallback(IntrusivePtr<zeek::detail::trigger::Trigg
void OpenResultCallback::Complete(const ErrorResult& res) {
zeek::Val* result;
if ( res ) {
if ( res )
result = new StringVal(res.value());
}
else {
storage_mgr->AddBackendToMap(backend->backend);
else
result = backend;
}
ValComplete(result);
}

9
src/storage/Manager.cc
View file

@ -54,8 +54,7 @@ ErrorResult Manager::OpenBackend(BackendPtr backend, RecordValPtr options, TypeP
return util::fmt("Failed to open backend %s: %s", backend->Tag(), res.value().c_str());
}
if ( ! cb )
AddBackendToMap(std::move(backend));
RegisterBackend(std::move(backend));
// TODO: post Storage::backend_opened event
@ -83,16 +82,18 @@ ErrorResult Manager::CloseBackend(BackendPtr backend, ErrorResultCallback* cb) {
void Manager::Expire() {
DBG_LOG(DBG_STORAGE, "Expire running, have %zu backends to check", backends.size());
std::unique_lock<std::mutex> lk(backends_mtx);
for ( const auto& b : backends )
for ( const auto& b : backends ) {
if ( b->IsOpen() )
b->Expire();
}
}
void Manager::StartExpirationTimer() {
zeek::detail::timer_mgr->Add(
new detail::ExpirationTimer(run_state::network_time + zeek::BifConst::Storage::expire_interval));
}
void Manager::AddBackendToMap(BackendPtr backend) {
void Manager::RegisterBackend(BackendPtr backend) {
std::unique_lock<std::mutex> lk(backends_mtx);
backends.push_back(std::move(backend));
}

2
src/storage/Manager.h
View file

@ -71,7 +71,7 @@ protected:
void StartExpirationTimer();
friend class storage::OpenResultCallback;
void AddBackendToMap(BackendPtr backend);
void RegisterBackend(BackendPtr backend);
private:
std::vector<BackendPtr> backends;