using Binance.TradeRobot.Model.Base;
using System.Collections.Generic;

namespace Binance.TradeRobot.Business.Extensions
{
    public static class RobotExtension
    {
        private static IDictionary<Enums.TradePolicy, Enums.BusinessType> BusinessTypeDic = new Dictionary<Enums.TradePolicy, Enums.BusinessType>()
        {
            {Enums.TradePolicy.Pyramid, Enums.BusinessType.UPrep },
            { Enums.TradePolicy.D2, Enums.BusinessType.IsolateMargin},
            { Enums.TradePolicy.D21, Enums.BusinessType.IsolateMargin}
        };

        /// <summary>
        /// 获取交易策略对应的业务类型
        /// </summary>
        /// <param name="tradePolicy"></param>
        /// <returns></returns>
        public static Enums.BusinessType GetBusinessType(this Enums.TradePolicy tradePolicy)
        {
            return BusinessTypeDic[tradePolicy];
        }
    }
}