EntityRepository.cs 2.7 KB

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