runtime: fix per cpu map low performance (#255)

* add static to ensure * fix
eunomia-bpf · Mar 23, 2024 · 3a831ac · 3a831ac
1 parent 9b0cc4e
commit 3a831ac
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/runtime/src/bpf_map/map_common_def.hpp b/runtime/src/bpf_map/map_common_def.hpp
@@ -23,11 +23,15 @@ using bytes_vec = boost::interprocess::vector<uint8_t, bytes_vec_allocator>;
 template <class T>
 static inline T ensure_on_current_cpu(std::function<T(int cpu)> func)
 {
+	static thread_local int currcpu = -1;
+	if (currcpu == sched_getcpu()) {
+		return func(currcpu);
+	}
 	cpu_set_t orig, set;
 	CPU_ZERO(&orig);
 	CPU_ZERO(&set);
 	sched_getaffinity(0, sizeof(orig), &orig);
-	int currcpu = sched_getcpu();
+	currcpu = sched_getcpu();
 	CPU_SET(currcpu, &set);
 	sched_setaffinity(0, sizeof(set), &set);
 	T ret = func(currcpu);
@@ -38,6 +42,10 @@ static inline T ensure_on_current_cpu(std::function<T(int cpu)> func)
 template <class T>
 static inline T ensure_on_certain_cpu(int cpu, std::function<T()> func)
 {
+	static thread_local int currcpu = -1;
+	if (currcpu == sched_getcpu()) {
+		return func(currcpu);
+	}
 	cpu_set_t orig, set;
 	CPU_ZERO(&orig);
 	CPU_ZERO(&set);

diff --git a/runtime/src/bpf_map/userspace/prog_array.hpp b/runtime/src/bpf_map/userspace/prog_array.hpp
@@ -23,7 +23,7 @@ class prog_array_map_impl {
 	i32_vec data;
 
     public:
-	const static bool should_lock = true;
+	const static bool should_lock = false;
 	prog_array_map_impl(boost::interprocess::managed_shared_memory &memory,
 			    uint32_t key_size, uint32_t value_size,
 			    uint32_t max_entries);