diff --git a/Orleans.Providers.MongoDB/Utils/CollectionBase.cs b/Orleans.Providers.MongoDB/Utils/CollectionBase.cs index 9dcbd22..77d9d91 100644 --- a/Orleans.Providers.MongoDB/Utils/CollectionBase.cs +++ b/Orleans.Providers.MongoDB/Utils/CollectionBase.cs @@ -22,12 +22,25 @@ public class CollectionBase private readonly IMongoDatabase mongoDatabase; private readonly IMongoClient mongoClient; - private readonly Lazy> mongoCollection; + private readonly Action collectionConfigurator; + private IMongoCollection mongoCollection; + private readonly object mongoCollectionInitializerLock = new(); private readonly bool createShardKey; protected IMongoCollection Collection { - get { return mongoCollection.Value; } + get + { + if (mongoCollection == null) + { + lock (mongoCollectionInitializerLock) + { + mongoCollection ??= CreateCollection(collectionConfigurator); + } + } + + return mongoCollection; + } } protected IMongoDatabase Database @@ -44,9 +57,9 @@ public IMongoClient Client Action collectionConfigurator, bool createShardKey) { this.mongoClient = mongoClient; + this.collectionConfigurator = collectionConfigurator; mongoDatabase = mongoClient.GetDatabase(databaseName); - mongoCollection = CreateCollection(collectionConfigurator); this.createShardKey = createShardKey; } @@ -65,51 +78,40 @@ protected virtual void SetupCollection(IMongoCollection collection) { } - private Lazy> CreateCollection(Action collectionConfigurator) + private IMongoCollection CreateCollection(Action collectionConfigurator) { - return new Lazy>(() => - { - var collectionFilter = new ListCollectionNamesOptions - { - Filter = Builders.Filter.Eq("name", CollectionName()) - }; + var collectionName = CollectionName(); - if (!mongoDatabase.ListCollectionNames(collectionFilter).Any()) - { - mongoDatabase.CreateCollection(CollectionName()); - } - - var collectionSettings = CollectionSettings() ?? new MongoCollectionSettings(); + var collectionSettings = CollectionSettings() ?? new MongoCollectionSettings(); - collectionConfigurator?.Invoke(collectionSettings); + collectionConfigurator?.Invoke(collectionSettings); - var databaseCollection = mongoDatabase.GetCollection( - CollectionName(), - collectionSettings); + var databaseCollection = mongoDatabase.GetCollection( + collectionName, + collectionSettings); - if (this.createShardKey) + if (createShardKey) + { + try { - try + mongoClient.GetDatabase("admin").RunCommand(new BsonDocument { - Database.RunCommand(new BsonDocument + ["shardCollection"] = $"{mongoDatabase.DatabaseNamespace.DatabaseName}.{collectionName}", + ["key"] = new BsonDocument { - ["key"] = new BsonDocument - { - ["_id"] = "hashed" - }, - ["shardCollection"] = $"{mongoDatabase.DatabaseNamespace.DatabaseName}.{CollectionName()}" - }); - } - catch (MongoException) - { - // Shared key probably created already. - } + ["_id"] = "hashed" + } + }); + } + catch (MongoException) + { + // Shared key probably created already. } + } - SetupCollection(databaseCollection); + SetupCollection(databaseCollection); - return databaseCollection; - }); + return databaseCollection; } } } \ No newline at end of file