k-space/dex
Archived
10
0
Fork 0
Code Issues Pull Requests Packages Projects Activity

google: Retrieve all the groups for a user

Browse Source 
The list of groups is paginated (default page is 200), so when a user
has more than 200 groups, only the first 200 are retrieve.

This change is retrieving all the groups for a user by querying all the
pages.
This commit is contained in:
Fabrice Rabaute 2020-01-14 12:18:39 -08:00
parent 1cdb2b1d74
commit b85d7849ad
No known key found for this signature in database
GPG Key ID: 230B0BA8E423D606
1 changed files with 17 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
connector/google/google.go
View File

@ -240,15 +240,24 @@ func (c *googleConnector) createIdentity(ctx context.Context, identity connector
// getGroups creates a connection to the admin directory service and lists // getGroups creates a connection to the admin directory service and lists
// all groups the user is a member of // all groups the user is a member of
func (c *googleConnector) getGroups(email string) ([]string, error) { func (c *googleConnector) getGroups(email string) ([]string, error) {
groupsList, err := c.adminSrv.Groups.List().UserKey(email).Do()
if err != nil {
return nil, fmt.Errorf("could not list groups: %v", err)
}
var userGroups []string var userGroups []string
for _, group := range groupsList.Groups { var err error
// TODO (joelspeed): Make desried group key configurable groupsList := &admin.Groups{}
userGroups = append(userGroups, group.Email) for {
groupsList, err = c.adminSrv.Groups.List().
UserKey(email).PageToken(groupsList.NextPageToken).Do()
if err != nil {
return nil, fmt.Errorf("could not list groups: %v", err)
}
for _, group := range groupsList.Groups {
// TODO (joelspeed): Make desried group key configurable
userGroups = append(userGroups, group.Email)
}
if groupsList.NextPageToken == "" {
break
}
} }
return userGroups, nil return userGroups, nil