docs/csharp/asynchronous-programming/snippets/generate-consume-asynchronous-streams/start/Program.cs - main - aprogrammer/dotnet-docs

docs/csharp/asynchronous-programming/snippets/generate-consume-asynchronous-streams/start/Program.cs
149 строк5 KB
Bartosz Klonowski
Mention the GraphQLRequest class and ToJsonText method in async streams tutorial (#37669)
23 окт 2023, 18:48
23 окт 2023, 18:48f8c0210
using Newtonsoft.Json;
using Newtonsoft.Json.Linq;
using Octokit;

namespace GitHubActivityReport;

// <SnippetStarterGraphQLRequest>
public class GraphQLRequest
{
    [JsonProperty("query")]
    public string? Query { get; set; }

    [JsonProperty("variables")]
    public IDictionary<string, object> Variables { get; } = new Dictionary<string, object>();

    public string ToJsonText() =>
        JsonConvert.SerializeObject(this);
}
// </SnippetStarterGraphQLRequest>

class Program
{
    private const string PagedIssueQuery =
@"query ($repo_name: String!,  $start_cursor:String) {
  repository(owner: ""dotnet"", name: $repo_name) {
    issues(last: 25, before: $start_cursor)
     {
        totalCount
        pageInfo {
          hasPreviousPage
          startCursor
        }
        nodes {
          title
          number
          createdAt
        }
      }
    }
  }
";
    private class progressStatus : IProgress<int>
    {
        Action<int> action;
        public progressStatus(Action<int> progressAction) =>
            action = progressAction;

        public void Report(int value) => action(value);
    }

    // <SnippetStarterAppMain>
    static async Task Main(string[] args)
    {
        //Follow these steps to create a GitHub Access Token
        // https://help.github.com/articles/creating-a-personal-access-token-for-the-command-line/#creating-a-token
        //Select the following permissions for your GitHub Access Token:
        // - repo:status
        // - public_repo
        // Replace the 3rd parameter to the following code with your GitHub access token.
        var key = GetEnvVariable("GitHubKey",
        "You must store your GitHub key in the 'GitHubKey' environment variable",
        "");

        var client = new GitHubClient(new Octokit.ProductHeaderValue("IssueQueryDemo"))
        {
            Credentials = new Octokit.Credentials(key)
        };

        // <SnippetEnumerateOldStyle>
        var progressReporter = new progressStatus((num) =>
        {
            Console.WriteLine($"Received {num} issues in total");
        });
        CancellationTokenSource cancellationSource = new CancellationTokenSource();

        try
        {
            var results = await RunPagedQueryAsync(client, PagedIssueQuery, "docs",
                cancellationSource.Token, progressReporter);
            foreach(var issue in results)
                Console.WriteLine(issue);
        }
        catch (OperationCanceledException)
        {
            Console.WriteLine("Work has been cancelled");
        }
        // </SnippetEnumerateOldStyle>
    }
    // </SnippetStarterAppMain>

    // <SnippetRunPagedQuery>
    private static async Task<JArray> RunPagedQueryAsync(GitHubClient client, string queryText, string repoName, CancellationToken cancel, IProgress<int> progress)
    {
        var issueAndPRQuery = new GraphQLRequest
        {
            Query = queryText
        };
        issueAndPRQuery.Variables["repo_name"] = repoName;

        JArray finalResults = new JArray();
        bool hasMorePages = true;
        int pagesReturned = 0;
        int issuesReturned = 0;

        // Stop with 10 pages, because these are large repos:
        while (hasMorePages && (pagesReturned++ < 10))
        {
            var postBody = issueAndPRQuery.ToJsonText();
            var response = await client.Connection.Post<string>(new Uri("https://api.github.com/graphql"),
                postBody, "application/json", "application/json");

            JObject results = JObject.Parse(response.HttpResponse.Body.ToString()!);

            int totalCount = (int)issues(results)["totalCount"]!;
            hasMorePages = (bool)pageInfo(results)["hasPreviousPage"]!;
            issueAndPRQuery.Variables["start_cursor"] = pageInfo(results)["startCursor"]!.ToString();
            issuesReturned += issues(results)["nodes"]!.Count();
            // <SnippetProcessPage>
            finalResults.Merge(issues(results)["nodes"]!);
            progress?.Report(issuesReturned);
            cancel.ThrowIfCancellationRequested();
            // </SnippetProcessPage>
        }
        return finalResults;

        JObject issues(JObject result) => (JObject)result["data"]!["repository"]!["issues"]!;
        JObject pageInfo(JObject result) => (JObject)issues(result)["pageInfo"]!;
    }
    // </SnippetRunPagedQuery>

    private static string GetEnvVariable(string item, string error, string defaultValue)
    {
        var value = Environment.GetEnvironmentVariable(item);
        if (string.IsNullOrWhiteSpace(value))
        {
            if (!string.IsNullOrWhiteSpace(defaultValue))
            {
                return defaultValue;
            }

            if (!string.IsNullOrWhiteSpace(error))
            {
                Console.WriteLine(error);
                Environment.Exit(0);
            }
        }
        return value ?? string.Empty;
    }
}