From 2d6da0352057bd5ef2099cd65fd01eb091612ad8 Mon Sep 17 00:00:00 2001
From: Vicki Pfau <vi@endrift.com>
Date: Thu, 13 Jun 2024 13:12:19 -0700
Subject: [PATCH] daemon/root: Move ftrace initialization

---
 src/daemon/root.rs | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/daemon/root.rs b/src/daemon/root.rs
index cb33bc4..b872757 100644
--- a/src/daemon/root.rs
+++ b/src/daemon/root.rs
@@ -119,6 +119,11 @@ impl DaemonContext for RootContext {
         daemon: &mut Daemon<RootContext>,
     ) -> Result<()> {
         self.state = state;
+
+        let connection = daemon.get_connection();
+        let ftrace = Ftrace::init(connection).await?;
+        daemon.add_service(ftrace);
+
         self.reload_ds_inhibit(daemon).await?;
 
         Ok(())
@@ -187,8 +192,5 @@ pub async fn daemon() -> Result<()> {
     let context = RootContext::new(tx);
     let mut daemon = Daemon::new(subscriber, connection.clone(), rx).await?;
 
-    let ftrace = Ftrace::init(connection).await?;
-    daemon.add_service(ftrace);
-
     daemon.run(context).await
 }