Skip to content

Commit df93f6a

Browse files
committed
[server] Pass "useRetryClient" on all requests to image-builder
1 parent f911fb2 commit df93f6a

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

components/server/src/workspace/workspace-starter.ts

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1242,11 +1242,19 @@ export class WorkspaceStarter {
12421242
additionalAuth,
12431243
);
12441244

1245+
// Resolve feature flag with user context
1246+
const useRetryClient = await getExperimentsClientForBackend().getValueAsync(
1247+
"imagebuilder_retry_resolve",
1248+
false,
1249+
{ user },
1250+
);
1251+
12451252
const req = new BuildRequest();
12461253
req.setSource(src);
12471254
req.setAuth(auth);
12481255
req.setForceRebuild(forceRebuild);
12491256
req.setTriggeredBy(user.id);
1257+
req.setUseRetryClient(useRetryClient);
12501258
if (!ignoreBaseImageresolvedAndRebuildBase && !forceRebuild && workspace.baseImageNameResolved) {
12511259
req.setBaseImageNameResolved(workspace.baseImageNameResolved);
12521260
}
@@ -2078,8 +2086,16 @@ export class WorkspaceStarter {
20782086
region?: WorkspaceRegion,
20792087
organizationId?: string,
20802088
) {
2089+
// Resolve feature flag with user context
2090+
const useRetryClient = await getExperimentsClientForBackend().getValueAsync(
2091+
"imagebuilder_retry_resolve",
2092+
false,
2093+
{ user },
2094+
);
2095+
20812096
const req = new ResolveBaseImageRequest();
20822097
req.setRef(imageRef);
2098+
req.setUseRetryClient(useRetryClient);
20832099
const allowAll = new BuildRegistryAuthTotal();
20842100
allowAll.setAllowAll(true);
20852101
const auth = new BuildRegistryAuth();

0 commit comments

Comments
 (0)