Skip to content

Commit 9ec9da0

Browse files
committed
fix: update DeletionEvent to retrieve MCPUsage before updating deletion timestamp
1 parent 76c2c7c commit 9ec9da0

File tree

2 files changed

+9
-11
lines changed

2 files changed

+9
-11
lines changed

VERSION

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
v0.0.13-dev
1+
v0.0.14

internal/usage/tracking.go

Lines changed: 8 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -168,19 +168,17 @@ func (u *UsageTracker) DeletionEvent(ctx context.Context, project string, worksp
168168
return fmt.Errorf("error getting object key: %w", err)
169169
}
170170

171-
var mcpUsage = v1.MCPUsage{
172-
ObjectMeta: metav1.ObjectMeta{
173-
Name: objectKey.Name,
174-
Namespace: objectKey.Namespace,
175-
},
176-
Spec: v1.MCPUsageSpec{
177-
MCPDeletedAt: metav1.NewTime(time.Now().UTC()),
178-
},
179-
}
180-
err = u.client.Patch(ctx, &mcpUsage, client.Merge)
171+
var mcpUsage v1.MCPUsage
172+
err = u.client.Get(ctx, objectKey, &mcpUsage)
181173
if k8serrors.IsNotFound(err) {
182174
return nil
183175
}
176+
if err != nil {
177+
return fmt.Errorf("error getting MCPUsage resource: %w", err)
178+
}
179+
180+
mcpUsage.Spec.MCPDeletedAt = metav1.NewTime(time.Now().UTC())
181+
err = u.client.Update(ctx, &mcpUsage)
184182
if err != nil {
185183
return fmt.Errorf("error when setting deletion timestamp on MCPUsage element: %w", err)
186184
}

0 commit comments

Comments
 (0)