Include automatic incrementing counter to missionRun model

equinor · Feb 5, 2024 · ab2685b · ab2685b
1 parent e37c570
commit ab2685b
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 0 deletions.
diff --git a/backend/api/Database/Context/FlotillaDbContext.cs b/backend/api/Database/Context/FlotillaDbContext.cs
@@ -55,6 +55,7 @@ protected override void OnModelCreating(ModelBuilder modelBuilder)
                 .Property(m => m.InspectionFrequency)
                 .HasConversion(new TimeSpanToTicksConverter());
 
+            modelBuilder.Entity<MissionRun>().Property(m => m.MissionRunCount).ValueGeneratedOnAdd();
             modelBuilder.Entity<MissionRun>().OwnsOne(m => m.Map).OwnsOne(t => t.TransformationMatrices);
             modelBuilder.Entity<MissionRun>().OwnsOne(m => m.Map).OwnsOne(b => b.Boundary);
             modelBuilder.Entity<Robot>().OwnsOne(r => r.Pose).OwnsOne(p => p.Orientation);
@@ -98,6 +99,30 @@ protected override void ConfigureConventions(ModelConfigurationBuilder configura
             configurationBuilder.Properties(typeof(Enum)).HaveConversion<string>();
         }
 
+        public static void CreateTriggerForMissionRunsInsert(FlotillaDbContext dbContext)
+        {
+            string sql = @"CREATE TRIGGER IF NOT EXISTS tr_insert_mission_runs_inspection_id 
+                            AFTER INSERT ON MissionRuns
+                            BEGIN
+                                UPDATE MissionRuns 
+                                SET InspectionId = (SELECT MAX(Id) FROM Inspections)
+                                WHERE Id = NEW.Id;
+                            END";
+
+            if (dbContext.Database.IsSqlite())
+            {
+                dbContext.Database.ExecuteSqlRaw(sql);
+            }
+            else if (dbContext.Database.IsSqlServer())
+            {
+                throw new NotImplementedException();
+            }
+            else
+            {
+                throw new InvalidOperationException("Unknown database provider.");
+            }
+        }
+
         private static void AddConverterForDateTimeOffsets<T>(ref EntityTypeBuilder<T> entity)
             where T : class
         {

diff --git a/backend/api/Database/Context/InitDb.cs b/backend/api/Database/Context/InitDb.cs
@@ -567,6 +567,7 @@ private static List<MissionRun> GetMissionRuns()
             var missionRun1 = new MissionRun
             {
                 Name = "Placeholder Mission 1",
+                MissionRunCount = 1,
                 Robot = robots[0],
                 InstallationCode = areas[0].Installation!.InstallationCode,
                 Area = areas[0],
@@ -580,6 +581,7 @@ private static List<MissionRun> GetMissionRuns()
             var missionRun2 = new MissionRun
             {
                 Name = "Placeholder Mission 2",
+                MissionRunCount = 2,
                 Robot = robots[1],
                 InstallationCode = areas[1].Installation!.InstallationCode,
                 Area = areas[1],
@@ -594,6 +596,7 @@ private static List<MissionRun> GetMissionRuns()
             var missionRun3 = new MissionRun
             {
                 Name = "Placeholder Mission 3",
+                MissionRunCount = 3,
                 Robot = robots[2],
                 InstallationCode = areas[1].Installation!.InstallationCode,
                 Area = areas[1],
@@ -607,6 +610,7 @@ private static List<MissionRun> GetMissionRuns()
             var missionRun4 = new MissionRun
             {
                 Name = "Placeholder Mission 4",
+                MissionRunCount = 4,
                 Robot = robots[2],
                 InstallationCode = areas[1].Installation.InstallationCode,
                 Area = areas[1],
@@ -624,6 +628,7 @@ private static List<MissionRun> GetMissionRuns()
             var missionRun5 = new MissionRun
             {
                 Name = "Placeholder Mission 5",
+                MissionRunCount = 5,
                 Robot = robots[2],
                 InstallationCode = areas[1].Installation.InstallationCode,
                 Area = areas[1],
@@ -641,6 +646,7 @@ private static List<MissionRun> GetMissionRuns()
             var missionRun6 = new MissionRun
             {
                 Name = "Placeholder Mission 6",
+                MissionRunCount = 6,
                 Robot = robots[2],
                 InstallationCode = areas[1].Installation.InstallationCode,
                 Area = areas[1],
@@ -658,6 +664,7 @@ private static List<MissionRun> GetMissionRuns()
             var missionRun7 = new MissionRun
             {
                 Name = "Some failed tasks",
+                MissionRunCount = 7,
                 Robot = robots[2],
                 InstallationCode = areas[1].Installation.InstallationCode,
                 Area = areas[1],

diff --git a/backend/api/Database/Models/MissionRun.cs b/backend/api/Database/Models/MissionRun.cs
@@ -6,6 +6,9 @@ namespace Api.Database.Models
 {
     public class MissionRun : SortableRecord
     {
+        [Required]
+        [DatabaseGenerated(DatabaseGeneratedOption.Computed)]
+        public int MissionRunCount { get; set; } = 1;
 
         private MissionStatus _status;