added get async

2019-02-12 22:53:22 -05:00
parent f3cc1f2fe3
commit 2bf96b1588
3 changed files with 57 additions and 7 deletions
@@ -1,4 +1,5 @@
 using System;
+using System.Collections.Generic;
 using System.Linq;
 using System.Linq.Expressions;
 using System.Reflection;
@@ -38,12 +39,7 @@ namespace PoweredSoft.ObjectStorage.MongoDB

        protected virtual Expression<Func<TEntity, bool>> CreateEntityExpression(TEntity entity)
        {
-            var objectKey = typeof(TEntity)
-                            .GetProperties(BindingFlags.Public | BindingFlags.Instance)
-                            .FirstOrDefault(t => t.GetCustomAttribute<BsonIdAttribute>() != null);
-            if (objectKey == null)
-                throw new Exception("Must have a BsonIdAttribute on one of the properties."); 
-                
+            var objectKey = GetKeyProperty();
            var constant = objectKey.GetValue(entity);
            var expression = QueryableHelpers.CreateConditionExpression<TEntity>(objectKey.Name,
                DynamicLinq.ConditionOperators.Equal, constant, DynamicLinq.QueryConvertStrategy.LeaveAsIs);
@@ -51,6 +47,16 @@ namespace PoweredSoft.ObjectStorage.MongoDB
            return expression;
        }

+        protected virtual PropertyInfo GetKeyProperty()
+        {
+            var objectKey = typeof(TEntity)
+                            .GetProperties(BindingFlags.Public | BindingFlags.Instance)
+                            .FirstOrDefault(t => t.GetCustomAttribute<BsonIdAttribute>() != null);
+            if (objectKey == null)
+                throw new Exception("Must have a BsonIdAttribute on one of the properties.");
+            return objectKey;
+        }
+
        public async Task DeleteAsync(TEntity entity, CancellationToken cancellationToken = default(CancellationToken))
        {
            var expression = CreateEntityExpression(entity);
@@ -63,5 +69,24 @@ namespace PoweredSoft.ObjectStorage.MongoDB
            await Collection.ReplaceOneAsync(expression, entity);
            return entity;
        }
+
+        public Task<List<TEntity>> GetAllAsync(CancellationToken cancellationToken = default(CancellationToken))
+        {
+            return Collection.AsQueryable().ToListAsync(cancellationToken);
+        }
+
+        public Task<List<TEntity>> GetAllAsync(Expression<Func<TEntity, bool>> predicate, CancellationToken cancellationToken = default(CancellationToken))
+        {
+            return Collection.Find(predicate).ToListAsync(cancellationToken);
+        }
+
+        public Task<TEntity> GetAsync(object key, CancellationToken cancellationToken = default(CancellationToken))
+        {
+            var keyProp = GetKeyProperty();
+            var expression = QueryableHelpers.CreateConditionExpression<TEntity>(keyProp.Name,
+                DynamicLinq.ConditionOperators.Equal, key, DynamicLinq.QueryConvertStrategy.LeaveAsIs);
+            var result = Collection.Find(expression).FirstOrDefaultAsync();
+            return result;
+        }
    }
 }