azure-cosmos-dotnet-v3/Microsoft.Azure.Cosmos/src/HttpClient/HttpTimeoutPolicy.cs at 0f367b36489233fc40dd3bc7c6ded3e974ed9ff1 · Azure/azure-cosmos-dotnet-v3 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
//------------------------------------------------------------
// Copyright (c) Microsoft Corporation.  All rights reserved.
//------------------------------------------------------------
namespace Microsoft.Azure.Cosmos
{
    using System;
    using System.Collections.Generic;
    using System.Net.Http;
    using Microsoft.Azure.Documents;

    internal abstract class HttpTimeoutPolicy
    {
        public abstract string TimeoutPolicyName { get; }
        public abstract int TotalRetryCount { get; }
        public abstract IEnumerator<(TimeSpan requestTimeout, TimeSpan delayForNextRequest)> GetTimeoutEnumerator();
        public abstract bool IsSafeToRetry(HttpMethod httpMethod);

        public abstract bool ShouldRetryBasedOnResponse(HttpMethod requestHttpMethod, HttpResponseMessage responseMessage);

        public virtual bool ShouldThrow503OnTimeout => false;

        public static HttpTimeoutPolicy GetTimeoutPolicy(
           DocumentServiceRequest documentServiceRequest,
           bool isPartitionLevelFailoverEnabled = false,
           bool isThinClientEnabled = false)
        {
            //Query Plan Requests
            if (documentServiceRequest.ResourceType == ResourceType.Document
                && documentServiceRequest.OperationType == OperationType.QueryPlan)
            {
                return HttpTimeoutPolicyControlPlaneRetriableHotPath.InstanceShouldThrow503OnTimeout;
            }

            //Get Partition Key Range Requests
            if (documentServiceRequest.ResourceType == ResourceType.PartitionKeyRange
                && documentServiceRequest.OperationType == OperationType.ReadFeed)
            {
                return HttpTimeoutPolicyControlPlaneRetriableHotPath.InstanceShouldThrow503OnTimeout;
            }

            //Get Addresses Requests
            if (documentServiceRequest.ResourceType == ResourceType.Address)
            {
                return HttpTimeoutPolicyControlPlaneRetriableHotPath.InstanceShouldThrow503OnTimeout;
            }

            //Data Plane Operations
            if (!HttpTimeoutPolicy.IsMetaData(documentServiceRequest))
            {
                if (isThinClientEnabled)
                {
                    return documentServiceRequest.IsReadOnlyRequest
                        ? HttpTimeoutPolicyForThinClient.InstanceShouldRetryAndThrow503OnTimeout
                        : HttpTimeoutPolicyForThinClient.InstanceShouldNotRetryAndThrow503OnTimeout;
                }
                // Data Plane Reads.
                else if (documentServiceRequest.IsReadOnlyRequest)
                {
                    return isPartitionLevelFailoverEnabled
                        ? HttpTimeoutPolicyForPartitionFailover.InstanceShouldThrow503OnTimeout
                        : HttpTimeoutPolicyDefault.InstanceShouldThrow503OnTimeout;
                }
            }

            //Meta Data Read
            if (HttpTimeoutPolicy.IsMetaData(documentServiceRequest) && documentServiceRequest.IsReadOnlyRequest)
            {
                return HttpTimeoutPolicyControlPlaneRetriableHotPath.InstanceShouldThrow503OnTimeout;
            }

            //Default behavior
            return HttpTimeoutPolicyDefault.Instance;
        }

        private static bool IsMetaData(DocumentServiceRequest request)
        {
            return (request.OperationType != Documents.OperationType.ExecuteJavaScript && request.ResourceType == ResourceType.StoredProcedure) ||
                request.ResourceType != ResourceType.Document;

        }
    }
}