CSHARP-3549: CSOT: Add timeoutMS to settings

src/MongoDB.Driver/AggregateOptions.cs

@@ -1,4 +1,4 @@
-/* Copyright 2015-present MongoDB Inc.
+/* Copyright 2010-present MongoDB Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
@@ -34,6 +34,7 @@ public class AggregateOptions
         private BsonDocument _let;
         private TimeSpan? _maxAwaitTime;
         private TimeSpan? _maxTime;
+        private TimeSpan? _timeout;
         private ExpressionTranslationOptions _translationOptions;
         private bool? _useCursor;
 
@@ -127,6 +128,16 @@ public TimeSpan? MaxTime
             set { _maxTime = Ensure.IsNullOrInfiniteOrGreaterThanOrEqualToZero(value, nameof(value)); }
         }
 
+        /// <summary>
+        /// Gets or sets the operation timeout.
+        /// </summary>
+        // TODO: CSOT: Make it public when CSOT will be ready for GA
+        internal TimeSpan? Timeout
+        {
+            get => _timeout;
+            set => _timeout = Ensure.IsNullOrValidTimeout(value, nameof(Timeout));
+        }
+
         /// <summary>
         /// Gets or sets the translation options.
         /// </summary>

src/MongoDB.Driver/BulkWriteOptions.cs

@@ -13,7 +13,9 @@
 * limitations under the License.
 */
 
+using System;
 using MongoDB.Bson;
+using MongoDB.Driver.Core.Misc;
 
 namespace MongoDB.Driver
 {
@@ -27,6 +29,7 @@ public sealed class BulkWriteOptions
         private BsonValue _comment;
         private bool _isOrdered;
         private BsonDocument _let;
+        private TimeSpan? _timeout;
 
         // constructors
         /// <summary>
@@ -73,5 +76,15 @@ public BsonDocument Let
             get { return _let; }
             set { _let = value; }
         }
+
+        /// <summary>
+        /// Gets or sets the operation timeout.
+        /// </summary>
+        // TODO: CSOT: Make it public when CSOT will be ready for GA
+        internal TimeSpan? Timeout
+        {
+            get => _timeout;
+            set => _timeout = Ensure.IsNullOrValidTimeout(value, nameof(Timeout));
+        }
     }
 }

src/MongoDB.Driver/ChangeStreamOptions.cs

@@ -1,4 +1,4 @@
-/* Copyright 2017-present MongoDB Inc.
+/* Copyright 2010-present MongoDB Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
@@ -35,6 +35,7 @@ public class ChangeStreamOptions
         private bool? _showExpandedEvents;
         private BsonDocument _startAfter;
         private BsonTimestamp _startAtOperationTime;
+        private TimeSpan? _timeout;
 
         // public properties
         /// <summary>
@@ -166,5 +167,15 @@ public BsonTimestamp StartAtOperationTime
             get { return _startAtOperationTime; }
             set { _startAtOperationTime = value; }
         }
+
+        /// <summary>
+        /// Gets or sets the operation timeout.
+        /// </summary>
+        // TODO: CSOT: Make it public when CSOT will be ready for GA
+        internal TimeSpan? Timeout
+        {
+            get => _timeout;
+            set => _timeout = Ensure.IsNullOrValidTimeout(value, nameof(Timeout));
+        }
     }
 }

src/MongoDB.Driver/ClientBulkWriteOptions.cs

@@ -13,7 +13,9 @@
 * limitations under the License.
 */
 
+using System;
 using MongoDB.Bson;
+using MongoDB.Driver.Core.Misc;
 
 namespace MongoDB.Driver
 {
@@ -22,6 +24,8 @@ namespace MongoDB.Driver
     /// </summary>
     public sealed class ClientBulkWriteOptions
     {
+        private TimeSpan? _timeout;
+
         /// <summary>
         /// Initializes a new instance of the <see cref="BulkWriteOptions"/> class.
         /// </summary>
@@ -75,6 +79,16 @@ public ClientBulkWriteOptions(
         /// </summary>
         public BsonDocument Let { get; set; }
 
+        /// <summary>
+        /// Gets or sets the operation timeout.
+        /// </summary>
+        // TODO: CSOT: Make it public when CSOT will be ready for GA
+        internal TimeSpan? Timeout
+        {
+            get => _timeout;
+            set => _timeout = Ensure.IsNullOrValidTimeout(value, nameof(Timeout));
+        }
+
         /// <summary>
         /// Whether detailed results for each successful operation should be included in the returned results.
         /// </summary>

src/MongoDB.Driver/Core/Bindings/CoreSession.cs

@@ -146,7 +146,7 @@ public bool IsInTransaction
             EnsureAbortTransactionCanBeCalled(nameof(AbortTransaction));
 
             // TODO: CSOT implement proper way to obtain the operationContext
-            var operationContext = new OperationContext(Timeout.InfiniteTimeSpan, cancellationToken);
+            var operationContext = new OperationContext(null, cancellationToken);
             try
             {
                 if (_currentTransaction.IsEmpty)
@@ -197,7 +197,7 @@ public bool IsInTransaction
             EnsureAbortTransactionCanBeCalled(nameof(AbortTransaction));
 
             // TODO: CSOT implement proper way to obtain the operationContext
-            var operationContext = new OperationContext(Timeout.InfiniteTimeSpan, cancellationToken);
+            var operationContext = new OperationContext(null, cancellationToken);
             try
             {
                 if (_currentTransaction.IsEmpty)
@@ -297,7 +297,7 @@ public long AdvanceTransactionNumber()
             EnsureCommitTransactionCanBeCalled(nameof(CommitTransaction));
 
             // TODO: CSOT implement proper way to obtain the operationContext
-            var operationContext = new OperationContext(Timeout.InfiniteTimeSpan, cancellationToken);
+            var operationContext = new OperationContext(null, cancellationToken);
             try
             {
                 _isCommitTransactionInProgress = true;
@@ -334,7 +334,7 @@ public long AdvanceTransactionNumber()
             EnsureCommitTransactionCanBeCalled(nameof(CommitTransaction));
 
             // TODO: CSOT implement proper way to obtain the operationContext
-            var operationContext = new OperationContext(Timeout.InfiniteTimeSpan, cancellationToken);
+            var operationContext = new OperationContext(null, cancellationToken);
             try
             {
                 _isCommitTransactionInProgress = true;

src/MongoDB.Driver/Core/Configuration/ConnectionString.cs

@@ -98,6 +98,9 @@ public sealed class ConnectionString
         private TimeSpan? _socketTimeout;
         private int? _srvMaxHosts;
         private string _srvServiceName;
+#pragma warning disable CS0649 // Field is never assigned to, and will always have its default value
+        private TimeSpan? _timeout;
+#pragma warning restore CS0649 // Field is never assigned to, and will always have its default value
         private bool? _tls;
         private bool? _tlsDisableCertificateRevocationCheck;
         private bool? _tlsInsecure;
@@ -399,7 +402,6 @@ public bool? RetryReads
             get { return _retryReads; }
         }
 
-
         /// <summary>
         /// Gets a value indicating whether or not to retry writes.
         /// </summary>
@@ -468,6 +470,12 @@ public bool? Ssl
         [Obsolete("Use TlsInsecure instead.")]
         public bool? SslVerifyCertificate => !_tlsInsecure;
 
+        /// <summary>
+        /// Gets the per-operation timeout.
+        /// </summary>
+        // TODO: CSOT: Make it public when CSOT will be ready for GA
+        internal TimeSpan? Timeout => _timeout;
+
         /// <summary>
         /// Gets whether to use TLS.
         /// </summary>
@@ -1089,6 +1097,12 @@ private void ParseOption(string name, string value)
                     var sslVerifyCertificateValue = ParseBoolean(name, value);
                     _tlsInsecure = EnsureTlsInsecureIsValid(!sslVerifyCertificateValue);
                     break;
+#if DEBUG // TODO: CSOT: Make it public when CSOT will be ready for GA
+                case "timeout":
+                case "timeoutms":
+                    _timeout = value == "0" ? System.Threading.Timeout.InfiniteTimeSpan : ParseTimeSpan(name, value);
+                    break;
+#endif
                 case "tlsdisablecertificaterevocationcheck":
                     var tlsDisableCertificateRevocationCheckValue = ParseBoolean(name, value);
                     _tlsDisableCertificateRevocationCheck =

src/MongoDB.Driver/Core/Misc/Ensure.cs

@@ -1,4 +1,4 @@
-/* Copyright 2013-present MongoDB Inc.
+/* Copyright 2010-present MongoDB Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
@@ -461,8 +461,7 @@ public static TimeSpan IsValidTimeout(TimeSpan value, string paramName)
         {
             if (value < TimeSpan.Zero && value != Timeout.InfiniteTimeSpan)
             {
-                var message = string.Format("Invalid timeout: {0}.", value);
-                throw new ArgumentException(message, paramName);
+                throw new ArgumentOutOfRangeException($"Invalid timeout: {value}.", paramName);
             }
             return value;
         }

src/MongoDB.Driver/Core/Misc/Feature.cs

@@ -570,7 +570,7 @@ public void ThrowIfNotSupported(IMongoClient client, CancellationToken cancellat
         {
             var cluster = client.GetClusterInternal();
             // TODO: CSOT implement proper way to obtain the operationContext
-            var operationContext = new OperationContext(Timeout.InfiniteTimeSpan, cancellationToken);
+            var operationContext = new OperationContext(null, cancellationToken);
             using (var binding = new ReadWriteBindingHandle(new WritableServerBinding(cluster, NoCoreSession.NewHandle())))
             using (var channelSource = binding.GetWriteChannelSource(operationContext))
             using (var channel = channelSource.GetChannel(operationContext))
@@ -589,7 +589,7 @@ public async Task ThrowIfNotSupportedAsync(IMongoClient client, CancellationToke
         {
             var cluster = client.GetClusterInternal();
             // TODO: CSOT implement proper way to obtain the operationContext
-            var operationContext = new OperationContext(Timeout.InfiniteTimeSpan, cancellationToken);
+            var operationContext = new OperationContext(null, cancellationToken);
             using (var binding = new ReadWriteBindingHandle(new WritableServerBinding(cluster, NoCoreSession.NewHandle())))
             using (var channelSource = await binding.GetWriteChannelSourceAsync(operationContext).ConfigureAwait(false))
             using (var channel = await channelSource.GetChannelAsync(operationContext).ConfigureAwait(false))

src/MongoDB.Driver/Core/Operations/AggregateOperation.cs

@@ -288,7 +288,7 @@ public IAsyncCursor<TResult> Execute(OperationContext operationContext, Retryabl
 
             using (EventContext.BeginOperation())
             {
-                var operation = CreateOperation(context);
+                var operation = CreateOperation(operationContext, context);
                 var result = operation.Execute(operationContext, context);
 
                 context.ChannelSource.Session.SetSnapshotTimeIfNeeded(result.AtClusterTime);
@@ -317,7 +317,7 @@ public async Task<IAsyncCursor<TResult>> ExecuteAsync(OperationContext operation
 
             using (EventContext.BeginOperation())
             {
-                var operation = CreateOperation(context);
+                var operation = CreateOperation(operationContext, context);
                 var result = await operation.ExecuteAsync(operationContext, context).ConfigureAwait(false);
 
                 context.ChannelSource.Session.SetSnapshotTimeIfNeeded(result.AtClusterTime);
@@ -326,15 +326,15 @@ public async Task<IAsyncCursor<TResult>> ExecuteAsync(OperationContext operation
             }
         }
 
-        internal BsonDocument CreateCommand(ConnectionDescription connectionDescription, ICoreSession session)
+        internal BsonDocument CreateCommand(OperationContext operationContext, ICoreSession session, ConnectionDescription connectionDescription)
         {
             var readConcern = ReadConcernHelper.GetReadConcernForCommand(session, connectionDescription, _readConcern);
             var command = new BsonDocument
             {
                 { "aggregate", _collectionNamespace == null ? (BsonValue)1 : _collectionNamespace.CollectionName },
                 { "pipeline", new BsonArray(_pipeline) },
                 { "allowDiskUse", () => _allowDiskUse.Value, _allowDiskUse.HasValue },
-                { "maxTimeMS", () => MaxTimeHelper.ToMaxTimeMS(_maxTime.Value), _maxTime.HasValue },
+                { "maxTimeMS", () => MaxTimeHelper.ToMaxTimeMS(_maxTime.Value), _maxTime.HasValue && !operationContext.IsRootContextTimeoutConfigured() },
                 { "collation", () => _collation.ToBsonDocument(), _collation != null },
                 { "hint", _hint, _hint != null },
                 { "let", _let, _let != null },
@@ -354,10 +354,10 @@ internal BsonDocument CreateCommand(ConnectionDescription connectionDescription,
 
         private IDisposable BeginOperation() => EventContext.BeginOperation(null, "aggregate");
 
-        private ReadCommandOperation<AggregateResult> CreateOperation(RetryableReadContext context)
+        private ReadCommandOperation<AggregateResult> CreateOperation(OperationContext operationContext, RetryableReadContext context)
         {
             var databaseNamespace = _collectionNamespace == null ? _databaseNamespace : _collectionNamespace.DatabaseNamespace;
-            var command = CreateCommand(context.Channel.ConnectionDescription, context.Binding.Session);
+            var command = CreateCommand(operationContext, context.Binding.Session, context.Channel.ConnectionDescription);
             var serializer = new AggregateResultDeserializer(_resultSerializer);
             return new ReadCommandOperation<AggregateResult>(databaseNamespace, command, serializer, MessageEncoderSettings)
             {

src/MongoDB.Driver/Core/Operations/AggregateToCollectionOperation.cs

@@ -1,4 +1,4 @@
-/* Copyright 2013-present MongoDB Inc.
+/* Copyright 2010-present MongoDB Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
@@ -157,7 +157,7 @@ public BsonDocument Execute(OperationContext operationContext, IWriteBinding bin
             using (var channel = channelSource.GetChannel(operationContext))
             using (var channelBinding = new ChannelReadWriteBinding(channelSource.Server, channel, binding.Session.Fork()))
             {
-                var operation = CreateOperation(channelBinding.Session, channel.ConnectionDescription, mayUseSecondary.EffectiveReadPreference);
+                var operation = CreateOperation(operationContext, channelBinding.Session, channel.ConnectionDescription, mayUseSecondary.EffectiveReadPreference);
                 return operation.Execute(operationContext, channelBinding);
             }
         }
@@ -172,12 +172,12 @@ public async Task<BsonDocument> ExecuteAsync(OperationContext operationContext,
             using (var channel = await channelSource.GetChannelAsync(operationContext).ConfigureAwait(false))
             using (var channelBinding = new ChannelReadWriteBinding(channelSource.Server, channel, binding.Session.Fork()))
             {
-                var operation = CreateOperation(channelBinding.Session, channel.ConnectionDescription, mayUseSecondary.EffectiveReadPreference);
+                var operation = CreateOperation(operationContext, channelBinding.Session, channel.ConnectionDescription, mayUseSecondary.EffectiveReadPreference);
                 return await operation.ExecuteAsync(operationContext, channelBinding).ConfigureAwait(false);
             }
         }
 
-        public BsonDocument CreateCommand(ICoreSessionHandle session, ConnectionDescription connectionDescription)
+        public BsonDocument CreateCommand(OperationContext operationContext, ICoreSessionHandle session, ConnectionDescription connectionDescription)
         {
             var readConcern = _readConcern != null
                 ? ReadConcernHelper.GetReadConcernForCommand(session, connectionDescription, _readConcern)
@@ -189,7 +189,7 @@ public BsonDocument CreateCommand(ICoreSessionHandle session, ConnectionDescript
                 { "pipeline", new BsonArray(_pipeline) },
                 { "allowDiskUse", () => _allowDiskUse.Value, _allowDiskUse.HasValue },
                 { "bypassDocumentValidation", () => _bypassDocumentValidation.Value, _bypassDocumentValidation.HasValue },
-                { "maxTimeMS", () => MaxTimeHelper.ToMaxTimeMS(_maxTime.Value), _maxTime.HasValue },
+                { "maxTimeMS", () => MaxTimeHelper.ToMaxTimeMS(_maxTime.Value), _maxTime.HasValue && !operationContext.IsRootContextTimeoutConfigured() },
                 { "collation", () => _collation.ToBsonDocument(), _collation != null },
                 { "readConcern", readConcern, readConcern != null },
                 { "writeConcern", writeConcern, writeConcern != null },
@@ -202,9 +202,9 @@ public BsonDocument CreateCommand(ICoreSessionHandle session, ConnectionDescript
 
         private IDisposable BeginOperation() => EventContext.BeginOperation("aggregate");
 
-        private WriteCommandOperation<BsonDocument> CreateOperation(ICoreSessionHandle session, ConnectionDescription connectionDescription, ReadPreference effectiveReadPreference)
+        private WriteCommandOperation<BsonDocument> CreateOperation(OperationContext operationContext, ICoreSessionHandle session, ConnectionDescription connectionDescription, ReadPreference effectiveReadPreference)
         {
-            var command = CreateCommand(session, connectionDescription);
+            var command = CreateCommand(operationContext, session, connectionDescription);
             var operation = new WriteCommandOperation<BsonDocument>(_databaseNamespace, command, BsonDocumentSerializer.Instance, MessageEncoderSettings);
             if (effectiveReadPreference != null)
             {