Skip to content

Commit 7938f89

Browse files
committed
Fix up stream init settings
1 parent a14b424 commit 7938f89

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/Share/Web/UCM/SyncV2/Impl.hs

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@ import Share.Web.UCM.Sync.HashJWT qualified as HashJWT
3333
import Share.Web.UCM.SyncV2.Queries qualified as SSQ
3434
import U.Codebase.Sqlite.Orphans ()
3535
import Unison.Debug qualified as Debug
36+
import Unison.Hash32 (Hash32)
3637
import Unison.Share.API.Hash (HashJWTClaims (..))
3738
import Unison.SyncV2.API qualified as SyncV2
3839
import Unison.SyncV2.Types (DownloadEntitiesChunk (..), EntityChunk (..), ErrorChunk (..), StreamInitInfo (..))
@@ -43,8 +44,8 @@ import UnliftIO.Async qualified as Async
4344
batchSize :: Int32
4445
batchSize = 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

4950
server :: Maybe UserId -> SyncV2.Routes WebAppServer
5051
server 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

Comments
 (0)