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)),
39 type rateLimitedInstanceSet struct {
44 func (is rateLimitedInstanceSet) Create(it arvados.InstanceType, image cloud.ImageID, tags cloud.InstanceTags, init cloud.InitCommand, pk ssh.PublicKey) (cloud.Instance, error) {
46 inst, err := is.InstanceSet.Create(it, image, tags, init, pk)
47 return &rateLimitedInstance{inst, is.ticker}, err
50 type rateLimitedInstance struct {
55 func (inst *rateLimitedInstance) Destroy() error {
57 return inst.Instance.Destroy()