@@ -33,6 +33,7 @@ import Share.Web.UCM.Sync.HashJWT qualified as HashJWT
3333import Share.Web.UCM.SyncV2.Queries qualified as SSQ
3434import U.Codebase.Sqlite.Orphans ()
3535import Unison.Debug qualified as Debug
36+ import Unison.Hash32 (Hash32 )
3637import Unison.Share.API.Hash (HashJWTClaims (.. ))
3738import Unison.SyncV2.API qualified as SyncV2
3839import Unison.SyncV2.Types (DownloadEntitiesChunk (.. ), EntityChunk (.. ), ErrorChunk (.. ), StreamInitInfo (.. ))
@@ -43,8 +44,8 @@ import UnliftIO.Async qualified as Async
4344batchSize :: Int32
4445batchSize = 1000
4546
46- streamSettings :: StreamInitInfo
47- streamSettings = StreamInitInfo {version = SyncV2. Version 1 , entitySorting = SyncV2. Unsorted , numEntities = Nothing }
47+ streamSettings :: Hash32 -> Maybe SyncV2. BranchRef -> StreamInitInfo
48+ streamSettings rootCausalHash rootBranchRef = StreamInitInfo {version = SyncV2. Version 1 , entitySorting = SyncV2. Unsorted , numEntities = Nothing , rootCausalHash, rootBranchRef }
4849
4950server :: Maybe UserId -> SyncV2. Routes WebAppServer
5051server mayUserId =
@@ -90,7 +91,7 @@ downloadEntitiesStreamImpl mayCallerUserId (SyncV2.DownloadEntitiesRequest {caus
9091 pure $ Codebase. codebaseEnv authZToken codebaseLoc
9192 q <- UnliftIO. atomically $ do
9293 q <- STM. newTBMQueue 10
93- STM. writeTBMQueue q (NEL. singleton $ InitialC $ streamSettings)
94+ STM. writeTBMQueue q (NEL. singleton $ InitialC $ streamSettings causalHash ( Just branchRef) )
9495 pure q
9596 streamResults <- lift $ UnliftIO. toIO do
9697 Logging. logInfoText " Starting download entities stream"
0 commit comments