EntityRepository.cs 3.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102
  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;
  9. using Quadarax.Foundation.Core.Data.Interface.Domain;
  10. using Quadarax.Foundation.Core.Data.Interface.Repository;
  11. namespace Quadarax.Foundation.Core.Data.Repository
  12. {
  13. public abstract class EntityRepository<TKey, TEntity> : IDaoRepository<TKey, TEntity>
  14. where TEntity : Entity<TKey>, new()
  15. where TKey : IEquatable<TKey>
  16. {
  17. private IUnitOfWork<IDataDomain> _unitOfWork;
  18. private DbSet<TEntity> DbSetInstance => _unitOfWork.Context.GetDbSet<DbSet<TEntity>>();
  19. protected IDataDomain Context => _unitOfWork.Context;
  20. protected IContext OperationContext { get; }
  21. protected IList<string> DefaultIncludes { get; }
  22. protected EntityRepository(IUnitOfWork<IDataDomain> unitOfWork)
  23. {
  24. _unitOfWork = unitOfWork ?? throw new ArgumentNullException(nameof(unitOfWork));
  25. DefaultIncludes = new List<string>();
  26. //_dbSet = domain.GetDbSet<DbSet<TEntity>>();
  27. }
  28. protected EntityRepository(IUnitOfWork<IDataDomain> unitOfWork, IContext currentOperationContext) : this(unitOfWork)
  29. {
  30. OperationContext = currentOperationContext ?? throw new ArgumentNullException(nameof(currentOperationContext));
  31. }
  32. public virtual IQueryable<TEntity> Query(IPaging paging,bool usingUncommited=false)
  33. {
  34. if (paging == null) throw new ArgumentNullException(nameof(paging));
  35. var local = DbSetInstance.Local.Page(paging.IsDisabled ? (int?) null : paging.Page, paging.PageSize).AsQueryable();
  36. if (local.Any() && usingUncommited)
  37. {
  38. return local.MultiInclude(DefaultIncludes.ToArray());
  39. }
  40. return DbSetInstance.Page(paging.IsDisabled ? (int?) null : paging.Page, paging.PageSize).MultiInclude(DefaultIncludes.ToArray());
  41. }
  42. public virtual TEntity Get(TKey id)
  43. {
  44. return Query(new Paging()).MultiInclude(DefaultIncludes.ToArray()).FirstOrDefault(x => Equals(x.Id, id));
  45. }
  46. public virtual IQueryable<TEntity> Get(IEnumerable<TKey> ids)
  47. {
  48. return Query(new Paging()).Where(x => ids.Contains(x.Id));
  49. }
  50. public virtual TEntity Set(TEntity entity)
  51. {
  52. //DbSetInstance.Update(entity);
  53. return entity;
  54. }
  55. public virtual IEnumerable<TEntity> Set(IEnumerable<TEntity> entities)
  56. {
  57. return entities.Select(Set).ToList();
  58. }
  59. public virtual TEntity New()
  60. {
  61. return DbSetInstance.Add(new TEntity()).Entity;
  62. }
  63. public virtual bool Remove(TKey id)
  64. {
  65. var dao = Get(id);
  66. if (dao == null) return false;
  67. DbSetInstance.Remove(dao);
  68. return true;
  69. }
  70. public virtual IEnumerable<bool> Remove(IEnumerable<TKey> ids)
  71. {
  72. return ids.Select(Remove).ToList();
  73. }
  74. public virtual bool Remove(TEntity entity)
  75. {
  76. DbSetInstance.Remove(entity);
  77. return true;
  78. }
  79. public virtual IEnumerable<bool> Remove(IEnumerable<TEntity> entities)
  80. {
  81. return entities.Select(Remove).ToList();
  82. }
  83. }
  84. }