using Microsoft.EntityFrameworkCore; using System; using System.Collections.Generic; using System.Linq; using System.Threading; using System.Threading.Tasks; using System.Data.Common; #nullable disable namespace AsbCloudDb.Model { //Scaffold-DbContext "Host=localhost;Database=postgres;Username=postgres;Password=q;Persist Security Info=True" Npgsql.EntityFrameworkCore.PostgreSQL -OutputDir Model -DataAnnotations public partial class AsbCloudDbContext : DbContext, IAsbCloudDbContext { public virtual DbSet Clusters { get; set; } public virtual DbSet Companies { get; set; } public virtual DbSet DataSaubBases { get; set; } public virtual DbSet Deposits { get; set; } public virtual DbSet Users { get; set; } public virtual DbSet UserRoles { get; set; } public virtual DbSet Wells { get; set; } public virtual DbSet ReportProperties { get; set; } public virtual DbSet Files { get; set; } public virtual DbSet FileCategories { get; set; } public virtual DbSet Telemetries { get; set; } public virtual DbSet TelemetryEvents { get; set; } public virtual DbSet TelemetryMessages { get; set; } public virtual DbSet TelemetryUsers { get; set; } public virtual DbSet TelemetryOperations { get; set; } public virtual DbSet TelemetryAnalysis { get; set; } public virtual DbSet WellSections { get; set; } public virtual DbSet WellSectionTypes { get; set; } public virtual DbSet WellOperations { get; set; } public virtual DbSet WellTypes { get; set; } public virtual DbSet LastData { get; set; } public virtual DbSet TvdData { get; set; } //var options = new DbContextOptionsBuilder() // .UseNpgsql("Host=localhost;Database=postgres;Username=postgres;Password=q;Persist Security Info=True") // .Options; //var context = new AsbCloudDbContext(options); public AsbCloudDbContext(DbContextOptions options) : base(options) { Database.EnsureCreated(); } //protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder) //{ // if (!optionsBuilder.IsConfigured) // { // optionsBuilder.UseNpgsql(connectionString); // } //} protected override void OnModelCreating(ModelBuilder modelBuilder) { modelBuilder.HasPostgresExtension("adminpack") .HasAnnotation("Relational:Collation", "Russian_Russia.1251"); modelBuilder.Entity(entity => { entity.HasOne(d => d.Deposit) .WithMany(p => p.Clusters) .HasForeignKey(d => d.IdDeposit) .HasConstraintName("t_cluster_t_deposit_id_fk"); }); modelBuilder.Entity(entity => { entity.HasOne(d => d.Telemetry) .WithMany(p => p.DataSaubBases) .HasForeignKey(d => d.IdTelemetry) .OnDelete(DeleteBehavior.ClientSetNull) .HasConstraintName("t_data_saub_base_t_telemetry_id_fk"); }); modelBuilder.Entity(entity => { entity.HasOne(d => d.Telemetry) .WithMany(p => p.Messages) .HasForeignKey(d => d.IdTelemetry) .HasConstraintName("t_messages_t_telemetry_id_fk"); }); modelBuilder.Entity(entity => { entity.HasKey(nameof(TelemetryUser.IdTelemetry), nameof(TelemetryUser.IdUser)); entity.HasOne(d => d.Telemetry) .WithMany(p => p.Users) .HasForeignKey(d => d.IdTelemetry) .OnDelete(DeleteBehavior.ClientSetNull) .HasConstraintName("t_telemetry_user_t_telemetry_id_fk"); }); modelBuilder.Entity(entity => { entity.HasOne(d => d.Telemetry) .WithMany(p => p.Analysis) .HasForeignKey(d => d.IdTelemetry) .HasConstraintName("t_analysis_t_telemetry_id_fk"); entity.HasOne(d => d.Operation) .WithMany(p => p.Analysis) .HasForeignKey(d => d.IdOperation) .OnDelete(DeleteBehavior.SetNull) .HasConstraintName("t_analysis_t_operation_id_fk"); }); modelBuilder.Entity(entity => { entity.HasKey(nameof(TelemetryEvent.IdTelemetry), nameof(TelemetryEvent.IdEvent)); entity.HasOne(d => d.Telemetry) .WithMany(p => p.Events) .HasForeignKey(d => d.IdTelemetry) .HasConstraintName("t_event_t_telemetry_id_fk"); }); modelBuilder.Entity(entity => { entity.HasOne(d => d.Company) .WithMany(p => p.Users) .HasForeignKey(d => d.IdCompany) .HasConstraintName("t_user_t_company_id_fk"); entity.HasIndex(d => d.Login) .IsUnique(); }); modelBuilder.Entity(entity => { entity.HasOne(d => d.Cluster) .WithMany(p => p.Wells) .HasForeignKey(d => d.IdCluster) .HasConstraintName("t_well_t_cluster_id_fk"); entity.HasOne(d => d.Telemetry) .WithOne(p => p.Well) .HasForeignKey(d => d.IdTelemetry) .HasConstraintName("t_well_t_telemetry_id_fk"); }); modelBuilder.Entity(entity => { entity.HasIndex(e => e.WellDepthPlan) .HasDatabaseName("IX_t_well_section_well_depth_plan"); }); modelBuilder.Entity(entity => { entity.HasKey(nameof(RelationCompanyWell.IdCompany), nameof(RelationCompanyWell.IdWell)); entity.HasOne(r => r.Well) .WithMany(w => w.RelationCompaniesWells) .HasForeignKey(r => r.IdWell) .HasConstraintName("t_relation_company_well_t_well_id_fk"); entity.HasOne(r => r.Company) .WithMany(w => w.RelationCompaniesWells) .HasForeignKey(r => r.IdCompany) .HasConstraintName("t_relation_company_well_t_company_id_fk"); }); modelBuilder.Entity(entity => { entity.HasIndex(d => d.Depth); entity.HasIndex(d => d.Date); }); FillData(modelBuilder); FillDemoData(modelBuilder); } private static void FillData(ModelBuilder modelBuilder) { modelBuilder.Entity(entity => { entity.HasData(new List{ new UserRole{ Id = 1, Caption = "Администратор", }, }); }); modelBuilder.Entity(entity => { entity.HasData(new List{ new User{ Id = 1, IdCompany = 1, IdRole = 1, Level = int.MaxValue, Login = "dev", PasswordHash = "Vlcj|4fa529103dde7ff72cfe76185f344d4aa87931f8e1b2044e8a7739947c3d18923464eaad93843e4f809c5e126d013072", // dev Name = "Разработчик", }, }); }); modelBuilder.Entity(entity => { entity.HasData(new List{ new Company{ Id = 1, Caption = "ООО \"АСБ\"", IdCompanyType = 3}, }); }); modelBuilder.Entity(entity => { entity.HasData(new List { new Operation {Id = 1, OperationType = 0, Name = "Невозможно определить операцию"}, new Operation {Id = 2, OperationType = 0, Name = "Роторное бурение" }, new Operation {Id = 3, OperationType = 0, Name = "Слайдирование" }, new Operation {Id = 4, OperationType = 0, Name = "Подъем с проработкой" }, new Operation {Id = 5, OperationType = 0, Name = "Спуск с проработкой" }, new Operation {Id = 6, OperationType = 0, Name = "Подъем с промывкой" }, new Operation {Id = 7, OperationType = 0, Name = "Спуск с промывкой" }, new Operation {Id = 8, OperationType = 0, Name = "Спуск в скважину" }, new Operation {Id = 9, OperationType = 0, Name = "Спуск с вращением" }, new Operation {Id = 10, OperationType = 0, Name = "Подъем из скважины" }, new Operation {Id = 11, OperationType = 0, Name = "Подъем с вращением" }, new Operation {Id = 12, OperationType = 0, Name = "Промывка в покое" }, new Operation {Id = 13, OperationType = 0, Name = "Промывка с вращением" }, new Operation {Id = 14, OperationType = 0, Name = "Удержание в клиньях" }, new Operation {Id = 15, OperationType = 0, Name = "Неподвижное состояние" }, new Operation {Id = 16, OperationType = 0, Name = "Вращение без циркуляции" }, new Operation {Id = 17, OperationType = 0, Name = "На поверхности" } }); }); modelBuilder.Entity(entity => { entity.HasData(new List { new FileCategory {Id = 1, Name = "Растворный сервис", ShortName = "fluidService"}, new FileCategory {Id = 2, Name = "Цементирование", ShortName = "cement"}, new FileCategory {Id = 3, Name = "ННБ", ShortName = "nnb"}, new FileCategory {Id = 4, Name = "ГТИ", ShortName = "gti"}, new FileCategory {Id = 5, Name = "Документы по скважине", ShortName = "wellDocuments"}, new FileCategory {Id = 6, Name = "Супервайзер", ShortName = "supervisor"}, new FileCategory {Id = 7, Name = "Мастер", ShortName = "master"}, new FileCategory {Id = 8, Name = "Последний замер бурового раствора ПЛАН", ShortName = "fluidPlanLastData"}, new FileCategory {Id = 9, Name = "Последний замер бурового раствора ФАКТ", ShortName = "fluidFactLastData"}, new FileCategory {Id = 10, Name = "Последние данные Шламограммы", ShortName = "mudLastData"}, new FileCategory {Id = 11, Name = "Последние данные ННБ", ShortName = "nnbLastData"}, new FileCategory {Id = 12, Name = "Рапорт", ShortName = "report"} }); }); modelBuilder.Entity(entity => { entity.HasData(new List{ new WellSectionType{ Id = 1, Caption = "Пилотный ствол"}, new WellSectionType{ Id = 2, Caption = "Направление"}, new WellSectionType{ Id = 3, Caption = "Кондуктор"}, new WellSectionType{ Id = 4, Caption = "Эксплуатационная колонна"}, new WellSectionType{ Id = 5, Caption = "Транспортный ствол"}, new WellSectionType{ Id = 6, Caption = "Хвостовик"}, }); }); } private static void FillDemoData(ModelBuilder modelBuilder) { modelBuilder.Entity(entity => { entity.HasData(new List{ new CompanyType{ Id = 1, Caption = "Недрапользователь", }, new CompanyType{ Id = 2, Caption = "Буровой подрядчик", }, new CompanyType{ Id = 3, Caption = "Сервис автоматизации бурения", }, }); }); modelBuilder.Entity(entity => { entity.HasData(new List { new Deposit{Id = 1, Caption = "м/р 1", Latitude = 60.8705722222222, Longitude = 70.3811888888889 }, }); }); modelBuilder.Entity(entity => { entity.HasData(new List { new Cluster{Id = 1, Caption = "к39.1", IdDeposit = 1, Latitude = 60.8705722222222, Longitude = 70.3811888888889}, new Cluster{Id = 2, Caption = "к151", IdDeposit = 1, Latitude = 60.8205750000000, Longitude = 70.1343833333334}, new Cluster{Id = 3, Caption = "к611", IdDeposit = 1, Latitude = 60.8100666666667, Longitude = 69.7778388888889}, new Cluster{Id = 4, Caption = "к203", IdDeposit = 1, Latitude = 60.8928805555556, Longitude = 70.3272055555556}, new Cluster{Id = 5, Caption = "к221", IdDeposit = 1, Latitude = 60.6672055555556, Longitude = 69.6603861111111}, }); }); modelBuilder.Entity(entity => { entity.HasData(new List{ new Telemetry{ Id = 1, RemoteUid = "123", Info = new TelemetryInfo { Well = "1", Cluster = "1", Comment = "", Deposit = "1", Customer = "1", HmiVersion = "1", PlcVersion = "1", TimeZoneId = "1", DrillingStartDate = DateTime.Parse("2021-06-29T12:01:19.000000"), TimeZoneOffsetTotalHours = 5.0 }, } }); }); modelBuilder.Entity(entity => { entity.HasData(new List { new Well{Id = 1, IdCluster = 1, IdTelemetry = 1, Caption = "скв 16314", Latitude = 60.8705722222222, Longitude = 70.3811888888889}, new Well{Id = 2, IdCluster = 1, Caption = "скв 16311", Latitude = 60.8705722222222, Longitude = 70.3811888888889}, new Well{Id = 3, IdCluster = 2, Caption = "скв 16315", Latitude = 60.8205750000000, Longitude = 70.1343833333334}, new Well{Id = 4, IdCluster = 2, Caption = "скв 16318", Latitude = 60.8205750000000, Longitude = 70.1343833333334}, new Well{Id = 5, IdCluster = 3, Caption = "скв 16310", Latitude = 60.8100666666667, Longitude = 69.7778388888889}, new Well{Id = 6, IdCluster = 4, Caption = "скв 16316", Latitude = 60.8928805555556, Longitude = 70.3272055555556}, new Well{Id = 7, IdCluster = 5, Caption = "скв 16312", Latitude = 60.6672055555556, Longitude = 69.6603861111111}, new Well{Id = 8, IdCluster = 5, Caption = "скв 16313", Latitude = 60.6672055555556, Longitude = 69.6603861111111}, new Well{Id = 9, IdCluster = 5, Caption = "скв 42669", Latitude = 60.6672055555556, Longitude = 69.6603861111111}, }); }); modelBuilder.Entity(entity => { entity.HasData(new List { new RelationCompanyWell{ IdWell = 1, IdCompany = 1}, new RelationCompanyWell{ IdWell = 2, IdCompany = 1}, new RelationCompanyWell{ IdWell = 3, IdCompany = 1}, new RelationCompanyWell{ IdWell = 4, IdCompany = 1}, new RelationCompanyWell{ IdWell = 5, IdCompany = 1}, new RelationCompanyWell{ IdWell = 6, IdCompany = 1}, new RelationCompanyWell{ IdWell = 7, IdCompany = 1}, new RelationCompanyWell{ IdWell = 8, IdCompany = 1}, new RelationCompanyWell{ IdWell = 9, IdCompany = 1}, }); }); modelBuilder.Entity(entity => { entity.HasData(new List{ new TelemetryAnalysis { Id = 1, IdTelemetry = 1, IdOperation = 17, UnixDate = 1626870355, DurationSec = 10, OperationStartDepth = null, OperationEndDepth = 206, IsWellDepthIncreasing = false, IsWellDepthDecreasing = false, IsBitPositionIncreasing = false, IsBitPositionDecreasing = false, IsBitPositionLt20 = true, IsBlockPositionIncreasing = false, IsBlockPositionDecreasing = false, IsRotorSpeedLt3 = true, IsRotorSpeedGt3 = false, IsPressureLt20 = true, IsPressureGt20 = false, IsHookWeightNotChanges = true, IsHookWeightLt3 = true }, new TelemetryAnalysis { Id = 2, IdTelemetry = 1, IdOperation = 8, UnixDate = 1626870364, DurationSec = 6, OperationStartDepth = 206, OperationEndDepth = 206, IsWellDepthIncreasing = false, IsWellDepthDecreasing = false, IsBitPositionIncreasing = true, IsBitPositionDecreasing = false, IsBitPositionLt20 = true, IsBlockPositionIncreasing = true, IsBlockPositionDecreasing = false, IsRotorSpeedLt3 = true, IsRotorSpeedGt3 = false, IsPressureLt20 = true, IsPressureGt20 = false, IsHookWeightNotChanges = true, IsHookWeightLt3 = true }, new TelemetryAnalysis { Id = 3, IdTelemetry = 1, IdOperation = 10, UnixDate = 1626870370, DurationSec = 2, OperationStartDepth = 206, OperationEndDepth = 206, IsWellDepthIncreasing = false, IsWellDepthDecreasing = false, IsBitPositionIncreasing = false, IsBitPositionDecreasing = true, IsBitPositionLt20 = true, IsBlockPositionIncreasing = false, IsBlockPositionDecreasing = true, IsRotorSpeedLt3 = true, IsRotorSpeedGt3 = false, IsPressureLt20 = true, IsPressureGt20 = false, IsHookWeightNotChanges = true, IsHookWeightLt3 = true }, new TelemetryAnalysis { Id = 4, IdTelemetry = 1, IdOperation = 17, UnixDate = 1626870372, DurationSec = 7, OperationStartDepth = 206, OperationEndDepth = 206, IsWellDepthIncreasing = false, IsWellDepthDecreasing = false, IsBitPositionIncreasing = false, IsBitPositionDecreasing = false, IsBitPositionLt20 = true, IsBlockPositionIncreasing = false, IsBlockPositionDecreasing = false, IsRotorSpeedLt3 = true, IsRotorSpeedGt3 = false, IsPressureLt20 = true, IsPressureGt20 = false, IsHookWeightNotChanges = true, IsHookWeightLt3 = true }, new TelemetryAnalysis { Id = 5, IdTelemetry = 1, IdOperation = 8, UnixDate = 1626870379, DurationSec = 7, OperationStartDepth = 206, OperationEndDepth = 206, IsWellDepthIncreasing = false, IsWellDepthDecreasing = false, IsBitPositionIncreasing = true, IsBitPositionDecreasing = false, IsBitPositionLt20 = true, IsBlockPositionIncreasing = true, IsBlockPositionDecreasing = false, IsRotorSpeedLt3 = true, IsRotorSpeedGt3 = false, IsPressureLt20 = true, IsPressureGt20 = false, IsHookWeightNotChanges = true, IsHookWeightLt3 = true } }); }); } public IQueryable GetWellsForCompany(int idCompany) { return from well in Wells .Include(w => w.RelationCompaniesWells) .ThenInclude(r => r.Company) .Include(w => w.Cluster) .ThenInclude(c => c.Deposit) where well.RelationCompaniesWells.Any(c => c.IdCompany == idCompany) select well; } public IQueryable GetUsersByLogin(string login) => Users .Include(e => e.Role) .Include(e => e.Company) .Where(e => e.Login == login); public async Task<(DateTime From, DateTime To)> GetDatesRangeAsync(int idTelemetry, CancellationToken token = default) where TEntity : class, IIdTelemetryDate { var dbSet = Set(); var datesRange = await (from m in dbSet where m.IdTelemetry == idTelemetry group m by m.IdTelemetry into g select new { From = g.Min(d => d.Date), To = g.Max(d => d.Date) }).AsNoTracking().FirstOrDefaultAsync(token); if (datesRange is null) return (DateTime.MinValue, DateTime.MaxValue); return (datesRange.From, datesRange.To); } public async Task> GetDepthToIntervalAsync(int telemetryId, int intervalHoursTimestamp, int workStartTimestamp, double timezoneOffset, CancellationToken token) { //TODO: Сменить на LINQ группирование using var command = Database.GetDbConnection().CreateCommand(); command.CommandText = $@"SELECT Min(t.bit_depth) AS MinDepth, Max(t.bit_depth) AS MaxDepth, Min(t.Date) AS dateStart FROM t_data_saub_base AS t WHERE id_telemetry = {telemetryId} AND t.Id % 10 = 0 GROUP BY floor((extract(epoch from t.date) - {workStartTimestamp} + {timezoneOffset}) / {intervalHoursTimestamp});"; Database.OpenConnection(); using var reader = await command.ExecuteReaderAsync(token); IEnumerable<(double? MinDepth, double? MaxDepth, DateTime BeginPeriodDate)> GetResult(DbDataReader rd) { if (rd.HasRows) { while (reader.Read()) { yield return ( (double?)reader.GetValue(0), (double?)reader.GetValue(1), (DateTime)reader.GetValue(2) ); } } } return GetResult(reader); } public async Task CreatePartitionAsync(string propertyName, int id, CancellationToken token = default) where TEntity : class { var dbSet = Set(); var baseTableName = dbSet.EntityType.GetTableName(); var schema = dbSet.EntityType.GetSchema(); var tableObject = Microsoft.EntityFrameworkCore.Metadata.StoreObjectIdentifier.Table(baseTableName, schema); var tableName = baseTableName.Replace("_base", ""); var property = dbSet.EntityType.GetProperty(propertyName).GetColumnName(tableObject); var query = $"CREATE TABLE {tableName}_{id} (like {baseTableName} including all, constraint partitioning_check check ({property} = 1)) INHERITS ({baseTableName});"; return await Database.ExecuteSqlRawAsync(query, token).ConfigureAwait(false); } } }