v2 Added Contact,Address

Added Indices...

1 files changed, 23 insertions, 1 deletions

diff --git a/Software/Visual_Studio_v2/Tango.DAL.Mongo/MongoRepository.cs b/Software/Visual_Studio_v2/Tango.DAL.Mongo/MongoRepository.cs
index d268f434d..864bb17cc 100644
--- a/Software/Visual_Studio_v2/Tango.DAL.Mongo/MongoRepository.cs
+++ b/Software/Visual_Studio_v2/Tango.DAL.Mongo/MongoRepository.cs
@@ -6,6 +6,8 @@ using System.Reflection;
 using Tango.DAL.Attributes;
 using System.Linq.Expressions;
 using MongoDB.Driver.Linq;
+using System.Linq;
+using Newtonsoft.Json;
 
 namespace Tango.DAL.Mongo
 {
@@ -25,7 +27,27 @@ namespace Tango.DAL.Mongo
         {
             if (_collection == null)
             {
-                _collection = _database.GetCollection<T>(typeof(T).GetCustomAttribute<MongoCollectionAttribute>().Name);
+                _collection = _database.GetCollection<T>(typeof(T).GetCustomAttribute<CollectionAttribute>().Name);
+
+                foreach (var prop in typeof(T).GetProperties())
+                {
+                    if (prop.GetCustomAttribute<UniqueAttribute>() != null)
+                    {
+                        var indices = _collection.Indexes.List().ToList().Select(x => JsonConvert.DeserializeObject<MongoIndex>(x.ToString())).ToList();
+
+                        if (!indices.Exists(x => x.name == prop.Name))
+                        {
+                            var builder = Builders<T>.IndexKeys;
+                            var indexModel = new CreateIndexModel<T>(builder.Ascending(prop.Name), new CreateIndexOptions()
+                            {
+                                Unique = true,
+                                Name = prop.Name,
+                            });
+
+                            _collection.Indexes.CreateOne(indexModel);
+                        }
+                    }
+                }
             }
 
             return _collection;