extern/netbird
1
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2025-08-16 10:08:12 +02:00
Code Issues Packages Projects Releases Wiki Activity

debug google workspace request

Browse Source
This commit is contained in:
bcmmbaga
2024-02-16 13:10:37 +03:00
parent cf87f1e702
commit e3d038da8a
1 changed files with 21 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
management/server/idp/google_workspace.go
View File

@ -5,6 +5,8 @@ import (
"encoding/base64" "encoding/base64"
"fmt" "fmt"
"net/http" "net/http"
"os"
"strconv"
"time" "time"
log "github.com/sirupsen/logrus" log "github.com/sirupsen/logrus"
@ -150,19 +152,37 @@ func (gm *GoogleWorkspaceManager) GetAllAccounts() (map[string][]*UserData, erro
// getAllUsers returns all users in a Google Workspace account filtered by customer ID. // getAllUsers returns all users in a Google Workspace account filtered by customer ID.
func (gm *GoogleWorkspaceManager) getAllUsers() ([]*UserData, error) { func (gm *GoogleWorkspaceManager) getAllUsers() ([]*UserData, error) {
var usersLimit int64 = 500
if maxUsersLimitEnv := os.Getenv("GOOGLE_WORKSPACE_USERS_LIMIT"); maxUsersLimitEnv != "" {
maxUsersLimit, err := strconv.Atoi(maxUsersLimitEnv)
if err == nil {
log.Debugf("GOOGLE_WORKSPACE_USERS_LIMIT env is set using %d as users limit", maxUsersLimit)
usersLimit = int64(maxUsersLimit)
}
} else {
log.Debugf("GOOGLE_WORKSPACE_USERS_LIMIT env is not set using default users limit 500")
}
users := make([]*UserData, 0) users := make([]*UserData, 0)
pageToken := "" pageToken := ""
for { for {
call := gm.usersService.List().Customer(gm.CustomerID).MaxResults(500) call := gm.usersService.List().Customer(gm.CustomerID).MaxResults(usersLimit)
if pageToken != "" { if pageToken != "" {
call.PageToken(pageToken) call.PageToken(pageToken)
} }
resp, err := call.Do() resp, err := call.Do()
if err != nil { if err != nil {
log.Debugf("failed to retrieve users from workspace error: %s, http status: %d, headers: %v",
err.Error(),
resp.HTTPStatusCode,
resp.Header,
)
return nil, err return nil, err
} }
log.Debugf("fetched %d users from workspace", len(resp.Users))
for _, user := range resp.Users { for _, user := range resp.Users {
users = append(users, parseGoogleWorkspaceUser(user)) users = append(users, parseGoogleWorkspaceUser(user))
} }