dd ctx to Set function

Signed-off-by: alice lyu <[email protected]>

api/client/volume/client.go

@@ -460,8 +460,7 @@ func (v *volumeClient) Unmount(ctx context.Context, volumeID string, mountPath s
 }
 
 // Update volume
-func (v *volumeClient) Set(volumeID string, locator *api.VolumeLocator,
- spec *api.VolumeSpec) error {
+func (v *volumeClient) Set(ctx context.Context, volumeID string, locator *api.VolumeLocator, spec *api.VolumeSpec) error {
  return v.doVolumeSet(correlation.TODO(),
  volumeID,
  &api.VolumeSetRequest{

api/server/middleware_auth.go

@@ -205,7 +205,7 @@ func (a *authMiddleware) setWithAuth(w http.ResponseWriter, r *http.Request, nex
 
  if req.Spec != nil && req.Spec.Size > 0 {
  isOpDone = true
- err = d.Set(volumeID, req.Locator, req.Spec)
+ err = d.Set(ctx, volumeID, req.Locator, req.Spec)
  }
 
  for err == nil && req.Action != nil {

api/server/sdk/volume_ops.go

@@ -729,7 +729,7 @@ func (s *VolumeServer) Update(
  maskUnModified(updatedSpec, req.GetSpec())
 
  // Send to driver
- if err := s.driver(ctx).Set(req.GetVolumeId(), locator, updatedSpec); err != nil {
+ if err := s.driver(ctx).Set(ctx, req.GetVolumeId(), locator, updatedSpec); err != nil {
  return nil, status.Errorf(codes.Internal, "Failed to update volume: %v", err)
  }
 

api/server/volume_test.go

@@ -123,7 +123,7 @@ func TestVolumeNoAuth(t *testing.T) {
 
  newspec := req.GetSpec()
  newspec.Size = newsize
- resp := driverclient.Set(id, req.GetLocator(), newspec)
+ resp := driverclient.Set(context.TODO(), id, req.GetLocator(), newspec)
  assert.Nil(t, resp)
 
  // INSPECT
@@ -793,7 +793,7 @@ func TestVolumeSetSuccess(t *testing.T) {
  Spec: &api.VolumeSpec{Size: newsize},
  }
 
- res := driverclient.Set(id, req.GetLocator(), req2.GetSpec())
+ res := driverclient.Set(context.TODO(), id, req.GetLocator(), req2.GetSpec())
  assert.Nil(t, res)
 
  // Assert volume information is correct
@@ -808,13 +808,11 @@ func TestVolumeSetSuccess(t *testing.T) {
  assert.Equal(t, newsize, r.GetVolume().GetSpec().GetSize())
 
  // Send HA request
- res = driverclient.Set(id,
- nil,
- &api.VolumeSpec{
- HaLevel: 2,
- ReplicaSet: &api.ReplicaSet{Nodes: []string{}},
- SnapshotInterval: math.MaxUint32,
- })
+ res = driverclient.Set(context.TODO(), id, nil, &api.VolumeSpec{
+ HaLevel: 2,
+ ReplicaSet: &api.ReplicaSet{Nodes: []string{}},
+ SnapshotInterval: math.MaxUint32,
+ })
  assert.Nil(t, res, fmt.Sprintf("Error: %v", res))
 
  // Assert volume information is correct
@@ -884,7 +882,7 @@ func TestVolumeSetFailed(t *testing.T) {
  Spec: &api.VolumeSpec{Size: size, HaLevel: halevel},
  }
  // Cannot get this to fail....
- err = driverclient.Set("doesnotexist", req2.GetLocator(), req2.GetSpec())
+ err = driverclient.Set(context.TODO(), "doesnotexist", req2.GetLocator(), req2.GetSpec())
  // assert.NotNil(t, err)
 
  // Assert volume information is correct
@@ -929,7 +927,7 @@ func TestMiddlewareVolumeSetSizeSuccess(t *testing.T) {
 
  // Not setting mock secrets
 
- err = driverclient.Set(id, nil, req.GetSpec())
+ err = driverclient.Set(context.TODO(), id, nil, req.GetSpec())
  assert.NoError(t, err, "Unexpected error on Set")
 
  // Assert volume information is correct
@@ -976,7 +974,7 @@ func TestMiddlewareVolumeSetFailure(t *testing.T) {
 
  // Not setting mock secrets
 
- err = driverclient.Set(id, &api.VolumeLocator{Name: "myvol"}, req.GetSpec())
+ err = driverclient.Set(context.TODO(), id, &api.VolumeLocator{Name: "myvol"}, req.GetSpec())
  assert.Error(t, err, "Unexpected error on Set")
 
 }

pkg/sanity/volume.go

@@ -617,7 +617,7 @@ var _ = Describe("Volume [Volume Tests]", func() {
  },
  }
 
- err = volumedriver.Set(volumeID, set.GetLocator(), set.GetSpec())
+ err = volumedriver.Set(context.TODO(), volumeID, set.GetLocator(), set.GetSpec())
  Expect(err).NotTo(HaveOccurred())
 
  By("Inspecting the volume for new updates")
@@ -679,7 +679,7 @@ var _ = Describe("Volume [Volume Tests]", func() {
  },
  }
 
- err = volumedriver.Set(volumeID, set.Locator, set.Spec)
+ err = volumedriver.Set(context.TODO(), volumeID, set.Locator, set.Spec)
  Expect(err).NotTo(HaveOccurred())
 
  By("Inspecting the volume for new updates")

volume/drivers/buse/buse.go

@@ -362,7 +362,7 @@ func (d *driver) SnapshotGroup(groupID string, labels map[string]string, volumeI
  return nil, volume.ErrNotSupported
 }
 
-func (d *driver) Set(volumeID string, locator *api.VolumeLocator, spec *api.VolumeSpec) error {
+func (d *driver) Set(ctx context.Context, volumeID string, locator *api.VolumeLocator, spec *api.VolumeSpec) error {
  if spec != nil {
  return volume.ErrNotSupported
  }

volume/drivers/fake/fake.go

@@ -336,7 +336,7 @@ func (d *driver) CloudMigrateStatus(request *api.CloudMigrateStatusRequest) (*ap
  }, nil
 }
 
-func (d *driver) Set(volumeID string, locator *api.VolumeLocator, spec *api.VolumeSpec) error {
+func (d *driver) Set(ctx context.Context, volumeID string, locator *api.VolumeLocator, spec *api.VolumeSpec) error {
  v, err := d.GetVol(volumeID)
  if err != nil {
  return err

volume/drivers/fake/fake_test.go

@@ -715,7 +715,7 @@ func TestFakeSet(t *testing.T) {
  assert.NotEmpty(t, volid)
 
  // Set values
- err = d.Set(volid, &api.VolumeLocator{
+ err = d.Set(context.TODO(), volid, &api.VolumeLocator{
  Name: "newname",
  VolumeLabels: map[string]string{
  "hello": "world",

volume/drivers/fuse/volume_driver.go

@@ -180,7 +180,7 @@ func (v *volumeDriver) Unmount(ctx context.Context, volumeID string, mountpath s
  return v.UpdateVol(volume)
 }
 
-func (v *volumeDriver) Set(volumeID string, locator *api.VolumeLocator, spec *api.VolumeSpec) error {
+func (v *volumeDriver) Set(ctx context.Context, volumeID string, locator *api.VolumeLocator, spec *api.VolumeSpec) error {
  return volume.ErrNotSupported
 
 }

volume/drivers/nfs/nfs.go

@@ -789,7 +789,7 @@ func (d *driver) Detach(ctx context.Context, volumeID string, options map[string
  return nil
 }
 
-func (d *driver) Set(volumeID string, locator *api.VolumeLocator, spec *api.VolumeSpec) error {
+func (d *driver) Set(ctx context.Context, volumeID string, locator *api.VolumeLocator, spec *api.VolumeSpec) error {
  if spec != nil {
  return volume.ErrNotSupported
  }

volume/drivers/test/driver.go

@@ -149,7 +149,7 @@ func set(t *testing.T, ctx *Context) {
  require.Equal(t, vols[0].Id, ctx.volID, "Expect volID %v actual %v", ctx.volID, vols[0].Id)
 
  vols[0].Locator.VolumeLabels["UpdateTest"] = "Success"
- err = ctx.Set(ctx.volID, vols[0].Locator, nil)
+ err = ctx.Set(correlation.TODO(), ctx.volID, vols[0].Locator, nil)
  if err != volume.ErrNotSupported {
  require.NoError(t, err, "Failed in Update")
  vols, err = ctx.Inspect(correlation.TODO(), []string{ctx.volID})

volume/drivers/vfs/vfs.go

@@ -171,7 +171,7 @@ func (d *driver) Unmount(ctx context.Context, volumeID string, mountpath string,
  return d.UpdateVol(v)
 }
 
-func (d *driver) Set(volumeID string, locator *api.VolumeLocator, spec *api.VolumeSpec) error {
+func (d *driver) Set(ctx context.Context, volumeID string, locator *api.VolumeLocator, spec *api.VolumeSpec) error {
  if spec != nil {
  return volume.ErrNotSupported
  }

volume/volume.go

@@ -291,7 +291,7 @@ type ProtoDriver interface {
  Unmount(ctx context.Context, volumeID string, mountPath string, options map[string]string) error
  // Update not all fields of the spec are supported, ErrNotSupported will be thrown for unsupported
  // updates.
- Set(volumeID string, locator *api.VolumeLocator, spec *api.VolumeSpec) error
+ Set(ctx context.Context, volumeID string, locator *api.VolumeLocator, spec *api.VolumeSpec) error
  // Status returns a set of key-value pairs which give low
  // level diagnostic status about this driver.
  Status() [][2]string