CSHARP-3918: Support authorizedCollections option for listCollections helpers.

Author: DmitryLukyanov

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

@@ -1,4 +1,4 @@
-/* Copyright 2013-present MongoDB Inc.
+/* Copyright 2013-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.
@@ -13,9 +13,11 @@
 * limitations under the License.
 */
 
+using System.Linq;
 using System.Threading;
 using System.Threading.Tasks;
 using MongoDB.Bson;
+using MongoDB.Bson.Serialization.Serializers;
 using MongoDB.Driver.Core.Bindings;
 using MongoDB.Driver.Core.Events;
 using MongoDB.Driver.Core.Misc;
@@ -26,9 +28,10 @@ namespace MongoDB.Driver.Core.Operations
     /// <summary>
     /// Represents a list collections operation.
     /// </summary>
-    public class ListCollectionsOperation : IReadOperation<IAsyncCursor<BsonDocument>>
+    public class ListCollectionsOperation : IReadOperation<IAsyncCursor<BsonDocument>>, IExecutableInRetryableReadContext<IAsyncCursor<BsonDocument>>
     {
         // fields
+        private bool? _authorizedCollections;
         private int? _batchSize;
         private BsonDocument _filter;
         private readonly DatabaseNamespace _databaseNamespace;
@@ -51,6 +54,18 @@ public ListCollectionsOperation(
         }
 
         // properties
+        /// <summary>
+        /// Gets or sets the AuthorizedCollections flag.
+        /// </summary>
+        /// <value>
+        /// Whether authorizedCollections flag is set.
+        /// </value>
+        public bool? AuthorizedCollections
+        {
+            get => _authorizedCollections;
+            set => _authorizedCollections = value;
+        }
+
         /// <summary>
         /// Gets or sets the batch size.
         /// </summary>
@@ -127,11 +142,22 @@ public IAsyncCursor<BsonDocument> Execute(IReadBinding binding, CancellationToke
         {
             Ensure.IsNotNull(binding, nameof(binding));
 
-            using (EventContext.BeginOperation())
             using (var context = RetryableReadContext.Create(binding, _retryRequested, cancellationToken))
             {
-                var operation = CreateOperation(context.Channel);
-                return operation.Execute(context, cancellationToken);
+                return Execute(context, cancellationToken);
+            }
+        }
+
+        /// <inheritdoc/>
+        public IAsyncCursor<BsonDocument> Execute(RetryableReadContext context, CancellationToken cancellationToken)
+        {
+            Ensure.IsNotNull(context, nameof(context));
+
+            using (EventContext.BeginOperation())
+            {
+                var operation = CreateOperation();
+                var result = operation.Execute(context, cancellationToken);
+                return CreateCursor(context.ChannelSource, context.Channel, result);
             }
         }
 
@@ -140,24 +166,58 @@ public async Task<IAsyncCursor<BsonDocument>> ExecuteAsync(IReadBinding binding,
         {
             Ensure.IsNotNull(binding, nameof(binding));
 
-            using (EventContext.BeginOperation())
             using (var context = await RetryableReadContext.CreateAsync(binding, _retryRequested, cancellationToken).ConfigureAwait(false))
             {
-                var operation = CreateOperation(context.Channel);
-                return await operation.ExecuteAsync(context, cancellationToken).ConfigureAwait(false);
+                return await ExecuteAsync(context, cancellationToken).ConfigureAwait(false);
+            }
+        }
+
+        /// <inheritdoc/>
+        public async Task<IAsyncCursor<BsonDocument>> ExecuteAsync(RetryableReadContext context, CancellationToken cancellationToken)
+        {
+            Ensure.IsNotNull(context, nameof(context));
+
+            using (EventContext.BeginOperation())
+            {
+                var operation = CreateOperation();
+                var result = await operation.ExecuteAsync(context, cancellationToken).ConfigureAwait(false);
+                return CreateCursor(context.ChannelSource, context.Channel, result);
             }
         }
 
         // private methods
-        private IExecutableInRetryableReadContext<IAsyncCursor<BsonDocument>> CreateOperation(IChannel channel)
+        private ReadCommandOperation<BsonDocument> CreateOperation()
         {
-            return new ListCollectionsUsingCommandOperation(_databaseNamespace, _messageEncoderSettings)
+            var command = new BsonDocument
+            {
+                { "listCollections", 1 },
+                { "filter", _filter, _filter != null },
+                { "nameOnly", () => _nameOnly.Value, _nameOnly.HasValue },
+                { "cursor", () => new BsonDocument("batchSize", _batchSize.Value), _batchSize.HasValue },
+                { "authorizedCollections", () => _authorizedCollections.Value, _authorizedCollections.HasValue }
+            };
+            return new ReadCommandOperation<BsonDocument>(_databaseNamespace, command, BsonDocumentSerializer.Instance, _messageEncoderSettings)
             {
-                BatchSize = _batchSize,
-                Filter = _filter,
-                NameOnly = _nameOnly,
                 RetryRequested = _retryRequested // might be overridden by retryable read context
             };
         }
+
+        private IAsyncCursor<BsonDocument> CreateCursor(IChannelSourceHandle channelSource, IChannelHandle channel, BsonDocument result)
+        {
+            var cursorDocument = result["cursor"].AsBsonDocument;
+            var cursorId = cursorDocument["id"].ToInt64();
+            var getMoreChannelSource = ChannelPinningHelper.CreateGetMoreChannelSource(channelSource, channel, cursorId);
+            var cursor = new AsyncCursor<BsonDocument>(
+                getMoreChannelSource,
+                CollectionNamespace.FromFullName(cursorDocument["ns"].AsString),
+                cursorDocument["firstBatch"].AsBsonArray.OfType<BsonDocument>().ToList(),
+                cursorId,
+                batchSize: _batchSize ?? 0,
+                0,
+                BsonDocumentSerializer.Instance,
+                _messageEncoderSettings);
+
+            return cursor;
+        }
     }
 }

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

@@ -23,9 +23,19 @@ namespace MongoDB.Driver
     public sealed class ListCollectionNamesOptions
     {
         // fields
+        private bool? authorizedCollections;
         private FilterDefinition<BsonDocument> _filter;
 
         // properties
+        /// <summary>
+        /// Gets or sets the AuthorizedCollections flag.
+        /// </summary>
+        public bool? AuthorizedCollections
+        {
+            get { return authorizedCollections; }
+            set { authorizedCollections = value; }
+        }
+
         /// <summary>
         /// Gets or sets the filter.
         /// </summary>

src/MongoDB.Driver/ListCollectionNamesOptions.cs

@@ -693,6 +693,7 @@ private ListCollectionsOperation CreateListCollectionNamesOperation(ListCollecti
             var messageEncoderSettings = GetMessageEncoderSettings();
             return new ListCollectionsOperation(_databaseNamespace, messageEncoderSettings)
             {
+                AuthorizedCollections = options?.AuthorizedCollections,
                 Filter = options?.Filter?.Render(_settings.SerializerRegistry.GetSerializer<BsonDocument>(), _settings.SerializerRegistry, _linqProvider),
                 NameOnly = true,
                 RetryRequested = _client.Settings.RetryReads