From 45e8d7acecc85830b3295789246b5fedb4f78ef6 Mon Sep 17 00:00:00 2001 From: Tanner Stirrat Date: Tue, 17 Sep 2024 09:26:44 -0600 Subject: [PATCH] Add usage metrics for export --- internal/services/v1/permissions.go | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/internal/services/v1/permissions.go b/internal/services/v1/permissions.go index 72866dca7f..437d7fe32e 100644 --- a/internal/services/v1/permissions.go +++ b/internal/services/v1/permissions.go @@ -1075,6 +1075,8 @@ func ExportBulk(ctx context.Context, ds datastore.Datastore, batchSize uint64, r } emptyRels := make([]*v1.Relationship, limit) + // The number of batches/dispatches for the purpose of usage metrics + var batches uint32 for _, ns := range namespaces { rels := emptyRels @@ -1154,7 +1156,16 @@ func ExportBulk(ctx context.Context, ds datastore.Datastore, batchSize uint64, r }); err != nil { return shared.RewriteErrorWithoutConfig(ctx, err) } + // Increment batches for usagemetrics + batches++ } } + + // Record usage metrics + respMetadata := &dispatch.ResponseMeta{ + DispatchCount: batches, + } + usagemetrics.SetInContext(ctx, respMetadata) + return nil }