Also returning expanded principals for improved readability

gravitational · Jan 7, 2025 · a20ac6f · a20ac6f
1 parent 437e832
commit a20ac6f
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/lib/srv/discovery/fetchers/azure-sync/memberships.go b/lib/srv/discovery/fetchers/azure-sync/memberships.go
@@ -10,7 +10,10 @@ import (
 
 const parallelism = 10
 
-func expandMemberships(ctx context.Context, cli *msgraph.Client, principals []*accessgraphv1alpha.AzurePrincipal) error {
+func expandMemberships(
+	ctx context.Context,
+	cli *msgraph.Client, principals []*accessgraphv1alpha.AzurePrincipal,
+) ([]*accessgraphv1alpha.AzurePrincipal, error) {
 	var eg errgroup.Group
 	eg.SetLimit(parallelism)
 	for _, principal := range principals {
@@ -25,5 +28,5 @@ func expandMemberships(ctx context.Context, cli *msgraph.Client, principals []*a
 			return nil
 		})
 	}
-	return eg.Wait()
+	return principals, eg.Wait()
 }