refactor: use try register instead of doing it manually

Farfetch · Dec 6, 2023 · 2fff8f6 · 2fff8f6
1 parent 4f1b5c7
commit 2fff8f6
Showing 1 changed file with 11 additions and 17 deletions.
diff --git a/src/KafkaFlow.Retry.MongoDb/Model/DboConfigurations.cs b/src/KafkaFlow.Retry.MongoDb/Model/DboConfigurations.cs
@@ -18,15 +18,15 @@ internal static void TryAddIndexes(DbContext dbContext)
                     ),
                     new CreateIndexModel<RetryQueueDbo>(
                         Builders<RetryQueueDbo>.IndexKeys.Ascending(x => x.QueueGroupKey),
-                        new CreateIndexOptions{ Unique = true }
+                        new CreateIndexOptions { Unique = true }
                     ),
                     new CreateIndexModel<RetryQueueDbo>(
                         Builders<RetryQueueDbo>.IndexKeys.Ascending(x => x.Status)
                     ),
-                     new CreateIndexModel<RetryQueueDbo>(
+                    new CreateIndexModel<RetryQueueDbo>(
                         Builders<RetryQueueDbo>.IndexKeys.Descending(x => x.CreationDate)
                     ),
-                      new CreateIndexModel<RetryQueueDbo>(
+                    new CreateIndexModel<RetryQueueDbo>(
                         Builders<RetryQueueDbo>.IndexKeys.Ascending(x => x.LastExecution)
                     )
                 }
@@ -53,23 +53,17 @@ internal static void TryAddIndexes(DbContext dbContext)
 
         internal static void TryRegisterClassMapppings()
         {
-            if (!BsonClassMap.IsClassMapRegistered(typeof(RetryQueueDbo)))
+            BsonClassMap.TryRegisterClassMap<RetryQueueDbo>(cm =>
             {
-                BsonClassMap.RegisterClassMap<RetryQueueDbo>(cm =>
-                    {
-                        cm.AutoMap();
-                        cm.MapIdProperty(q => q.Id).SetIdGenerator(new GuidGenerator());
-                    });
-            }
+                cm.AutoMap();
+                cm.MapIdProperty(q => q.Id).SetIdGenerator(new GuidGenerator());
+            });
 
-            if (!BsonClassMap.IsClassMapRegistered(typeof(RetryQueueItemDbo)))
+            BsonClassMap.TryRegisterClassMap<RetryQueueItemDbo>(cm =>
             {
-                BsonClassMap.RegisterClassMap<RetryQueueItemDbo>(cm =>
-                    {
-                        cm.AutoMap();
-                        cm.MapIdProperty(q => q.Id).SetIdGenerator(new GuidGenerator());
-                    });
-            }
+                cm.AutoMap();
+                cm.MapIdProperty(q => q.Id).SetIdGenerator(new GuidGenerator());
+            });
         }
     }
 }