Add support for ObjectId type for the Id of the document.

2018-03-05 23:07:56 +00:00
parent 2f6515a809
commit 57d332ef2a
9 changed files with 91 additions and 72 deletions
@@ -0,0 +1,31 @@
+using MongoDB.Bson;
+using System;
+
+namespace MongoDbGenericRepository.Utils
+{
+    public static class IdGenerator
+    {
+        private static readonly Random Random = new Random();
+
+        public static TKey GetId<TKey>()
+        {
+            var idTypeName = typeof(TKey).Name;
+            switch (idTypeName)
+            {
+                case "Guid":
+                    return (TKey)(object)Guid.NewGuid();
+                case "Int16":
+                    return (TKey)(object)Random.Next(1, short.MaxValue);
+                case "Int32":
+                    return (TKey)(object)Random.Next(1, int.MaxValue);
+                case "Int64":
+                    return (TKey)(object)(Random.NextLong(1, long.MaxValue));
+                case "String":
+                    return (TKey)(object)Guid.NewGuid().ToString();
+                case "ObjectId":
+                    return (TKey)(object)ObjectId.GenerateNewId();
+            }
+            throw new ArgumentException($"{idTypeName} is not a supported Id type, the Id of the document cannot be set.");
+        }
+    }
+}