mongodb 和 C# 中的工作单元

使用MongoDB和C#的工作单元来提高开发效率

csharp
using MongoDB.Bson;
using MongoDB.Driver;
using MongoDB.Driver.Core.Bindings;
using System;
using System.Collections.Generic;
using System.Threading.Tasks;
public class UnitOfWork
{
    private readonly IMongoDatabase _database;
    private readonly IClientSessionHandle _session;
    public UnitOfWork(string connectionString, string databaseName)
    {
        var client = new MongoClient(connectionString);
        _database = client.GetDatabase(databaseName);
        _session = client.StartSession();
    }
    public IMongoCollection GetCollection(string collectionName)
    {
        return _database.GetCollection(collectionName);
    }
    public async Task CommitAsync()
    {
        await _session.CommitTransactionAsync();
    }
    public void Rollback()
    {
        _session.AbortTransaction();
    }
    public void Dispose()
    {
        _session.Dispose();
    }
}
public class UserRepository
{
    private readonly IMongoCollection _collection;
    public UserRepository(UnitOfWork unitOfWork)
    {
        _collection = unitOfWork.GetCollection("users");
    }
    public async Task> GetUsersAsync()
    {
        return await _collection.Find(new BsonDocument()).ToListAsync();
    }
    public async Task InsertUserAsync(BsonDocument user)
    {
        await _collection.InsertOneAsync(user);
    }
}
public class Program
{
    static async Task Main(string[] args)
    {
        var unitOfWork = new UnitOfWork("mongodb://localhost:27017", "test");
        var userRepository = new UserRepository(unitOfWork);
        try
        {
            var user = new BsonDocument
            {
                { "name", "John" },
                { "age", 30 }
            };
            await unitOfWork.BeginTransactionAsync();
            await userRepository.InsertUserAsync(user);
            var users = await userRepository.GetUsersAsync();
            foreach (var u in users)
            {
                Console.WriteLine(u);
            }
            await unitOfWork.CommitAsync();
        }
        catch (Exception ex)
        {
            unitOfWork.Rollback();
            Console.WriteLine(ex.Message);
        }
        finally
        {
            unitOfWork.Dispose();
        }
    }
}