using AssetManager.Infrastructure.StrategyEngine.Calculators;
using Microsoft.Extensions.DependencyInjection;

namespace AssetManager.Infrastructure.StrategyEngine;

/// <summary>
/// 策略引擎服务注册扩展
/// </summary>
public static class StrategyEngineExtensions
{
    /// <summary>
    /// 注册策略引擎及所有计算器
    /// </summary>
    public static IServiceCollection AddStrategyEngine(this IServiceCollection services)
    {
        // 注册策略计算器
        services.AddScoped<IStrategyCalculator, MaTrendCalculator>();
        services.AddScoped<IStrategyCalculator, ChandelierExitCalculator>();
        services.AddScoped<IStrategyCalculator, RiskParityCalculator>();

        // 注册策略引擎
        services.AddScoped<IStrategyEngine, StrategyEngine>();

        return services;
    }
}