public void Remove <TEntity>(TEntity entity) where TEntity : Entity { if (entity != null) { context.Entry(entity).State = System.Data.Entity.EntityState.Deleted; } }
private void CopyReturn(CopyReturn message, Return returnCopy) { returnCopy.ResetSubmitted(userContext.UserId.ToString(), message.ReturnId); context.Entry(returnCopy.Organisation).State = EntityState.Unchanged; context.Returns.Add(returnCopy); }
public async Task RemoveAatfData(Aatf aatf, IEnumerable <int> quarters) { foreach (var quarter in quarters) { var quarterWindow = await quarterWindowFactory.GetAnnualQuarter(new Quarter(aatf.ComplianceYear, (QuarterType)quarter)); var aatfCount = await context.Aatfs.CountAsync(a => a.Organisation.Id == aatf.Organisation.Id && a.ComplianceYear == aatf.ComplianceYear && a.FacilityType.Value == aatf.FacilityType.Value && a.ApprovalDate.Value <= quarterWindow.EndDate); IEnumerable <WeeeSentOn> weeeSentOn; IEnumerable <WeeeReused> weeeReused; IEnumerable <WeeeReceived> weeeReceived; IEnumerable <ReturnAatf> returnAatfs; var weeeReceivedAmounts = new List <WeeeReceivedAmount>().AsEnumerable(); var weeeReusedAmounts = new List <WeeeReusedAmount>().AsEnumerable(); var weeeReusedSites = new List <WeeeReusedSite>().AsEnumerable(); var weeeSentOnAmounts = new List <WeeeSentOnAmount>().AsEnumerable(); if (aatfCount > 0) { if (aatfCount == 1) { var returns = context.Returns.Where(r => r.Organisation.Id == aatf.Organisation.Id && (int)r.Quarter.Q == quarter && r.Quarter.Year == aatf.ComplianceYear && r.FacilityType.Value == aatf.FacilityType.Value); var returnIds = returns.Select(r => r.Id).ToList(); returnAatfs = context.ReturnAatfs.Where(r => returnIds.Contains(r.Return.Id)); var returnReportsOn = context.ReturnReportOns.Where(r => returnIds.Contains(r.Return.Id)); var returnScheme = context.ReturnScheme.Where(r => returnIds.Contains(r.Return.Id)); var nonObligated = context.NonObligatedWeee.Where(r => returnIds.Contains(r.Return.Id)); weeeSentOn = context.WeeeSentOn.Where(w => returnIds.Contains(w.Return.Id)).Cast <WeeeSentOn>(); weeeReused = context.WeeeReused.Where(w => returnIds.Contains(w.Return.Id)).Cast <WeeeReused>(); weeeReceived = context.WeeeReceived.Where(w => returnIds.Contains(w.Return.Id)).Cast <WeeeReceived>(); foreach (var @return in returns) { context.Entry(@return).Entity.ParentId = null; context.Entry(@return).State = EntityState.Modified; } await context.SaveChangesAsync(); context.ReturnScheme.RemoveRange(returnScheme); context.ReturnReportOns.RemoveRange(returnReportsOn); context.Returns.RemoveRange(returns); context.NonObligatedWeee.RemoveRange(nonObligated); } else { weeeSentOn = context.WeeeSentOn.Where(ObligatedByAatfComplianceYearAndQuarter(aatf, quarter)).Cast <WeeeSentOn>(); weeeReused = context.WeeeReused.Where(ObligatedByAatfComplianceYearAndQuarter(aatf, quarter)).Cast <WeeeReused>(); weeeReceived = context.WeeeReceived.Where(ObligatedByAatfComplianceYearAndQuarter(aatf, quarter)).Cast <WeeeReceived>(); returnAatfs = context.ReturnAatfs.Where(r => r.Aatf.Id == aatf.Id && r.Return.Organisation.Id == aatf.Organisation.Id && r.Return.Quarter.Year == aatf.ComplianceYear && (int)r.Return.Quarter.Q == quarter); } weeeReceivedAmounts = context.WeeeReceivedAmount.Where(w => weeeReceived.Select(wr => wr.Id).Contains(w.WeeeReceived.Id)); weeeReusedAmounts = context.WeeeReusedAmount.Where(w => weeeReused.Select(wr => wr.Id).Contains(w.WeeeReused.Id)); weeeReusedSites = context.WeeeReusedSite.Where(w => weeeReused.Select(wr => wr.Id).Contains(w.WeeeReused.Id)); weeeSentOnAmounts = context.WeeeSentOnAmount.Where(w => weeeSentOn.Select(wr => wr.Id).Contains(w.WeeeSentOn.Id)); context.WeeeSentOnAmount.RemoveRange(weeeSentOnAmounts); context.WeeeReusedSite.RemoveRange(weeeReusedSites); context.WeeeReusedAmount.RemoveRange(weeeReusedAmounts); context.WeeeReceivedAmount.RemoveRange(weeeReceivedAmounts); context.WeeeReceived.RemoveRange(weeeReceived); context.WeeeReused.RemoveRange(weeeReused); context.WeeeSentOn.RemoveRange(weeeSentOn); context.ReturnAatfs.RemoveRange(returnAatfs); } } await context.SaveChangesAsync(); }