slight refactor to InsertNotExists to return entity

threenine · Dec 20, 2021 · cfced34 · cfced34
1 parent 75052c4
commit cfced34
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/src/IRepository.cs b/src/IRepository.cs
@@ -35,7 +35,8 @@ public interface IRepository<T> : IReadRepository<T>, IDisposable where T : clas
         void Insert(params T[] entities);
         void Insert(IEnumerable<T> entities);
 
-        void InsertNotExists(Expression<Func<T, bool>> predicate, T entity);
+        T InsertNotExists(Expression<Func<T, bool>> predicate, T entity);
+
 
         void Update(T entity);
         void Update(params T[] entities);

diff --git a/src/Repository.cs b/src/Repository.cs
@@ -90,11 +90,14 @@ public void Insert(IEnumerable<T> entities)
             _dbSet.AddRange(entities);
         }
 
-        public void InsertNotExists(Expression<Func<T, bool>> predicate, T entity)
+        public T InsertNotExists(Expression<Func<T, bool>> predicate, T entity)
         {
-            var exists = predicate != null ? _dbSet.Any(predicate) : _dbSet.Any();
-            if (!exists) _dbSet.Add(entity);
+            if (_dbSet.Any(predicate)) return _dbSet.SingleOrDefault(predicate);
+            _dbSet.Add(entity);
+            return entity;
+
         }
+
 
         #endregion