diff mbox series

process: Add profile logging for main loop

Message ID 20240209235615.3473502-1-richard.purdie@linuxfoundation.org
State Accepted, archived
Commit d8d5cd43a60560f67e86f4f625113b0f73b944c0
Headers show
Series process: Add profile logging for main loop | expand

Commit Message

Richard Purdie Feb. 9, 2024, 11:56 p.m. UTC 
When the idle/main loop was added, we didn't include profiling information
for it. There is a performance issue in there, add logging for it.

Signed-off-by: Richard Purdie <richard.purdie@linuxfoundation.org>
---
 lib/bb/server/process.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)
diff mbox series

Patch

diff --git a/lib/bb/server/process.py b/lib/bb/server/process.py
index 6d77ce4786..1ca8e02ba0 100644
--- a/lib/bb/server/process.py
+++ b/lib/bb/server/process.py
@@ -402,6 +402,22 @@  class ProcessServer():
                 serverlog("".join(msg))
 
     def idle_thread(self):
+        if self.cooker.configuration.profile:
+            try:
+                import cProfile as profile
+            except:
+                import profile
+            prof = profile.Profile()
+
+            ret = profile.Profile.runcall(prof, self.idle_thread_internal)
+
+            prof.dump_stats("profile-mainloop.log")
+            bb.utils.process_profilelog("profile-mainloop.log")
+            serverlog("Raw profiling information saved to profilemainloop.log and processed statistics to profilemainloop.log.processed")
+        else:
+            self.idle_thread_internal()
+
+    def idle_thread_internal(self):
         def remove_idle_func(function):
             with bb.utils.lock_timeout(self._idlefuncsLock):
                 del self._idlefuns[function]