Refresh uuidMap after applying shardCollection DDL change and add tests.

Author: inelpandzic

pcsm/catalog.go

@@ -1188,7 +1188,6 @@ func (c *Catalog) ShardCollection(
 
 	log.Ctx(ctx).Debugf("Sharded collection %s.%s", db, coll)
 
-	// update c.databases[db].collections[coll].sharded = true
 	c.lock.Lock()
 	databaseEntry := c.Databases[db]
 	collectionEntry := databaseEntry.Collections[coll]

pcsm/repl.go

@@ -573,7 +573,7 @@ func (r *Repl) run(opts *options.ChangeStreamOptionsBuilder) {
 			metrics.AddEventsApplied(1)
 
 			switch change.OperationType { //nolint:exhaustive
-			case Create, Rename, Drop, DropDatabase:
+			case Create, Rename, Drop, DropDatabase, ShardCollection:
 				uuidMap = r.catalog.UUIDMap()
 			}
 		}
@@ -742,6 +742,8 @@ func (r *Repl) applyDDLChange(ctx context.Context, change *ChangeEvent) error {
 			event.OperationDescription.ShardKey,
 			event.OperationDescription.Unique)
 
+		lg.Infof("Collection %q has been sharded", change.Namespace)
+
 	case ReshardCollection:
 		fallthrough
 	case RefineCollectionShardKey:

tests/test_collections_sharded.py

@@ -57,6 +57,13 @@ def test_create_collection_with_collation(t: Testing, phase: Runner.Phase):
         t.source.admin.command(
             "shardCollection", "db_1.coll_1", key={"name": 1}, collation={"locale": "simple"}
         )
+        t.source["db_1"]["coll_1"].insert_many([{"name": "n3"}, {"name": "n2"}, {"name": "n3"}])
+
+        t.source["db_1"].create_collection("coll_2", collation={"locale": "en", "strength": 2})
+        t.source.admin.command(
+            "shardCollection", "db_1.coll_2", key={"_id": "hashed"}, collation={"locale": "simple"}
+        )
+        t.source["db_1"]["coll_2"].insert_many([{"_id": 11}, {"_id": 22}, {"_id": 33}])
 
     t.compare_all_sharded()