fix: wait for full startup of data transfer when making client

filecoin-project · Dec 19, 2023 · e7ab797 · e7ab797
1 parent 5e9f939
commit e7ab797
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/retrievalmarket/client/client.go b/retrievalmarket/client/client.go
@@ -161,10 +161,27 @@ func NewClientWithConfig(cfg *Config) (*Client, error) {
 		}
 	}
 
+	errCh := make(chan error)
+	startedCh := make(chan struct{})
+
+	mgr.OnReady(func(err error) {
+		if err != nil {
+			errCh <- err
+			return
+		}
+		close(startedCh)
+	})
+
 	if err := mgr.Start(context.Background()); err != nil {
 		return nil, err
 	}
 
+	select {
+	case <-startedCh:
+	case err := <-errCh:
+		return nil, err
+	}
+
 	c := &Client{
 		host:                       cfg.Host,
 		api:                        cfg.Api,