Adding coordinate to Location model

microservices/LocationsApi/Services/LocationStore.cs

@@ -1,4 +1,5 @@
 using LocationsApi.Models;
+using MongoDB.Bson;
 using MongoDB.Driver;
 
 namespace LocationsApi.Services;
@@ -13,8 +14,71 @@ public class LocationStore
         var dbName = cfg["MongoDB:DatabaseName"] ?? "GameDb";
         var client = new MongoClient(cs);
         var db = client.GetDatabase(dbName);
-        _col = db.GetCollection<Location>("Locations");
+        var collectionName = "Locations";
+        EnsureLocationSchema(db, collectionName);
+        _col = db.GetCollection<Location>(collectionName);
 
+        var coordIndex = Builders<Location>.IndexKeys
+            .Ascending(l => l.Coord.X)
+            .Ascending(l => l.Coord.Y);
+        var coordIndexOptions = new CreateIndexOptions { Unique = true };
+        _col.Indexes.CreateOne(new CreateIndexModel<Location>(coordIndex, coordIndexOptions));
+    }
+
+    private static void EnsureLocationSchema(IMongoDatabase db, string collectionName)
+    {
+        var validator = new BsonDocument
+        {
+            {
+                "$jsonSchema", new BsonDocument
+                {
+                    { "bsonType", "object" },
+                    { "required", new BsonArray { "name", "coord", "createdUtc" } },
+                    {
+                        "properties", new BsonDocument
+                        {
+                            { "name", new BsonDocument { { "bsonType", "string" } } },
+                            {
+                                "coord", new BsonDocument
+                                {
+                                    { "bsonType", "object" },
+                                    { "required", new BsonArray { "x", "y" } },
+                                    {
+                                        "properties", new BsonDocument
+                                        {
+                                            { "x", new BsonDocument { { "bsonType", "int" } } },
+                                            { "y", new BsonDocument { { "bsonType", "int" } } }
+                                        }
+                                    }
+                                }
+                            },
+                            { "createdUtc", new BsonDocument { { "bsonType", "date" } } }
+                        }
+                    }
+                }
+            }
+        };
+
+        var options = new CreateCollectionOptions
+        {
+            Validator = new BsonDocumentFilterDefinition<BsonDocument>(validator),
+            ValidationAction = DocumentValidationAction.Error
+        };
+
+        var collections = db.ListCollectionNames().ToList();
+        if (!collections.Contains(collectionName))
+        {
+            db.CreateCollection(collectionName, options);
+            return;
+        }
+
+        var command = new BsonDocument
+        {
+            { "collMod", collectionName },
+            { "validator", validator },
+            { "validationAction", "error" }
+        };
+        db.RunCommand<BsonDocument>(command);
     }
 
     public Task CreateAsync(Location location) => _col.InsertOneAsync(location);