From 3eaebaed8fef8e0c062b6b0640e814d1d92bfed4 Mon Sep 17 00:00:00 2001 From: Danny Date: Tue, 2 Aug 2022 12:21:35 -0400 Subject: [PATCH] Incremental change prior to selector (#460) Use of abstract variable for creating page iterator. --- src/internal/connector/graph_connector.go | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/internal/connector/graph_connector.go b/src/internal/connector/graph_connector.go index 03620bc0e..b165018b0 100644 --- a/src/internal/connector/graph_connector.go +++ b/src/internal/connector/graph_connector.go @@ -8,6 +8,7 @@ import ( "strings" "sync/atomic" + absser "github.com/microsoft/kiota-abstractions-go/serialization" msgraphsdk "github.com/microsoftgraph/msgraph-sdk-go" msgraphgocore "github.com/microsoftgraph/msgraph-sdk-go-core" "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 // to the GraphConnector struct. func (gc *GraphConnector) serializeMessages(ctx context.Context, user string) (map[string]*exchange.Collection, error) { + var transformer absser.ParsableFactory options := optionsForMessageSnapshot() response, err := gc.graphService.client.UsersById(user).Messages().GetWithRequestConfigurationAndResponseHandler(options, nil) if err != nil { 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 { return nil, err }