Fix conditional write when generation = 0.

transparency-dev · Dec 14, 2023 · c5b0dcc · c5b0dcc
1 parent d0688e1
commit c5b0dcc
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/experimental/gcp-log/internal/storage/storage.go b/experimental/gcp-log/internal/storage/storage.go
@@ -126,7 +126,14 @@ func (c *Client) WriteCheckpoint(ctx context.Context, newCPRaw []byte) error {
 	bkt := c.gcsClient.Bucket(c.bucket)
 	obj := bkt.Object(layout.CheckpointPath)
 
-	w := obj.If(gcs.Conditions{GenerationMatch: c.checkpointGen}).NewWriter(ctx)
+	var cond gcs.Conditions
+	if c.checkpointGen == 0 {
+		cond = gcs.Conditions{DoesNotExist: true}
+	} else {
+		cond = gcs.Conditions{GenerationMatch: c.checkpointGen}
+	}
+
+	w := obj.If(cond).NewWriter(ctx)
 	if _, err := w.Write(newCPRaw); err != nil {
 		return err
 	}