StatisticsDao.cs 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Linq;
  4. using System.Threading.Tasks;
  5. using MafiaTelegramBot.DataBase.Entity;
  6. using MafiaTelegramBot.Resources;
  7. using Microsoft.EntityFrameworkCore;
  8. namespace MafiaTelegramBot.DataBase.EntityDao
  9. {
  10. public static class StatisticsDao
  11. {
  12. private static readonly MafiaDataBase DataBase = MafiaDataBase.GetInstance();
  13. public static async Task<Dictionary<Roles, StatisticsEntity>> GetStatisticsListById(long userId)
  14. {
  15. if (await UserDataExists(userId))
  16. {
  17. var list = await DataBase.Statistics.Where(s => s.UserId == userId).ToListAsync();
  18. return list.ToDictionary(item => Enum.Parse<Roles>(item.Role), item => item);
  19. }
  20. var roles = Enum.GetValues(typeof(Roles)).Cast<Roles>();
  21. foreach (var role in roles)
  22. {
  23. var statsRow = new StatisticsEntity { UserId = userId, Role = role.ToString() };
  24. await DataBase.Statistics.AddAsync(statsRow);
  25. }
  26. await DataBase.SaveChangesAsync();
  27. var result = await DataBase.Statistics.Where(s => s.UserId == userId).ToListAsync();
  28. return result.ToDictionary(item => Enum.Parse<Roles>(item.Role), item => item);
  29. }
  30. private static async Task<bool> UserDataExists(long id)
  31. {
  32. return await DataBase.Statistics.AnyAsync(entity => entity.UserId == id);
  33. }
  34. }
  35. }