Update eclair-core/src/main/scala/fr/acinq/eclair/Eclair.scala

Co-authored-by: Pierre-Marie Padiou <pm47@users.noreply.github.com>
ACINQ · Nov 26, 2021 · 4c8bd39 · 4c8bd39
1 parent 42e3483
commit 4c8bd39
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/eclair-core/src/main/scala/fr/acinq/eclair/Eclair.scala b/eclair-core/src/main/scala/fr/acinq/eclair/Eclair.scala
@@ -165,8 +165,12 @@ class EclairImpl(appKit: Kit) extends Eclair with Logging {
   private val externalIdMaxLength = 66
 
   override def connect(target: Either[NodeURI, PublicKey])(implicit timeout: Timeout): Future[String] = target match {
-    case Left(uri) => appKit.switchboard.toTyped.ask((ref: typed.ActorRef[PeerConnection.ConnectionResult]) => Peer.Connect(uri, ref)).map(_.toString)
-    case Right(pubKey) => appKit.switchboard.toTyped.ask((ref: typed.ActorRef[PeerConnection.ConnectionResult]) => Peer.Connect(pubKey, None, ref)).map(_.toString)
+    implicit val typedScheduler: Scheduler = appKit.system.scheduler.toTyped
+    appKit.switchboard.toTyped
+      .ask[PeerConnection.ConnectionResult](ref => target.fold(
+        uri => Peer.Connect(uri, ref),
+        nodeId => Peer.Connect(nodeId, None, ref))
+      ).map(_.toString)
   }
 
   override def disconnect(nodeId: PublicKey)(implicit timeout: Timeout): Future[String] = {