using System;
using System.Threading.Tasks;
using Microsoft.Azure.Documents;
using Microsoft.Azure.Documents.Client;
using Microsoft.Azure.Documents.Linq;

namespace CosmosV2Testing
{
    class Program
    {
        static async Task Main(string[] args)
        {
            var endpoint = new Uri("<ENDPOINT>");
            var key = "<KEY>";
            var client = new DocumentClient(endpoint, key, new ConnectionPolicy { ConnectionMode = ConnectionMode.Gateway });

            var collectionUri = UriFactory.CreateDocumentCollectionUri("<DATABASENAME>", "<COLLECTIONNAME>");

            var result = client.CreateDocumentQuery(collectionUri, "SELECT * FROM c ORDER BY c.entryValue, c._ts", new FeedOptions { 
                MaxItemCount = 5,
                PartitionKey = new PartitionKey("d72edfb58d064e09b30445592730f412"),
                ResponseContinuationTokenLimitInKb = 1,
                PopulateQueryMetrics = true
            }).AsDocumentQuery();

            var pageResults = await result.ExecuteNextAsync();
            Console.WriteLine($"Fetched {pageResults.Count} documents costing {pageResults.RequestCharge} RU.");
            Console.WriteLine($"Diagnostics: {pageResults.RequestDiagnosticsString}");

            result = client.CreateDocumentQuery(collectionUri, "SELECT * FROM c ORDER BY c.entryValue, c._ts", new FeedOptions
            {
                MaxItemCount = 5,
                PartitionKey = new PartitionKey("d72edfb58d064e09b30445592730f412"),
                ResponseContinuationTokenLimitInKb = 1,
                RequestContinuation = pageResults.ResponseContinuation,
                PopulateQueryMetrics = true
            }).AsDocumentQuery();

            pageResults = await result.ExecuteNextAsync();
            Console.WriteLine($"Fetched {pageResults.Count} documents costing {pageResults.RequestCharge} RU.");
            Console.WriteLine($"Diagnostics: {pageResults.RequestDiagnosticsString}");

            Console.ReadKey();
        }
    }
}