SemianiakaVY/BotSharp
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

minor change

Browse source
This commit is contained in:
Jicheng Lu 2025-02-19 11:16:51 -06:00
parent 391aa0edef
commit c3459e34d4
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/Plugins/BotSharp.Plugin.MongoStorage/MongoDbContext.cs
View file

@ -48,10 +48,11 @@ public class MongoDbContext
private IMongoDatabase Database => _mongoClient.GetDatabase(_mongoDbDatabaseName);
#region Private methods
private bool CollectionExists(IMongoDatabase database, string collectionName)
{
var filter = Builders<BsonDocument>.Filter.Eq("name", collectionName);
var collections = database.ListCollections(new ListCollectionsOptions { Filter = filter });
var collections = database.ListCollectionNames(new ListCollectionNamesOptions { Filter = filter }).ToList();
return collections.Any();
}
@ -60,9 +61,11 @@ public class MongoDbContext
if (string.IsNullOrWhiteSpace(name))
throw new ArgumentException($"The collection {name} cannot be empty.");
string collectionName = $"{_collectionPrefix}_{name}";
var collectionName = $"{_collectionPrefix}_{name}";
if (!CollectionExists(Database, collectionName))
{
Database.CreateCollection(collectionName);
}
var collection = Database.GetCollection<TDocument>(collectionName);
return collection;
@ -134,6 +137,7 @@ public class MongoDbContext
return collection;
}
#endregion
#endregion
public IMongoCollection<AgentDocument> Agents
=> GetCollectionOrCreate<AgentDocument>("Agents");