public void UpdateCriteriaDrivenBudget(CriteriaDrivenBudgetEntity entity)
{
entity.BUDGET_NAME = BudgetName;
entity.CRITERIA = Criteria;
}
/// <summary>
/// Executes an upsert/delete operation on a simulation's investment library data
/// Throws a RowNotInTableException if no simulation is found
/// </summary>
/// <param name="model">InvestmentLibraryModel</param>
/// <param name="db">BridgeCareContext</param>
/// <returns>InvestmentLibraryModel</returns>
private InvestmentLibraryModel SaveSimulationInvestmentLibrary(InvestmentLibraryModel model, BridgeCareContext db)
{
var id = int.Parse(model.Id);
var simulation = db.Simulations
.Include(s => s.INVESTMENTS)
.Include(s => s.YEARLYINVESTMENTS)
.Include(s => s.CriteriaDrivenBudgets)
.Include(s => s.PRIORITIES).Include(s => s.PRIORITIES.Select(p => p.PRIORITYFUNDS))
.Single(s => s.SIMULATIONID == id);
if (simulation.CriteriaDrivenBudgets.Any())
{
simulation.CriteriaDrivenBudgets.ToList().ForEach(criteriaDrivenBudget =>
{
var criteriaDrivenBudgetModel =
model.CriteriaDrivenBudgets.SingleOrDefault(m =>
m.Id == criteriaDrivenBudget.BUDGET_CRITERIA_ID.ToString());
if (criteriaDrivenBudgetModel == null)
{
CriteriaDrivenBudgetEntity.DeleteEntry(criteriaDrivenBudget, db);
}
else
{
criteriaDrivenBudgetModel.matched = true;
criteriaDrivenBudgetModel.UpdateCriteriaDrivenBudget(criteriaDrivenBudget);
}
});
}
if (simulation.INVESTMENTS != null)
{
model.UpdateInvestment(simulation.INVESTMENTS);
}
if (simulation.YEARLYINVESTMENTS.Any())
{
simulation.YEARLYINVESTMENTS.ToList().ForEach(yearlyInvestment =>
{
var yearlyInvestmentModel =
model.BudgetYears.SingleOrDefault(m => m.Id == yearlyInvestment.YEARID.ToString());
if (yearlyInvestmentModel == null)
{
YearlyInvestmentEntity.DeleteEntry(yearlyInvestment, db);
}
else
{
yearlyInvestmentModel.matched = true;
yearlyInvestmentModel.UpdateYearlyInvestment(yearlyInvestment);
}
});
}
simulation.PRIORITIES.ToList().ForEach(priorityEntity =>
{
var budgetsForNewFunds = new List <string>();
if (priorityEntity.PRIORITYFUNDS.Any())
{
model.BudgetOrder.ForEach(budget =>
{
budgetsForNewFunds.Add(budget);
});
}
if (budgetsForNewFunds.Any())
{
priorityEntity.PRIORITYFUNDS
.ToList()
.ForEach(priorityFundEntity =>
{
PriorityFundEntity.DeleteEntry(priorityFundEntity, db);
});
db.PriorityFunds.AddRange(budgetsForNewFunds
.Select(budget => new PriorityFundEntity(priorityEntity.PRIORITYID, budget))
);
}
});
if (simulation.INVESTMENTS == null)
{
db.Investments.Add(new InvestmentsEntity(model));
}
if (model.BudgetYears.Any(m => !m.matched))
{
db.YearlyInvestments.AddRange(model.BudgetYears
.Where(yearlyInvestmentModel => !yearlyInvestmentModel.matched)
.Select(yearlyInvestmentModel => new YearlyInvestmentEntity(id, yearlyInvestmentModel))
.ToList()
);
}
if (model.CriteriaDrivenBudgets.Any(m => !m.matched))
{
db.CriteriaDrivenBudgets.AddRange(model.CriteriaDrivenBudgets
.Where(criteriaDrivenBudgetModel => !criteriaDrivenBudgetModel.matched)
.Select(criteriaDrivenBudgetModel => new CriteriaDrivenBudgetEntity(id, criteriaDrivenBudgetModel))
.ToList()
);
}
db.SaveChanges();
return(new InvestmentLibraryModel(simulation));
}
public CriteriaDrivenBudgetModel(CriteriaDrivenBudgetEntity criteriaDrivenBudgetEntity)
{
Id = criteriaDrivenBudgetEntity.BUDGET_CRITERIA_ID.ToString();
BudgetName = criteriaDrivenBudgetEntity.BUDGET_NAME;
Criteria = criteriaDrivenBudgetEntity.CRITERIA;
}
