From 27f689959decd391b047c8034d481267d500549e Mon Sep 17 00:00:00 2001 From: Taras Madan Date: Thu, 15 May 2025 15:01:02 +0200 Subject: vm: func Run accepts context It allows to use context as a single termination signal source. --- pkg/instance/execprog.go | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'pkg/instance') diff --git a/pkg/instance/execprog.go b/pkg/instance/execprog.go index 4fd656fc5..83461d07e 100644 --- a/pkg/instance/execprog.go +++ b/pkg/instance/execprog.go @@ -4,6 +4,7 @@ package instance import ( + "context" "fmt" "os" "time" @@ -123,7 +124,9 @@ func (inst *ExecProgInstance) runCommand(command string, duration time.Duration, if inst.BeforeContextLen != 0 { opts = append(opts, vm.OutputSize(inst.BeforeContextLen)) } - output, rep, err := inst.VMInstance.Run(duration, inst.reporter, command, opts...) + ctxTimeout, cancel := context.WithTimeout(context.Background(), duration) + defer cancel() + output, rep, err := inst.VMInstance.Run(ctxTimeout, inst.reporter, command, opts...) if err != nil { return nil, fmt.Errorf("failed to run command in VM: %w", err) } -- cgit mrf-deployment