Skip to content
This repository was archived by the owner on Oct 9, 2023. It is now read-only.

Commit ff9f55b

Browse files
authored
Make rendering template in agent client temporary (#395)
Signed-off-by: Edwin Yu <[email protected]>
1 parent 8eddca3 commit ff9f55b

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

go/tasks/plugins/webapi/agent/plugin.go

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,13 +68,15 @@ func (p Plugin) Create(ctx context.Context, taskCtx webapi.TaskExecutionContextR
6868
return nil, nil, err
6969
}
7070

71+
var argTemplate []string
7172
if taskTemplate.GetContainer() != nil {
7273
templateParameters := template.Parameters{
7374
TaskExecMetadata: taskCtx.TaskExecutionMetadata(),
7475
Inputs: taskCtx.InputReader(),
7576
OutputPath: taskCtx.OutputWriter(),
7677
Task: taskCtx.TaskReader(),
7778
}
79+
argTemplate = taskTemplate.GetContainer().Args
7880
modifiedArgs, err := template.Render(ctx, taskTemplate.GetContainer().Args, templateParameters)
7981
if err != nil {
8082
return nil, nil, err
@@ -101,6 +103,11 @@ func (p Plugin) Create(ctx context.Context, taskCtx webapi.TaskExecutionContextR
101103
return nil, nil, err
102104
}
103105

106+
// Restore unrendered template for subsequent renders.
107+
if taskTemplate.GetContainer() != nil {
108+
taskTemplate.GetContainer().Args = argTemplate
109+
}
110+
104111
return &ResourceMetaWrapper{
105112
OutputPrefix: outputPrefix,
106113
AgentResourceMeta: res.GetResourceMeta(),

0 commit comments

Comments
 (0)