1 files changed, 6 insertions, 1 deletions
diff --git a/rpkid/rpki/irdbd.py b/rpkid/rpki/irdbd.py
index 60f6a549..a0fc0ec1 100644
--- a/rpkid/rpki/irdbd.py
+++ b/rpkid/rpki/irdbd.py
@@ -174,7 +174,12 @@ class main(object):
 
     if profile:
       import cProfile
-      cProfile.run("self.main()", profile)
+      prof = cProfile.Profile()
+      try:
+        prof.runcall(self.main)
+      finally:
+        prof.dump_stats(profile)
+        rpki.log.info("Dumped profile data to %s" % profile)
     else:
       self.main()