src/02_Data/Data.Core/DbBuilder.cs

这里只显示每个文件的开头 60 行。登录后可以解锁完整代码。

微信扫码关注公众号，代码整理好后第一时间通知你。

using System;
using System.Collections.Generic;
using System.Linq;
using System.Reflection;
using System.Runtime.Loader;
using Microsoft.Extensions.DependencyInjection;
using Mkh.Data.Abstractions;
using Mkh.Data.Abstractions.Adapter;
using Mkh.Data.Abstractions.Logger;
using Mkh.Data.Abstractions.Options;
using Mkh.Data.Abstractions.Schema;
using Mkh.Data.Abstractions.Sharding;
using Mkh.Data.Core.Descriptors;
using Mkh.Data.Core.Sharding;
using Mkh.Data.Core.Sharding.Providers;

namespace Mkh.Data.Core;

internal class DbBuilder : IDbBuilder
{
    private readonly IList<Assembly> _repositoryAssemblies = new List<Assembly>();
    private readonly List<Action> _actions = new();
    private readonly Type _dbContextType;

    public IServiceCollection Services { get; set; }

    public DbOptions Options { get; set; }

    public CodeFirstOptions CodeFirstOptions { get; set; }

    public IDbContext DbContext { get; set; }

    public DbBuilder(IServiceCollection services, DbOptions options, Type dbContextType)
    {
        Services = services;
        Options = options;
        _dbContextType = dbContextType;
    }

    public IDbBuilder AddRepositoriesFromAssembly(Assembly assembly)
    {
        if (assembly == null)
            return this;

        _repositoryAssemblies.Add(assembly);
        return this;
    }

    public IDbBuilder AddAction(Action action)
    {
        if (action == null)
            return this;

        _actions.Add(action);
        return this;
    }

    public void Build()
    {
        if (Options.Provider != DbProvider.Sqlite)