Add retention days to volume snapshots (#55)

superfly · May 8, 2024 · 9c32a87 · 9c32a87
1 parent acab0fd
commit 9c32a87
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 10 deletions.
diff --git a/resource_volumes.go b/resource_volumes.go
@@ -68,6 +68,7 @@ query GetSnapshotsFromVolume($id: ID!) {
 					size
 					digest
 					createdAt
+					retentionDays
 				}
 			}
 		}
@@ -86,6 +87,7 @@ query GetSnapshotsFromVolume($id: ID!) {
 	}
 	var snapshots []VolumeSnapshot
 	for _, snapshot := range data.Volume.Snapshots.Nodes {
+		snapshot.Status = "created"
 		snapshots = append(snapshots, NewVolumeSnapshotFrom(snapshot))
 	}
 	return snapshots, nil

diff --git a/volume_types.go b/volume_types.go
@@ -56,19 +56,21 @@ type UpdateVolumeRequest struct {
 }
 
 type VolumeSnapshot struct {
-	ID        string    `json:"id"`
-	Size      int       `json:"size"`
-	Digest    string    `json:"digest"`
-	CreatedAt time.Time `json:"created_at"`
-	Status    string    `json:"status"`
+	ID            string    `json:"id"`
+	Size          int       `json:"size"`
+	Digest        string    `json:"digest"`
+	CreatedAt     time.Time `json:"created_at"`
+	Status        string    `json:"status"`
+	RetentionDays *int      `json:"retention_days"`
 }
 
 type VolumeSnapshotGql struct {
-	ID        string    `json:"id"`
-	Size      string    `json:"size"`
-	Digest    string    `json:"digest"`
-	CreatedAt time.Time `json:"createdAt"`
-	Status    string    `json:"status"`
+	ID            string    `json:"id"`
+	Size          string    `json:"size"`
+	Digest        string    `json:"digest"`
+	CreatedAt     time.Time `json:"createdAt"`
+	Status        string    `json:"status"`
+	RetentionDays *int      `json:"retentionDays"`
 }
 
 func NewVolumeSnapshotFrom(v VolumeSnapshotGql) VolumeSnapshot {