aboutsummaryrefslogtreecommitdiffhomepage
path: root/infra/bots/gen_tasks.go
diff options
context:
space:
mode:
authorGravatar Ben Wagner <benjaminwagner@google.com>2018-04-17 12:45:29 -0400
committerGravatar Skia Commit-Bot <skia-commit-bot@chromium.org>2018-04-17 16:56:24 +0000
commit053d04697acd74c45883058e81e422d731f06a0f (patch)
tree46dafd067f2cdf66df924d46a1c03382a6252290 /infra/bots/gen_tasks.go
parent1cd6098d52a531451f194437861cd7a548f825a7 (diff)
[infra] Add service account override in gen_tasks.go.
This CL is a prerequisite for re-landing https://skia-review.googlesource.com/c/skia/+/121481 Bug: skia:7050 No-Try: true Change-Id: Ice9d40c1df75734c1ec429be9cf4605a6098ce80 Reviewed-on: https://skia-review.googlesource.com/121490 Commit-Queue: Eric Boren <borenet@google.com> Auto-Submit: Ben Wagner <benjaminwagner@google.com> Reviewed-by: Eric Boren <borenet@google.com>
Diffstat (limited to 'infra/bots/gen_tasks.go')
-rw-r--r--infra/bots/gen_tasks.go5
1 files changed, 5 insertions, 0 deletions
diff --git a/infra/bots/gen_tasks.go b/infra/bots/gen_tasks.go
index a52b4b282e..4191ff4643 100644
--- a/infra/bots/gen_tasks.go
+++ b/infra/bots/gen_tasks.go
@@ -68,6 +68,11 @@ var (
Pool string `json:"pool"`
}
+ // alternateServiceAccount can be set in an init function to override the normal service accounts.
+ // Takes one of SERVICE_ACCOUNT_* constants as an argument and returns the service account that
+ // should be used, or uses sklog.Fatal to indicate a problem.
+ alternateServiceAccount func(serviceAccountEnum string) string
+
// alternateSwarmDimensions can be set in an init function to override the default swarming bot
// dimensions for the given task.
alternateSwarmDimensions func(parts map[string]string) []string