deals: Better client deal tracking

2019-09-13 19:59:10 +02:00
parent bfd4624712
commit 72a406ec7e
8 changed files with 305 additions and 115 deletions
@@ -59,10 +59,12 @@ func HandleIncomingMessages(mctx helpers.MetricsCtx, lc fx.Lifecycle, pubsub *pu
 	go sub.HandleIncomingMessages(ctx, mpool, msgsub)
 }

-func RunDealClient(lc fx.Lifecycle, c *deals.Client) {
+func RunDealClient(mctx helpers.MetricsCtx, lc fx.Lifecycle, c *deals.Client) {
+	ctx := helpers.LifecycleCtx(mctx, lc)
+
 	lc.Append(fx.Hook{
 		OnStart: func(context.Context) error {
-			c.Run()
+			c.Run(ctx)
 			return nil
 		},
 		OnStop: func(context.Context) error {