tag snapshots at creation (#145)

* tag snapshots at creation
digitalocean · Apr 25, 2019 · 07f200d · 07f200d
1 parent ed8c3e7
commit 07f200d
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/driver/controller.go b/driver/controller.go
@@ -631,11 +631,16 @@ func (d *Driver) CreateSnapshot(ctx context.Context, req *csi.CreateSnapshotRequ
 		}
 	}
 
-	snap, resp, err := d.storage.CreateSnapshot(ctx, &godo.SnapshotCreateRequest{
+	snapReq := &godo.SnapshotCreateRequest{
 		VolumeID:    req.GetSourceVolumeId(),
 		Name:        req.GetName(),
 		Description: createdByDO,
-	})
+	}
+	if d.doTag != "" {
+		snapReq.Tags = append(snapReq.Tags, d.doTag)
+	}
+
+	snap, resp, err := d.storage.CreateSnapshot(ctx, snapReq)
 	if err != nil {
 		if resp != nil && resp.StatusCode == http.StatusConflict {
 			// 409 is returned when we try to snapshot a volume with the same