public async Task <SpendingResponse> GetSpendingData([FromBody] SpendingRequest request)
{
var data = request.Mode switch
{
SpendingMode.Month => await CalculateSpendingData(
request, x => new MonthAndYear(x.Date.Month, x.Date.Year),
x => $"{x.Month:D2}.{x.Year}"),
SpendingMode.Year => await CalculateSpendingData(
request, x => x.Date.Year,
x => x.ToString()),
_ => throw new ArgumentOutOfRangeException()
};
return(data);
}
private (LocalDate, LocalDate) GetDateRange(SpendingRequest request)
{
if (request.Mode == SpendingMode.Month)
{
var monthRange = request.MonthRange ?? throw new InvalidOperationException("Month range is not specified");
return(new LocalDate(monthRange.From.Year, monthRange.From.Month, 1), AdjsutEndMonthRange(monthRange.To));
}
if (request.Mode == SpendingMode.Year)
{
var yearRange = request.YearRange ?? throw new InvalidOperationException("Year range is not specified");
return(new LocalDate(yearRange.From, 1, 1), new LocalDate(yearRange.To + 1, 1, 1));
}
throw new InvalidOperationException("Unknown display mode");
}
private async Task <SpendingResponse> CalculateSpendingData <TGroupKey>(
SpendingRequest request,
Func <Transaction, TGroupKey> groupKey,
Func <TGroupKey, string> groupFormatter)
where TGroupKey : notnull
{
var tags = await GetTagsWithChildren(request.Tags);
var(startDate, endDate) = GetDateRange(request);
var allTransactions = await _context.Transactions
.Include(x => x.Tags)
.ThenInclude(x => x.ParentTag)
.Where(x => ((x.IncomeAccount != null && x.IncomeAccount.InBalance) ||
(x.OutcomeAccount != null && x.OutcomeAccount.InBalance)) &&
x.Income == 0 &&
x.Outcome > 0 &&
x.Tags.Any(t => tags.Contains(t.Id)) &&
x.Date >= startDate &&
x.Date < endDate)
.ToArrayAsync();
var grouped = new SortedDictionary <TGroupKey, Dictionary <Tag, List <Transaction> > >();
var tagTotals = new Dictionary <Tag, decimal>();
foreach (var group in allTransactions.GroupBy(groupKey).OrderBy(x => x.Key))
{
var byTag = new Dictionary <Tag, List <Transaction> >();
foreach (var transaction in group)
{
var tag = transaction.Tags.Count > 0
? transaction.Tags[0]
: new Tag {
Id = Guid.Empty.ToString(), Title = "<Unspecified>"
};
while (tag.ParentTag != null)
{
tag = tag.ParentTag;
}
byTag.GetOrAdd(tag, () => new List <Transaction>()).Add(transaction);
tagTotals.AddOrUpdate(tag, () => transaction.Amount, cur => cur + transaction.Amount);
}
grouped[group.Key] = byTag;
}
var sortedTags = tagTotals.OrderBy(x => x.Value).Select(x => x.Key).ToArray();
var series = new List <SpendingSerie>();
foreach (var tag in sortedTags)
{
var totals = new decimal[grouped.Count];
foreach (var(month, i) in grouped.Enumerate())
{
if (!month.Value.TryGetValue(tag, out var transactions))
{
continue;
}
totals[i] = transactions.Sum(x => x.Amount);
}
series.Add(new(tag.Title ?? "???", totals));
}
return(new(grouped.Select(x => groupFormatter(x.Key)).ToArray(), series.ToArray()));
}
