DD.WellWorkover.Cloud/AsbCloudInfrastructure/Services/ClusterService.cs

using AsbCloudApp.Data;
using AsbCloudApp.Services;
using AsbCloudDb.Model;
using Microsoft.EntityFrameworkCore;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;

namespace AsbCloudInfrastructure.Services
{
    public class ClusterService : IClusterService
    {
        private readonly IAsbCloudDbContext db;

        public ClusterService(IAsbCloudDbContext db)
        {
            this.db = db;
        }

        public async Task<IEnumerable<DepositDto>> GetDepositsAsync(int idCompany,
            CancellationToken token = default)
        {
            var wellEntities = await (from well in db.Wells
                .Include(w => w.RelationCompaniesWells)
                .Include(w => w.WellType)
                .Include(w => w.Cluster)
                .ThenInclude(c => c.Deposit)
                                      where well.RelationCompaniesWells.Any(r => r.IdCompany == idCompany)
                                      select well).ToListAsync(token)
                                .ConfigureAwait(false);

            var gDepositEntities = wellEntities
                .GroupBy(w => w.Cluster)
                .GroupBy(c => c.Key.Deposit);

            var dtos = gDepositEntities.Select(gDeposit => new DepositDto
            {
                Id = gDeposit.Key.Id,
                Caption = gDeposit.Key.Caption,
                Latitude = gDeposit.Key.Latitude,
                Longitude = gDeposit.Key.Longitude,
                Description = "",
                Clusters = gDeposit.Select(gCluster => new ClusterDto
                {
                    Id = gCluster.Key.Id,
                    Caption = gCluster.Key.Caption,
                    Latitude = gCluster.Key.Latitude,
                    Longitude = gCluster.Key.Longitude,
                    Description = "",
                    Wells = gCluster.Select(well => new WellDto
                    {
                        Id = well.Id,
                        Caption = well.Caption,
                        Latitude = well.Latitude,
                        Longitude = well.Longitude,
                        WellType = well.WellType?.Caption,
                        Cluster = gCluster.Key.Caption,
                        Deposit = gDeposit.Key.Caption,
                    }),
                }),
            });

            return dtos;
        }

        public async Task<IEnumerable<ClusterDto>> GetClustersAsync(int idCompany,
            CancellationToken token = default)
        {
            var entities = await db.GetWellsForCompany(idCompany)
                        .Select(e => e.Cluster)
                        .Distinct()
                        .AsNoTracking()
                        .ToListAsync(token)
                        .ConfigureAwait(false);

            var dtos = entities.Select(e => new ClusterDto
            {
                Id = e.Id,
                Caption = e.Caption,
                Latitude = e.Latitude,
                Longitude = e.Longitude,
            });

            return dtos;
        }

        public async Task<IEnumerable<ClusterDto>> GetClustersAsync(int idCompany,
            int depositId, CancellationToken token = default)
        {
            var entities = await db.GetWellsForCompany(idCompany)
                        .Select(e => e.Cluster)
                        .Where(e => e.IdDeposit == depositId)
                        .Distinct()
                        .AsNoTracking()
                        .ToListAsync(token)
                        .ConfigureAwait(false);

            var dtos = entities.Select(e => new ClusterDto
            {
                Id = e.Id,
                Caption = e.Caption,
                Latitude = e.Latitude,
                Longitude = e.Longitude,
            });

            return dtos;
        }

        public async Task<IEnumerable<WellDto>> GetWellsAsync(int idCompany,
            int idCluster, CancellationToken token = default)
        {
            var entities = await db.GetWellsForCompany(idCompany)
                .Where(e => e.IdCluster == idCluster)
                .AsNoTracking()
                .ToListAsync(token)
                .ConfigureAwait(false);

            var dtos = entities.Select(e => new WellDto
            {
                Id = e.Id,
                Caption = e.Caption,
                Latitude = e.Latitude,
                Longitude = e.Longitude,
                Cluster = e.Cluster.Caption,
                Deposit = e.Cluster.Deposit.Caption,
            });

            return dtos;
        }

        public async Task<ClusterStatDto> GetStatAsync(int idCompany,
            int idCluster, CancellationToken token = default)
        {
            var wellEntities = from w in db.Wells
                               where w.IdCluster == idCluster && w.RelationCompaniesWells.Any(c => c.IdCompany == idCompany)
                               select w;

            var wellStatDtos = await wellEntities.Select(e => new WellStatDto
            {
                Id = e.Id,
                Caption = e.Caption,
                Cluster = e.Cluster.Caption,
                Deposit = e.Cluster.Deposit.Caption,
                Latitude = e.Latitude,
                Longitude = e.Longitude,
                //TODO: Fill data from operations service
                Companies = e.RelationCompaniesWells.Select(c => new CompanyDto
                {
                    Id = c.Company.Id,
                    Caption = c.Company.Caption,
                    CompanyType = c.Company.CompanyType.Caption,
                }),
                WellType = e.WellType.Caption,
            }).AsNoTracking().ToListAsync(token)
            .ConfigureAwait(false);

            if (!wellStatDtos.Any())
                return null;

            var clusterById = await db.Clusters.AsNoTracking()
                .FirstOrDefaultAsync(c => c.Id == idCluster, token)
                .ConfigureAwait(false);

            return new ClusterStatDto
            {
                Id = clusterById.Id,
                Description = "",
                Caption = clusterById.Caption,
                Latitude = clusterById.Latitude,
                Longitude = clusterById.Longitude,
                WellsStat = wellStatDtos,
            };
        }
    }
}