CalvinBackup/notesnook-sync-server_customPrivateV1
1
0
Fork 0
mirror of https://github.com/streetwriters/notesnook-sync-server.git synced 2026-07-08 20:07:52 +02:00
Code Issues Packages Projects Releases Wiki Activity

sync: make collection & db name usage more obvious

Browse Source
This commit is contained in:
Abdullah Atta
2023-09-09 20:28:46 +05:00
parent 8bbb4d0b9e
commit 55a7e9fd1c
12 changed files with 46 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Streetwriters.Data/Repositories/Repository.cs
+4 -9
View File
@@ -24,7 +24,6 @@ using System.Linq.Expressions;
using System.Threading.Tasks;
using MongoDB.Bson;
using MongoDB.Driver;
using Streetwriters.Data.Attributes;
using Streetwriters.Data.Interfaces;
namespace Streetwriters.Data.Repositories
@@ -34,19 +33,15 @@ namespace Streetwriters.Data.Repositories
protected readonly IDbContext dbContext;
protected IMongoCollection<TEntity> Collection { get; set; }
public Repository(IDbContext _dbContext)
public Repository(IDbContext _dbContext, string databaseName, string collectionName)
{
dbContext = _dbContext;
Collection = GetCollection();
Collection = GetCollection(databaseName, collectionName);
}
private protected IMongoCollection<TEntity> GetCollection()
private protected IMongoCollection<TEntity> GetCollection(string databaseName, string collectionName)
{
var attribute = (BsonCollectionAttribute)typeof(TEntity).GetCustomAttributes(
typeof(BsonCollectionAttribute),
true).FirstOrDefault();
if (string.IsNullOrEmpty(attribute.CollectionName) || string.IsNullOrEmpty(attribute.DatabaseName)) throw new Exception("Could not get a valid collection or database name.");
return dbContext.GetCollection<TEntity>(attribute.DatabaseName, attribute.CollectionName);
return dbContext.GetCollection<TEntity>(databaseName, collectionName);
}