Merge pull request #634 from input-output-hk/jhbertra/fix-sync-connec…

…tion Connect to chain sync query port with traced client
input-output-hk · Jun 26, 2023 · 6966b4d · 6966b4d
2 parents 9e412eb + 1c2dfbb
commit 6966b4d
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/marlowe-runtime/marlowe-sync/Main.hs b/marlowe-runtime/marlowe-sync/Main.hs
@@ -26,8 +26,8 @@ import Language.Marlowe.Runtime.Sync (MarloweSync (..), SyncDependencies (..), s
 import Language.Marlowe.Runtime.Sync.Database (hoistDatabaseQueries, logDatabaseQueries)
 import qualified Language.Marlowe.Runtime.Sync.Database.PostgreSQL as Postgres
 import Logging (RootSelector (..), renderRootSelectorOTel)
-import Network.Protocol.Driver (TcpServerDependencies (..), tcpClient)
-import Network.Protocol.Driver.Trace (tcpServerTraced)
+import Network.Protocol.Driver (TcpServerDependencies (..))
+import Network.Protocol.Driver.Trace (tcpClientTraced, tcpServerTraced)
 import Network.Protocol.Query.Client (queryClientPeer)
 import Network.Protocol.Query.Server (queryServerPeer)
 import Network.Socket (HostName, PortNumber)
@@ -79,7 +79,8 @@ run Options{..} = bracket (Pool.acquire 100 (Just 5000000) (fromString databaseU
                       (either throwIO pure <=< liftIO . Pool.use pool)
                       Postgres.databaseQueries
               , runtimeVersion = version
-              , chainSyncQueryConnector = tcpClient chainSyncHost chainQueryPort queryClientPeer
+              , chainSyncQueryConnector =
+                  tcpClientTraced (injectSelector ChainSyncQueryClient) chainSyncHost chainQueryPort queryClientPeer
               }
 
       tcpServerTraced "marlowe-sync" (injectSelector MarloweSyncServer)