1 // Copyright (C) The Arvados Authors. All rights reserved.
3 // SPDX-License-Identifier: AGPL-3.0
11 "git.curoverse.com/arvados.git/lib/cloud"
12 "git.curoverse.com/arvados.git/lib/cloud/azure"
13 "git.curoverse.com/arvados.git/lib/cloud/ec2"
14 "git.curoverse.com/arvados.git/sdk/go/arvados"
15 "github.com/sirupsen/logrus"
16 "golang.org/x/crypto/ssh"
19 var drivers = map[string]cloud.Driver{
20 "azure": azure.Driver,
24 func newInstanceSet(cluster *arvados.Cluster, setID cloud.InstanceSetID, logger logrus.FieldLogger) (cloud.InstanceSet, error) {
25 driver, ok := drivers[cluster.Containers.CloudVMs.Driver]
27 return nil, fmt.Errorf("unsupported cloud driver %q", cluster.Containers.CloudVMs.Driver)
29 is, err := driver.InstanceSet(cluster.Containers.CloudVMs.DriverParameters, setID, logger)
30 if maxops := cluster.Containers.CloudVMs.MaxCloudOpsPerSecond; maxops > 0 {
31 is = &rateLimitedInstanceSet{
33 ticker: time.NewTicker(time.Second / time.Duration(maxops)),
36 is = defaultTaggingInstanceSet{
38 defaultTags: cloud.InstanceTags(cluster.Containers.CloudVMs.ResourceTags),
43 type rateLimitedInstanceSet struct {
48 func (is rateLimitedInstanceSet) Create(it arvados.InstanceType, image cloud.ImageID, tags cloud.InstanceTags, init cloud.InitCommand, pk ssh.PublicKey) (cloud.Instance, error) {
50 inst, err := is.InstanceSet.Create(it, image, tags, init, pk)
51 return &rateLimitedInstance{inst, is.ticker}, err
54 type rateLimitedInstance struct {
59 func (inst *rateLimitedInstance) Destroy() error {
61 return inst.Instance.Destroy()
64 // Adds the specified defaultTags to every Create() call.
65 type defaultTaggingInstanceSet struct {
67 defaultTags cloud.InstanceTags
70 func (is defaultTaggingInstanceSet) Create(it arvados.InstanceType, image cloud.ImageID, tags cloud.InstanceTags, init cloud.InitCommand, pk ssh.PublicKey) (cloud.Instance, error) {
71 allTags := cloud.InstanceTags{}
72 for k, v := range is.defaultTags {
75 for k, v := range tags {
78 return is.InstanceSet.Create(it, image, allTags, init, pk)