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

Incremental change prior to selector (#460)

Browse Source 
Use of abstract variable for creating page iterator.
This commit is contained in:
Danny 2022-08-02 12:21:35 -04:00 committed by GitHub
parent bcbf0b1fb8
commit 3eaebaed8f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/internal/connector/graph_connector.go
View File

@ -8,6 +8,7 @@ import (
"strings" "strings"
"sync/atomic" "sync/atomic"
absser "github.com/microsoft/kiota-abstractions-go/serialization"
msgraphsdk "github.com/microsoftgraph/msgraph-sdk-go" msgraphsdk "github.com/microsoftgraph/msgraph-sdk-go"
msgraphgocore "github.com/microsoftgraph/msgraph-sdk-go-core" msgraphgocore "github.com/microsoftgraph/msgraph-sdk-go-core"
"github.com/microsoftgraph/msgraph-sdk-go/models" "github.com/microsoftgraph/msgraph-sdk-go/models"
@ -299,12 +300,14 @@ func (gc *GraphConnector) RestoreMessages(ctx context.Context, dcs []data.Collec
// serializeMessages: Temp Function as place Holder until Collections have been added // serializeMessages: Temp Function as place Holder until Collections have been added
// to the GraphConnector struct. // to the GraphConnector struct.
func (gc *GraphConnector) serializeMessages(ctx context.Context, user string) (map[string]*exchange.Collection, error) { func (gc *GraphConnector) serializeMessages(ctx context.Context, user string) (map[string]*exchange.Collection, error) {
var transformer absser.ParsableFactory
options := optionsForMessageSnapshot() options := optionsForMessageSnapshot()
response, err := gc.graphService.client.UsersById(user).Messages().GetWithRequestConfigurationAndResponseHandler(options, nil) response, err := gc.graphService.client.UsersById(user).Messages().GetWithRequestConfigurationAndResponseHandler(options, nil)
if err != nil { if err != nil {
return nil, err return nil, err
} }
pageIterator, err := msgraphgocore.NewPageIterator(response, &gc.graphService.adapter, models.CreateMessageCollectionResponseFromDiscriminatorValue) transformer = models.CreateMessageCollectionResponseFromDiscriminatorValue
pageIterator, err := msgraphgocore.NewPageIterator(response, &gc.graphService.adapter, transformer)
if err != nil { if err != nil {
return nil, err return nil, err
} }