using AssetManager.Data.Repositories;
using Microsoft.Extensions.DependencyInjection;
using SqlSugar;

namespace AssetManager.Data;

public static class DatabaseExtensions
{
    public static IServiceCollection AddDatabase(this IServiceCollection services)
    {
        // 使用 Singleton 注册 SqlSugarScope（线程安全，内部使用 AsyncLocal）
        // SqlSugarScope 设计上支持并发，会自动管理连接池
        services.AddSingleton<ISqlSugarClient>(s =>
        {
            return SqlSugarConfig.GetSqlSugarClient();
        });

        services.AddScoped<DatabaseService>();
        
        // Repository 层
        services.AddScoped<IPortfolioRepository, PortfolioRepository>();
        services.AddScoped<IMarketDataRepository, MarketDataRepository>();

        return services;
    }

    public static void InitializeDatabase(this IServiceProvider serviceProvider)
    {
        using var scope = serviceProvider.CreateScope();
        var dbService = scope.ServiceProvider.GetRequiredService<DatabaseService>();
        dbService.InitializeDatabase();
    }
}