src/Orm/SqlSugar/SqlSugarServiceCollectionExt.cs

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

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

using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using SqlSugar;
using System;
using YL.Utils.Log;

namespace YL.Core.Orm.SqlSugar
{
    public static class SqlSugarServiceCollectionExt
    {
        public static IServiceCollection AddSqlSugarDao(this IServiceCollection services, Action<ConnectionConfig> configAction, ServiceLifetime lifetime = ServiceLifetime.Scoped)
        {
            var config = new ConnectionConfig();
            configAction.Invoke(config);
            switch (lifetime)
            {
                case ServiceLifetime.Singleton:
                    services.AddSingleton(new SqlSugarDao(config));
                    break;

                case ServiceLifetime.Scoped:
                    services.AddScoped(serviceProvider =>
                    {
                        return new SqlSugarDao(config);
                    });
                    break;

                case ServiceLifetime.Transient:
                    services.AddTransient(serviceProvider =>
                    {
                        return new SqlSugarDao(config);
                    });
                    break;

                default:
                    services.AddScoped(serviceProvider =>
                    {
                        return new SqlSugarDao(config);
                    });
                    break;
            }

            return services;
        }

        public static IServiceCollection AddSqlSugarClient<T>(this IServiceCollection services, Action<ConnectionConfig> configAction, ServiceLifetime lifetime = ServiceLifetime.Scoped) where T : SqlSugarClient
        {
            switch (lifetime)
            {
                case ServiceLifetime.Singleton:

                    services.AddSingleton(serviceProvider =>
                    {
                        var config = new ConnectionConfig()
                        {
                            ConfigureExternalServices = new ConfigureExternalServices
                            {
                                SqlFuncServices = SqlSugarConfig.GetLambda()
                            }
                        };