EntityRepository.cs 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Linq;
  4. using Microsoft.EntityFrameworkCore;
  5. using Quadarax.Foundation.Core.Data.Domain;
  6. using Quadarax.Foundation.Core.Data.Entity;
  7. using Quadarax.Foundation.Core.Data.Extensions;
  8. using Quadarax.Foundation.Core.Data.Interface.Domain;
  9. using Quadarax.Foundation.Core.Data.Interface.Repository;
  10. namespace Quadarax.Foundation.Core.Data.Repository
  11. {
  12. public abstract class EntityRepository<TKey, TEntity> : IDaoRepository<TKey, TEntity>
  13. where TEntity : Entity<TKey>, new()
  14. where TKey : IEquatable<TKey>
  15. {
  16. private IUnitOfWork<DataDomain> _unitOfWork;
  17. private DbSet<TEntity>? DbSetInstance => _unitOfWork.Context.GetDbSet<DbSet<TEntity>>();
  18. protected DataDomain Context => _unitOfWork.Context;
  19. protected IList<string> DefaultIncludes { get; }
  20. protected EntityRepository(IUnitOfWork<DataDomain> unitOfWork)
  21. {
  22. _unitOfWork = unitOfWork ?? throw new ArgumentNullException(nameof(unitOfWork));
  23. DefaultIncludes = new List<string>();
  24. //_dbSet = domain.GetDbSet<DbSet<TEntity>>();
  25. }
  26. protected virtual IQueryable<TEntity> OnQuery(IPaging paging,bool usingUncommited=false)
  27. {
  28. var local = DbSetInstance?.Local.Page(paging.IsDisabled ? (int?) null : paging.Page, paging.PageSize).AsQueryable();
  29. if (local !=null && local.Any() && usingUncommited)
  30. {
  31. return local.MultiInclude(DefaultIncludes.ToArray());
  32. }
  33. if (DbSetInstance == null)
  34. throw new InvalidOperationException("DbSet is null");
  35. return DbSetInstance.Page(paging.IsDisabled ? (int?) null : paging.Page, paging.PageSize).MultiInclude(DefaultIncludes.ToArray());
  36. }
  37. public IQueryable<TEntity> Query(IPaging paging, bool usingUncomited = false)
  38. {
  39. var query = OnQuery(paging, usingUncomited);
  40. if (query== null)
  41. throw new InvalidOperationException("DbSet is null");
  42. return query;
  43. }
  44. public virtual TEntity? Get(TKey id)
  45. {
  46. return Query(Paging.NoPaging).MultiInclude(DefaultIncludes.ToArray()).FirstOrDefault(x => Equals(x.Id, id));
  47. }
  48. public virtual IQueryable<TEntity> Get(IEnumerable<TKey> ids)
  49. {
  50. return Query(Paging.NoPaging).Where(x => ids.Contains(x.Id));
  51. }
  52. public virtual TEntity Set(TEntity entity)
  53. {
  54. //DbSetInstance.Update(entity);
  55. return entity;
  56. }
  57. public virtual IEnumerable<TEntity> Set(IEnumerable<TEntity> entities)
  58. {
  59. return entities.Select(Set).ToList();
  60. }
  61. public virtual TEntity New()
  62. {
  63. if (DbSetInstance == null)
  64. throw new InvalidOperationException("DbSet is null");
  65. return DbSetInstance.Add(new TEntity()).Entity;
  66. }
  67. public virtual bool Remove(TKey id)
  68. {
  69. if (DbSetInstance == null)
  70. throw new InvalidOperationException("DbSet is null");
  71. var entity = Get(id);
  72. if (entity == null)
  73. return false;
  74. DbSetInstance.Remove(entity);
  75. return true;
  76. }
  77. public virtual IEnumerable<bool> Remove(IEnumerable<TKey> ids)
  78. {
  79. return ids.Select(Remove).ToList();
  80. }
  81. public virtual bool Remove(TEntity entity)
  82. {
  83. if (DbSetInstance == null)
  84. throw new InvalidOperationException("DbSet is null");
  85. DbSetInstance.Remove(entity);
  86. return true;
  87. }
  88. public virtual IEnumerable<bool> Remove(IEnumerable<TEntity> entities)
  89. {
  90. return entities.Select(Remove).ToList();
  91. }
  92. public void Commit()
  93. {
  94. _unitOfWork.Commit();
  95. }
  96. }
  97. }