CSHARP-551: Add SystemFlags and UserFlags to CollectionStatsResult (and obsolete Flags).

Author: rstam

Driver/Core/CommandResults/CollectionStatsResult.cs

@@ -23,6 +23,38 @@
 
 namespace MongoDB.Driver
 {
+    /// <summary>
+    /// Represents collection system flags.
+    /// </summary>
+    [Flags]
+    public enum CollectionSystemFlags
+    {
+        /// <summary>
+        /// No flags.
+        /// </summary>
+        None = 0,
+        /// <summary>
+        /// The collection has an _id index.
+        /// </summary>
+        HaveIdIndex = 1
+    }
+
+    /// <summary>
+    /// Represents collection user flags.
+    /// </summary>
+    [Flags]
+    public enum CollectionUserFlags
+    {
+        /// <summary>
+        /// No flags.
+        /// </summary>
+        None = 0,
+        /// <summary>
+        /// User power of 2 size.
+        /// </summary>
+        UsePowerOf2Size = 1
+    }
+
     /// <summary>
     /// Represents the results of the collection stats command.
     /// </summary>
@@ -68,9 +100,22 @@ public int ExtentCount
         /// <summary>
         /// Gets the flags.
         /// </summary>
+        [Obsolete("Use SystemFlags or UserFlags instead.")]
         public int Flags
         {
-            get { return Response["flags"].AsInt32; }
+            get
+            {
+                // flags was renamed to systemFlags in server version 2.2
+                BsonValue flags;
+                if (Response.TryGetValue("flags", out flags) || Response.TryGetValue("systemFlags", out flags))
+                {
+                    return flags.AsInt32;
+                }
+                else
+                {
+                    return 0;
+                }
+            }
         }
 
         /// <summary>
@@ -153,6 +198,26 @@ public long StorageSize
             get { return Response["storageSize"].ToInt64(); }
         }
 
+        /// <summary>
+        /// Gets the system flags.
+        /// </summary>
+        public CollectionSystemFlags SystemFlags
+        {
+            get
+            {
+                // systemFlags was called flags prior to server version 2.2
+                BsonValue systemFlags;
+                if (Response.TryGetValue("systemFlags", out systemFlags) || Response.TryGetValue("flags", out systemFlags))
+                {
+                    return (CollectionSystemFlags)systemFlags.AsInt32;
+                }
+                else
+                {
+                    return CollectionSystemFlags.None;
+                }
+            }
+        }
+
         /// <summary>
         /// Gets the total index size.
         /// </summary>
@@ -161,6 +226,26 @@ public long TotalIndexSize
             get { return Response["totalIndexSize"].ToInt64(); }
         }
 
+        /// <summary>
+        /// Gets the user flags.
+        /// </summary>
+        public CollectionUserFlags UserFlags
+        {
+            get
+            {
+                // userFlags was first introduced in server version 2.2
+                BsonValue userFlags;
+                if (Response.TryGetValue("userFlags", out userFlags))
+                {
+                    return (CollectionUserFlags)userFlags.AsInt32;
+                }
+                else
+                {
+                    return CollectionUserFlags.None;
+                }
+            }
+        }
+
         // nested classes
         /// <summary>
         /// Represents a collection of index sizes.

DriverUnitTests/Core/CommandResults/CollectionStatsResultTests.cs

@@ -44,7 +44,7 @@ public void Setup()
         public void Test()
         {
             // make sure collection exists and has exactly one document
-            _collection.RemoveAll();
+            _collection.Drop();
             _collection.Insert(new BsonDocument());
 
             var result = _collection.GetStats();
@@ -54,17 +54,26 @@ public void Test()
             Assert.IsTrue(result.AverageObjectSize > 0.0);
             Assert.IsTrue(result.DataSize > 0);
             Assert.IsTrue(result.ExtentCount > 0);
+#pragma warning disable 618
+            Assert.AreEqual(1, result.Flags);
+#pragma warning restore
             Assert.IsTrue(result.IndexCount > 0);
             Assert.IsTrue(result.IndexSizes["_id_"] > 0);
             Assert.IsTrue(result.IndexSizes.ContainsKey("_id_"));
             Assert.IsTrue(result.IndexSizes.Count > 0);
             Assert.IsTrue(result.IndexSizes.Keys.Contains("_id_"));
             Assert.IsTrue(result.IndexSizes.Values.Count() > 0);
             Assert.IsTrue(result.IndexSizes.Values.First() > 0);
+            Assert.IsFalse(result.IsCapped);
             Assert.IsTrue(result.LastExtentSize > 0);
+            Assert.IsFalse(result.Response.Contains("max"));
+            Assert.AreEqual(_collection.FullName, result.Namespace);
+            Assert.AreEqual(1, result.ObjectCount);
             Assert.IsTrue(result.PaddingFactor > 0.0);
             Assert.IsTrue(result.StorageSize > 0);
+            Assert.AreEqual(CollectionSystemFlags.HaveIdIndex, result.SystemFlags);
             Assert.IsTrue(result.TotalIndexSize > 0);
+            Assert.AreEqual(CollectionUserFlags.None, result.UserFlags);
         }
     }
 }