< Summary

Class:Azure.Core.PageableHelpers
Assembly:Azure.Template
File(s):C:\Git\azure-sdk-for-net\sdk\core\Azure.Core\src\Shared\AutoRest\PageableHelpers.cs
Covered lines:0
Uncovered lines:32
Coverable lines:32
Total lines:85
Line coverage:0% (0 of 32)
Covered branches:0
Total branches:26
Branch coverage:0% (0 of 26)

Metrics

MethodCyclomatic complexity Line coverage Branch coverage
CreateEnumerable(...)-0%0%
CreateAsyncEnumerable(...)-0%0%
.ctor(...)-0%100%
AsPages()-0%0%
.ctor(...)-0%100%
AsPages()-0%0%

File(s)

C:\Git\azure-sdk-for-net\sdk\core\Azure.Core\src\Shared\AutoRest\PageableHelpers.cs

#LineLine coverage
 1// Copyright (c) Microsoft Corporation. All rights reserved.
 2// Licensed under the MIT License.
 3
 4#nullable enable
 5
 6using System;
 7using System.Collections.Generic;
 8using System.Threading.Tasks;
 9
 10namespace Azure.Core
 11{
 12    internal static class PageableHelpers
 13    {
 14        public static Pageable<T> CreateEnumerable<T>(Func<int?, Page<T>> firstPageFunc, Func<string?, int?, Page<T>>? n
 15        {
 016            PageFunc<T> first = (continuationToken, pageSizeHint) => firstPageFunc(pageSizeHint);
 017            PageFunc<T>? next = nextPageFunc != null ? new PageFunc<T>(nextPageFunc) : null;
 018            return new FuncPageable<T>(first, next, pageSize);
 19        }
 20
 21        public static AsyncPageable<T> CreateAsyncEnumerable<T>(Func<int?, Task<Page<T>>> firstPageFunc, Func<string?, i
 22        {
 023            AsyncPageFunc<T> first = (continuationToken, pageSizeHint) => firstPageFunc(pageSizeHint);
 024            AsyncPageFunc<T>? next = nextPageFunc != null ? new AsyncPageFunc<T>(nextPageFunc) : null;
 025            return new FuncAsyncPageable<T>(first, next, pageSize);
 26        }
 27
 28        internal delegate Task<Page<T>> AsyncPageFunc<T>(string? continuationToken = default, int? pageSizeHint = defaul
 29        internal delegate Page<T> PageFunc<T>(string? continuationToken = default, int? pageSizeHint = default);
 30
 31        internal class FuncAsyncPageable<T> : AsyncPageable<T> where T : notnull
 32        {
 33            private readonly AsyncPageFunc<T> _firstPageFunc;
 34            private readonly AsyncPageFunc<T>? _nextPageFunc;
 35            private readonly int? _defaultPageSize;
 36
 037            public FuncAsyncPageable(AsyncPageFunc<T> firstPageFunc, AsyncPageFunc<T>? nextPageFunc, int? defaultPageSiz
 38            {
 039                _firstPageFunc = firstPageFunc;
 040                _nextPageFunc = nextPageFunc;
 041                _defaultPageSize = defaultPageSize;
 042            }
 43
 44            public override async IAsyncEnumerable<Page<T>> AsPages(string? continuationToken = default, int? pageSizeHi
 45            {
 046                AsyncPageFunc<T>? pageFunc = _firstPageFunc;
 047                int? pageSize = pageSizeHint ?? _defaultPageSize;
 48                do
 49                {
 050                    Page<T> pageResponse = await pageFunc(continuationToken, pageSize).ConfigureAwait(false);
 051                    yield return pageResponse;
 052                    continuationToken = pageResponse.ContinuationToken;
 053                    pageFunc = _nextPageFunc;
 054                } while (!string.IsNullOrEmpty(continuationToken) && pageFunc != null);
 055            }
 56        }
 57
 58        internal class FuncPageable<T> : Pageable<T> where T : notnull
 59        {
 60            private readonly PageFunc<T> _firstPageFunc;
 61            private readonly PageFunc<T>? _nextPageFunc;
 62            private readonly int? _defaultPageSize;
 63
 064            public FuncPageable(PageFunc<T> firstPageFunc, PageFunc<T>? nextPageFunc, int? defaultPageSize = default)
 65            {
 066                _firstPageFunc = firstPageFunc;
 067                _nextPageFunc = nextPageFunc;
 068                _defaultPageSize = defaultPageSize;
 069            }
 70
 71            public override IEnumerable<Page<T>> AsPages(string? continuationToken = default, int? pageSizeHint = defaul
 72            {
 073                PageFunc<T>? pageFunc = _firstPageFunc;
 074                int? pageSize = pageSizeHint ?? _defaultPageSize;
 75                do
 76                {
 077                    Page<T> pageResponse = pageFunc(continuationToken, pageSize);
 078                    yield return pageResponse;
 079                    continuationToken = pageResponse.ContinuationToken;
 080                    pageFunc = _nextPageFunc;
 081                } while (!string.IsNullOrEmpty(continuationToken) && pageFunc != null);
 082            }
 83        }
 84    }
 85}