From 45dccac0afeeb92930a67d72b482e773b6a4d057 Mon Sep 17 00:00:00 2001
From: shanj <18996038927@163.com>
Date: Tue, 1 Mar 2022 17:23:20 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E4=BA=A4=E6=98=93=E6=89=80?=
 =?UTF-8?q?=E8=B4=A6=E5=8F=B7=E6=8E=A5=E5=8F=A3=E6=9B=B4=E6=96=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../Exchange/ExchangeBusiness.cs              | 19 ++++++++++++++++---
 1 file changed, 16 insertions(+), 3 deletions(-)

diff --git a/Binance.TradeRobot.Business/Exchange/ExchangeBusiness.cs b/Binance.TradeRobot.Business/Exchange/ExchangeBusiness.cs
index 7409949..3e19a6d 100644
--- a/Binance.TradeRobot.Business/Exchange/ExchangeBusiness.cs
+++ b/Binance.TradeRobot.Business/Exchange/ExchangeBusiness.cs
@@ -21,10 +21,15 @@ namespace Binance.TradeRobot.Business.Exchange
 
         public void AddExchangeAccount(AddExchangeAccountRequest addExchangeAccountRequest)
         {
-            if (addExchangeAccountRequest.Id == 0 || string.IsNullOrEmpty(addExchangeAccountRequest.LoginName))
+            if (addExchangeAccountRequest.Id == 0 ||
+                string.IsNullOrEmpty(addExchangeAccountRequest.LoginName) ||
+                string.IsNullOrEmpty(addExchangeAccountRequest.APIKey) ||
+                string.IsNullOrEmpty(addExchangeAccountRequest.SecretKey))
                 throw new BusinessException("交易所账号参数有误");
             if (fsql.Select<ExchangeAccount>(addExchangeAccountRequest.Id).Any())
                 throw new BusinessException("交易所账号重复");
+            if (fsql.Select<ExchangeAPIKey>().Where(k => k.APIKey == addExchangeAccountRequest.APIKey || k.SecretKey == addExchangeAccountRequest.SecretKey).Any())
+                throw new BusinessException("重复的APIKey或SecretKey");
 
             var exchangeAccount = addExchangeAccountRequest.Map<ExchangeAccount>();
             if (addExchangeAccountRequest.TradePolicy == Enums.TradePolicy.金字塔)
@@ -33,12 +38,20 @@ namespace Binance.TradeRobot.Business.Exchange
             if (addExchangeAccountRequest.TradePolicy == Enums.TradePolicy.动量趋势v2)
                 exchangeAccount.BusinessType = Enums.BusinessType.Spot_Margin;
 
-            fsql.Transaction(() =>
+            var exchangeAPIKey = new ExchangeAPIKey()
             {
+                Id = idGenerator.NewLong(),
+                AccountId = addExchangeAccountRequest.Id,
+                APIKey = addExchangeAccountRequest.APIKey,
+                SecretKey = addExchangeAccountRequest.SecretKey
+            };
 
+            fsql.Transaction(() =>
+            {
+                fsql.Insert(exchangeAPIKey).ExecuteAffrows();
                 fsql.Insert(exchangeAccount).ExecuteAffrows();
             });
-           
+
         }
 
         public void AddExchangeAPIKey(AddExchangeAPIKeyRequest addExchangeAPIKeyRequest)