public SplitTreatmentModel(SplitTreatmentEntity entity)
{
Id = entity.SPLIT_TREATMENT_ID.ToString();
Description = entity.DESCRIPTION;
Criteria = entity.CRITERIA;
SplitTreatmentLimits = entity.SPLIT_TREATMENT_LIMITS.Any()
? entity.SPLIT_TREATMENT_LIMITS.Select(stl => new SplitTreatmentLimitModel(stl)).ToList()
: new List <SplitTreatmentLimitModel>();
}
/// <summary>
/// Executes an upsert/delete operation on a simulation's cash flow library data
/// Throws a RowNotInTableException if no simulation is found
/// </summary>
/// <param name="model">CashFlowLibraryModel</param>
/// <param name="db">BridgeCareContext</param>
/// <returns></returns>
public CashFlowLibraryModel SaveSimulationCashFlowLibrary(CashFlowLibraryModel model, BridgeCareContext db)
{
var id = int.Parse(model.Id);
if (!db.Simulations.Any(s => s.SIMULATIONID == id))
{
throw new RowNotInTableException($"No scenario found with id {id}");
}
var simulation = db.Simulations
.Include(s => s.SPLIT_TREATMENTS)
.Include(s => s.SPLIT_TREATMENTS.Select(st => st.SPLIT_TREATMENT_LIMITS))
.Single(s => s.SIMULATIONID == id);
if (simulation.SPLIT_TREATMENTS.Any())
{
simulation.SPLIT_TREATMENTS.ToList().ForEach(splitTreatment =>
{
var splitTreatmentModel =
model.SplitTreatments.SingleOrDefault(m =>
m.Id == splitTreatment.SPLIT_TREATMENT_ID.ToString());
if (splitTreatmentModel == null)
{
SplitTreatmentEntity.DeleteEntry(splitTreatment, db);
}
else
{
splitTreatmentModel.matched = true;
splitTreatmentModel.UpdateSplitTreatment(splitTreatment);
if (splitTreatment.SPLIT_TREATMENT_LIMITS.Any())
{
splitTreatment.SPLIT_TREATMENT_LIMITS.ToList().ForEach(splitTreatmentLimit =>
{
var splitTreatmentLimitModel =
splitTreatmentModel.SplitTreatmentLimits.SingleOrDefault(m =>
m.Id == splitTreatmentLimit.SPLIT_TREATMENT_LIMIT_ID.ToString());
if (splitTreatmentLimitModel == null)
{
SplitTreatmentLimitEntity.DeleteEntry(splitTreatmentLimit, db);
}
else
{
splitTreatmentLimitModel.matched = true;
splitTreatmentLimitModel.UpdateSplitTreatment(splitTreatmentLimit);
}
});
}
if (splitTreatmentModel.SplitTreatmentLimits.Any(m => !m.matched))
{
db.SplitTreatmentLimits.AddRange(splitTreatmentModel.SplitTreatmentLimits
.Where(splitTreatmentLimitModel => !splitTreatmentLimitModel.matched)
.Select(splitTreatmentLimitModel =>
new SplitTreatmentLimitEntity(splitTreatment.SPLIT_TREATMENT_ID, splitTreatmentLimitModel))
.ToList()
);
}
}
});
}
if (model.SplitTreatments.Any(m => !m.matched))
{
db.SplitTreatments.AddRange(model.SplitTreatments
.Where(splitTreatmentModel => !splitTreatmentModel.matched)
.Select(splitTreatmentModel => new SplitTreatmentEntity(id, splitTreatmentModel))
.ToList()
);
}
db.SaveChanges();
return(new CashFlowLibraryModel(simulation));
}
public void UpdateSplitTreatment(SplitTreatmentEntity entity)
{
entity.DESCRIPTION = Description;
entity.CRITERIA = Criteria;
}