added MongoDbCollectionNameAttribute

Giorgio Tresoldi · Giorgio Tresoldi · commit 64588543deb3 · 2018-06-26T10:56:45.000+02:00
diff --git a/SharpRepository.MongoDbRepository/MongoDbCollectionNameAttribute.cs b/SharpRepository.MongoDbRepository/MongoDbCollectionNameAttribute.cs
@@ -0,0 +1,15 @@
+﻿using System;
+
+namespace SharpRepository.MongoDbRepository
+{
+    [AttributeUsage(AttributeTargets.Class)]
+    public class MongoDbCollectionNameAttribute: Attribute
+    {
+        public string CollectionName { get; }
+
+        public MongoDbCollectionNameAttribute(string collectionName)
+        {
+            CollectionName = collectionName;
+        }
+    }
+}
diff --git a/SharpRepository.MongoDbRepository/MongoDbRepositoryBase.cs b/SharpRepository.MongoDbRepository/MongoDbRepositoryBase.cs
@@ -22,6 +22,7 @@ namespace SharpRepository.MongoDbRepository
     public class MongoDbRepositoryBase<T, TKey> : LinqRepositoryBase<T, TKey> where T : class, new()
     {
         private readonly string _databaseName;
+        private string _collectionName;
         protected IMongoDatabase Database;
         static readonly object _lock = new object();
 
@@ -107,11 +108,17 @@ private void Initialize(IMongoDatabase mongoDatabase = null)
                         }
                     );
                 }
+
+            var collectionNameAttributes = EntityType.GetTypeInfo().GetOneAttribute<MongoDbCollectionNameAttribute>(inherit: true);
+            if (collectionNameAttributes != null)
+                _collectionName = collectionNameAttributes.CollectionName;
+            else
+                _collectionName = TypeName;
         }
 
         private IMongoCollection<T> BaseCollection()
         {
-            return Database.GetCollection<T>(TypeName);
+            return Database.GetCollection<T>(_collectionName);
         }
 
         protected override IQueryable<T> BaseQuery(IFetchStrategy<T> fetchStrategy = null)

Original file line number	Diff line number	Diff line change
`@@ -22,6 +22,7 @@ namespace SharpRepository.MongoDbRepository`
`22`	`22`	`public class MongoDbRepositoryBase<T, TKey> : LinqRepositoryBase<T, TKey> where T : class, new()`
`23`	`23`	`{`
`24`	`24`	`private readonly string _databaseName;`
	`25`	`+ private string _collectionName;`
`25`	`26`	`protected IMongoDatabase Database;`
`26`	`27`	`static readonly object _lock = new object();`
`27`	`28`
`@@ -107,11 +108,17 @@ private void Initialize(IMongoDatabase mongoDatabase = null)`
`107`	`108`	`}`
`108`	`109`	`);`
`109`	`110`	`}`
	`111`	`+`
	`112`	`+ var collectionNameAttributes = EntityType.GetTypeInfo().GetOneAttribute<MongoDbCollectionNameAttribute>(inherit: true);`
	`113`	`+ if (collectionNameAttributes != null)`
	`114`	`+ _collectionName = collectionNameAttributes.CollectionName;`
	`115`	`+ else`
	`116`	`+ _collectionName = TypeName;`
`110`	`117`	`}`
`111`	`118`
`112`	`119`	`private IMongoCollection<T> BaseCollection()`
`113`	`120`	`{`
`114`		`- return Database.GetCollection<T>(TypeName);`
	`121`	`+ return Database.GetCollection<T>(_collectionName);`
`115`	`122`	`}`
`116`	`123`
`117`	`124`	`protected override IQueryable<T> BaseQuery(IFetchStrategy<T> fetchStrategy = null)`