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

Add cpu affinity option to supervised node config

Browse source
This commit is contained in:
Jon Siwek 2020-01-10 18:25:42 -08:00
parent 00cd04b0ae
commit 263a5f404a
7 changed files with 94 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/main.cc
View file

@ -78,6 +78,8 @@ extern "C" {
#include "setsignal.h"
};
#include "zeek-affinity.h"
#ifdef USE_PERFTOOLS_DEBUG
HeapLeakChecker* heap_checker = 0;
int perftools_leaks = 0;
@ -944,7 +946,7 @@ int main(int argc, char** argv)
{
if ( setenv("CLUSTER_NODE", node_name.data(), true) == -1 )
{
fprintf(stderr, "cluster node %s failed to setenv: %s\n",
fprintf(stderr, "node '%s' failed to setenv: %s\n",
node_name.data(), strerror(errno));
exit(1);
}
@ -954,7 +956,7 @@ int main(int argc, char** argv)
{
if ( chdir(zeek::supervised_node->directory->data()) )
{
fprintf(stderr, "supervised node %s failed to chdir to %s: %s\n",
fprintf(stderr, "node '%s' failed to chdir to %s: %s\n",
node_name.data(),
zeek::supervised_node->directory->data(),
strerror(errno));
@ -962,6 +964,15 @@ int main(int argc, char** argv)
}
}
if ( zeek::supervised_node->cpu_affinity )
{
auto res = zeek::set_affinity(*zeek::supervised_node->cpu_affinity);
if ( ! res )
fprintf(stderr, "node '%s' failed to set CPU affinity: %s\n",
node_name.data(), strerror(errno));
}
for ( const auto& s : zeek::supervised_node->scripts )
options.scripts_to_load.emplace_back(s);
}