Skip to content

Commit 67b9f5f

Browse files
committed
[server] Pass "useRetryClient" on all requests to image-builder
1 parent 22640fa commit 67b9f5f

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
@@ -1238,11 +1238,19 @@ export class WorkspaceStarter {
12381238
additionalAuth,
12391239
);
12401240

1241+
// Resolve feature flag with user context
1242+
const useRetryClient = await getExperimentsClientForBackend().getValueAsync(
1243+
"imagebuilder_retry_resolve",
1244+
false,
1245+
{ user },
1246+
);
1247+
12411248
const req = new BuildRequest();
12421249
req.setSource(src);
12431250
req.setAuth(auth);
12441251
req.setForceRebuild(forceRebuild);
12451252
req.setTriggeredBy(user.id);
1253+
req.setUseRetryClient(useRetryClient);
12461254
if (!ignoreBaseImageresolvedAndRebuildBase && !forceRebuild && workspace.baseImageNameResolved) {
12471255
req.setBaseImageNameResolved(workspace.baseImageNameResolved);
12481256
}
@@ -2068,8 +2076,16 @@ export class WorkspaceStarter {
20682076
region?: WorkspaceRegion,
20692077
organizationId?: string,
20702078
) {
2079+
// Resolve feature flag with user context
2080+
const useRetryClient = await getExperimentsClientForBackend().getValueAsync(
2081+
"imagebuilder_retry_resolve",
2082+
false,
2083+
{ user },
2084+
);
2085+
20712086
const req = new ResolveBaseImageRequest();
20722087
req.setRef(imageRef);
2088+
req.setUseRetryClient(useRetryClient);
20732089
const allowAll = new BuildRegistryAuthTotal();
20742090
allowAll.setAllowAll(true);
20752091
const auth = new BuildRegistryAuth();

0 commit comments

Comments
 (0)