EntityRepository.cs 3.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394
  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 IList<string> DefaultIncludes { get; }
  21. protected EntityRepository(IUnitOfWork<IDataDomain> unitOfWork)
  22. {
  23. _unitOfWork = unitOfWork ?? throw new ArgumentNullException(nameof(unitOfWork));
  24. DefaultIncludes = new List<string>();
  25. //_dbSet = domain.GetDbSet<DbSet<TEntity>>();
  26. }
  27. public virtual IQueryable<TEntity> Query(IPaging paging,bool usingUncommited=false)
  28. {
  29. var local = DbSetInstance.Local.Page(paging.IsDisabled ? (int?) null : paging.Page, paging.PageSize).AsQueryable();
  30. if (local.Any() && usingUncommited)
  31. {
  32. return local.MultiInclude(DefaultIncludes.ToArray());
  33. }
  34. return DbSetInstance.Page(paging.IsDisabled ? (int?) null : paging.Page, paging.PageSize).MultiInclude(DefaultIncludes.ToArray());
  35. }
  36. public virtual TEntity Get(TKey id)
  37. {
  38. return Query(new Paging()).MultiInclude(DefaultIncludes.ToArray()).FirstOrDefault(x => Equals(x.Id, id));
  39. }
  40. public virtual IQueryable<TEntity> Get(IEnumerable<TKey> ids)
  41. {
  42. return Query(new Paging()).Where(x => ids.Contains(x.Id));
  43. }
  44. public virtual TEntity Set(TEntity entity)
  45. {
  46. //DbSetInstance.Update(entity);
  47. return entity;
  48. }
  49. public virtual IEnumerable<TEntity> Set(IEnumerable<TEntity> entities)
  50. {
  51. return entities.Select(Set).ToList();
  52. }
  53. public virtual TEntity New()
  54. {
  55. return DbSetInstance.Add(new TEntity()).Entity;
  56. }
  57. public virtual bool Remove(TKey id)
  58. {
  59. var dao = Get(id);
  60. if (dao == null) return false;
  61. DbSetInstance.Remove(dao);
  62. return true;
  63. }
  64. public virtual IEnumerable<bool> Remove(IEnumerable<TKey> ids)
  65. {
  66. return ids.Select(Remove).ToList();
  67. }
  68. public virtual bool Remove(TEntity entity)
  69. {
  70. DbSetInstance.Remove(entity);
  71. return true;
  72. }
  73. public virtual IEnumerable<bool> Remove(IEnumerable<TEntity> entities)
  74. {
  75. return entities.Select(Remove).ToList();
  76. }
  77. }
  78. }