fix: fix update network metrics not updated

Signed-off-by: Sacha Froment <sfroment42@gmail.com>
berty · Feb 1, 2019 · f01552c · f01552c
1 parent 3f05b5f
commit f01552c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/core/manager/account/p2p.go b/core/manager/account/p2p.go
@@ -156,5 +156,7 @@ func (a *Account) UpdateP2PNetwork(ctx context.Context, opts *P2PNetworkOptions)
 		return err
 	}
 
+	a.node.UseNetworkMetrics(ctx, a.metrics)
+
 	return a.node.UseNetworkDriver(ctx, a.network)
 }
diff --git a/core/node/network.go b/core/node/network.go
@@ -21,6 +21,14 @@ func WithNetworkMetrics(metrics network.Metrics) NewNodeOption {
 	}
 }
 
+func (n *Node) UseNetworkMetrics(ctx context.Context, metrics network.Metrics) {
+	tracer := tracing.EnterFunc(ctx, metrics)
+	defer tracer.Finish()
+	ctx = tracer.Context()
+
+	n.networkMetrics = metrics
+}
+
 func (n *Node) UseNetworkDriver(ctx context.Context, driver network.Driver) error {
 	tracer := tracing.EnterFunc(ctx, driver)
 	defer tracer.Finish()