From 0db2ff34096896a27c1f2239fecfae4f5c32b744 Mon Sep 17 00:00:00 2001 From: Přemysl Eric Janouch Date: Fri, 19 Apr 2024 04:26:48 +0200 Subject: Set a time limit on runners --- acid.go | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) (limited to 'acid.go') diff --git a/acid.go b/acid.go index 17ca23b..b206057 100644 --- a/acid.go +++ b/acid.go @@ -81,8 +81,9 @@ type ConfigProject struct { } type ConfigProjectRunner struct { - Setup string `yaml:"setup"` // project setup script (SSH) - Build string `yaml:"build"` // project build script (SSH) + Setup string `yaml:"setup"` // project setup script (SSH) + Build string `yaml:"build"` // project build script (SSH) + Timeout string `yaml:"timeout"` // timeout duration } func parseConfig(path string) error { @@ -937,6 +938,17 @@ func executorRunTask(ctx context.Context, task Task) error { return fmt.Errorf("script: %w", err) } + // Lenient or not, some kind of a time limit is desirable. + timeout := time.Hour + if rt.ProjectRunner.Timeout != "" { + timeout, err = time.ParseDuration(rt.ProjectRunner.Timeout) + if err != nil { + return fmt.Errorf("timeout: %w", err) + } + } + ctx, cancelTimeout := context.WithTimeout(ctx, timeout) + defer cancelTimeout() + privateKey, err := os.ReadFile(rt.Runner.SSH.Identity) if err != nil { return fmt.Errorf( -- cgit v1.2.3-70-g09d2