public GatewayStoreClient(
     CosmosHttpClient httpClient,
     ICommunicationEventSource eventSource,
     JsonSerializerSettings serializerSettings = null)
 {
     this.httpClient         = httpClient;
     this.SerializerSettings = serializerSettings;
     this.eventSource        = eventSource;
 }
 // Backlog: Auth abstractions are spilling through. 4 arguments for this CTOR are result of it.
 public GatewayAccountReader(Uri serviceEndpoint,
                             AuthorizationTokenProvider cosmosAuthorization,
                             ConnectionPolicy connectionPolicy,
                             CosmosHttpClient httpClient)
 {
     this.httpClient          = httpClient;
     this.serviceEndpoint     = serviceEndpoint;
     this.cosmosAuthorization = cosmosAuthorization ?? throw new ArgumentNullException(nameof(AuthorizationTokenProvider));
     this.connectionPolicy    = connectionPolicy;
 }
Esempio n. 3
0
 public GatewayAccountReader(Uri serviceEndpoint,
                             IComputeHash stringHMACSHA256Helper,
                             bool hasResourceToken,
                             string resourceToken,
                             ConnectionPolicy connectionPolicy,
                             CosmosHttpClient httpClient)
 {
     this.httpClient              = httpClient;
     this.serviceEndpoint         = serviceEndpoint;
     this.authKeyHashFunction     = stringHMACSHA256Helper;
     this.hasAuthKeyResourceToken = hasResourceToken;
     this.authKeyResourceToken    = resourceToken;
     this.connectionPolicy        = connectionPolicy;
 }
        public GatewayStoreModel(
            GlobalEndpointManager endpointManager,
            ISessionContainer sessionContainer,
            ConsistencyLevel defaultConsistencyLevel,
            DocumentClientEventSource eventSource,
            JsonSerializerSettings serializerSettings,
            CosmosHttpClient httpClient)
        {
            this.endpointManager         = endpointManager;
            this.sessionContainer        = sessionContainer;
            this.defaultConsistencyLevel = defaultConsistencyLevel;
            this.eventSource             = eventSource;

            this.gatewayStoreClient = new GatewayStoreClient(
                httpClient,
                this.eventSource,
                serializerSettings);
        }
Esempio n. 5
0
        public async Task GatewayStatsDurationTest()
        {
            bool failedOnce = false;
            Func <HttpRequestMessage, Task <HttpResponseMessage> > sendFunc = async request =>
            {
                await Task.Delay(1000);

                if (!failedOnce)
                {
                    failedOnce = true;
                    throw new OperationCanceledException();
                }

                return(new HttpResponseMessage(HttpStatusCode.OK)
                {
                    Content = new StringContent("Response")
                });
            };

            HttpMessageHandler mockMessageHandler = new MockMessageHandler(sendFunc);
            CosmosHttpClient   cosmosHttpClient   = MockCosmosUtil.CreateCosmosHttpClient(() => new HttpClient(mockMessageHandler),
                                                                                          DocumentClientEventSource.Instance);

            Tracing.TraceData.ClientSideRequestStatisticsTraceDatum clientSideRequestStatistics = new Tracing.TraceData.ClientSideRequestStatisticsTraceDatum(DateTime.UtcNow);

            await cosmosHttpClient.SendHttpAsync(() => new ValueTask <HttpRequestMessage>(new HttpRequestMessage(HttpMethod.Get, "http://someuri.com")),
                                                 ResourceType.Document,
                                                 HttpTimeoutPolicyDefault.Instance,
                                                 clientSideRequestStatistics,
                                                 CancellationToken.None);

            Assert.AreEqual(clientSideRequestStatistics.HttpResponseStatisticsList.Count, 2);
            // The duration is calculated using date times which can cause the duration to be slightly off. This allows for up to 15 Ms of variance.
            // https://stackoverflow.com/questions/2143140/c-sharp-datetime-now-precision#:~:text=The%20precision%20is%20related%20to,35%2D40%20ms%20accuracy
            Assert.IsTrue(clientSideRequestStatistics.HttpResponseStatisticsList[0].Duration.TotalMilliseconds >= 985, $"First request did was not delayed by at least 1 second. {JsonConvert.SerializeObject(clientSideRequestStatistics.HttpResponseStatisticsList[0])}");
            Assert.IsTrue(clientSideRequestStatistics.HttpResponseStatisticsList[1].Duration.TotalMilliseconds >= 985, $"Second request did was not delayed by at least 1 second. {JsonConvert.SerializeObject(clientSideRequestStatistics.HttpResponseStatisticsList[1])}");
            Assert.IsTrue(clientSideRequestStatistics.HttpResponseStatisticsList[0].RequestStartTime <
                          clientSideRequestStatistics.HttpResponseStatisticsList[1].RequestStartTime);
        }
        public async Task DocumentClient_BuildHttpClientFactory_WithHandler()
        {
            HttpMessageHandler messageHandler   = new CustomMessageHandler();
            ConnectionPolicy   connectionPolicy = new ConnectionPolicy()
            {
                HttpClientFactory = () => new HttpClient(messageHandler)
            };

            CosmosHttpClient httpClient = CosmosHttpClientCore.CreateWithConnectionPolicy(
                apiType: ApiType.None,
                eventSource: DocumentClientEventSource.Instance,
                connectionPolicy: connectionPolicy,
                httpMessageHandler: null,
                sendingRequestEventArgs: null,
                receivedResponseEventArgs: null);

            Assert.IsNotNull(httpClient);
            HttpResponseMessage response = await httpClient.GetAsync(
                uri : new Uri("https://localhost"),
                additionalHeaders : new DictionaryNameValueCollection(),
                resourceType : ResourceType.Document,
                diagnosticsContext : null,
                cancellationToken : default);