14324: Azure driver for crunch-dispatch-cloud

[arvados.git] / lib / dispatchcloud / instance_set_proxy.go

diff --git a/lib/dispatchcloud/instance_set_proxy.go b/lib/dispatchcloud/instance_set_proxy.go
index e728b67cd21b013bb4b75649bc8505f21be6312a..80c5104580d9f44de9f392d197839c4b9c06a21c 100644 (file)
--- a/lib/dispatchcloud/instance_set_proxy.go
+++ b/lib/dispatchcloud/instance_set_proxy.go
@@ -5,6 +5,8 @@
 package dispatchcloud
 
 import (
+       "context"
+
        "git.curoverse.com/arvados.git/lib/cloud"
        "git.curoverse.com/arvados.git/sdk/go/arvados"
        "golang.org/x/crypto/ssh"
@@ -14,12 +16,12 @@ type instanceSetProxy struct {
        cloud.InstanceSet
 }
 
-func (is *instanceSetProxy) Create(it arvados.InstanceType, id cloud.ImageID, tags cloud.InstanceTags, pk ssh.PublicKey) (cloud.Instance, error) {
+func (is *instanceSetProxy) Create(ctx context.Context, it arvados.InstanceType, id cloud.ImageID, tags cloud.InstanceTags, pk ssh.PublicKey) (cloud.Instance, error) {
        // TODO: return if Create failed recently with a RateLimitError or QuotaError
-       return is.InstanceSet.Create(it, id, tags, pk)
+       return is.InstanceSet.Create(ctx, it, id, tags, pk)
 }
 
-func (is *instanceSetProxy) Instances(tags cloud.InstanceTags) ([]cloud.Instance, error) {
+func (is *instanceSetProxy) Instances(ctx context.Context, tags cloud.InstanceTags) ([]cloud.Instance, error) {
        // TODO: return if Instances failed recently with a RateLimitError
-       return is.InstanceSet.Instances(tags)
+       return is.InstanceSet.Instances(ctx, tags)
 }