larryh/corso
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Updates to /connector/sharepoint/data_collections.go

Browse Source 
CollectPages() method updated to create BetaService and to store service within collection.
This commit is contained in:
Danny Adams 2023-01-27 13:20:42 -05:00
parent 1aebe933ea
commit 560531a56c
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/internal/connector/sharepoint/data_collections.go
View File

@ -7,6 +7,7 @@ import (
"github.com/pkg/errors" "github.com/pkg/errors"
"github.com/alcionai/corso/src/internal/connector/graph" "github.com/alcionai/corso/src/internal/connector/graph"
"github.com/alcionai/corso/src/internal/connector/graph/betasdk"
"github.com/alcionai/corso/src/internal/connector/onedrive" "github.com/alcionai/corso/src/internal/connector/onedrive"
"github.com/alcionai/corso/src/internal/connector/support" "github.com/alcionai/corso/src/internal/connector/support"
"github.com/alcionai/corso/src/internal/data" "github.com/alcionai/corso/src/internal/data"
@ -176,9 +177,14 @@ func collectPages(
spcs := make([]data.Collection, 0) spcs := make([]data.Collection, 0)
// make the betaClient // make the betaClient
betaService := betaService adpt, err := graph.CreateAdapter(creds.AzureTenantID, creds.AzureClientID, creds.AzureClientSecret)
if err != nil {
return nil, errors.Wrap(err, "adapter for betaservice not created")
}
tuples, err := fetchPages(ctx, serv, siteID) betaService := betasdk.NewService(adpt)
tuples, err := fetchPages(ctx, betaService, siteID)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -195,6 +201,7 @@ func collectPages(
} }
collection := NewCollection(dir, serv, updater.UpdateStatus) collection := NewCollection(dir, serv, updater.UpdateStatus)
collection.betaService = betaService
collection.AddJob(tuple.id) collection.AddJob(tuple.id)
spcs = append(spcs, collection) spcs = append(spcs, collection)