From 171628f9a463f5af3ce55d897fa37b1b80ea4e8b Mon Sep 17 00:00:00 2001
From: zj <1772600164@qq.com>
Date: Thu, 15 Aug 2024 17:17:54 +0800
Subject: [PATCH] 1

---
 src/main/java/project/contract/job/ContractOrderCalculationServiceImpl.java |   80 +++++++++++++++++++++++++++++-----------
 1 files changed, 58 insertions(+), 22 deletions(-)

diff --git a/src/main/java/project/contract/job/ContractOrderCalculationServiceImpl.java b/src/main/java/project/contract/job/ContractOrderCalculationServiceImpl.java
index 49051f9..1fc7433 100644
--- a/src/main/java/project/contract/job/ContractOrderCalculationServiceImpl.java
+++ b/src/main/java/project/contract/job/ContractOrderCalculationServiceImpl.java
@@ -10,7 +10,10 @@
 
 import kernel.util.Arith;
 import kernel.util.ThreadUtils;
+import org.springframework.beans.BeansException;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.ApplicationContext;
+import org.springframework.context.ApplicationContextAware;
 import org.springframework.util.ObjectUtils;
 import project.contract.ContractLock;
 import project.contract.ContractOrder;
@@ -26,12 +29,14 @@
 import project.wallet.WalletService;
 import project.wallet.consumer.WalletMessage;
 
-public class ContractOrderCalculationServiceImpl implements ContractOrderCalculationService {
+public class ContractOrderCalculationServiceImpl implements ContractOrderCalculationService, ApplicationContextAware {
+	private ApplicationContext applicationContext;
+
 	private static final Logger logger = LoggerFactory.getLogger(ContractOrderCalculationServiceImpl.class);
 	private ContractOrderService contractOrderService;
 	private DataService dataService;
 	private WalletService walletService;
-	private RedisHandler redisHandler;
+//	private RedisHandler redisHandler;
 	private AssetService assetService;
 	public final static String STATE_SUBMITTED = "submitted";
 	public final static String STATE_CREATED = "created";
@@ -106,40 +111,58 @@
 		double amount = Arith.mul(div, Arith.sub(currentPrice, order.getTrade_avg_price()));
 		logger.info("---------盈亏金额:"+amount);
 		logger.info("---------盈亏:"+profit_loss);
+		RedisHandler redisHandler = getRedisHandler();
 		if ("profit".equals(profit_loss)) {
 			/**
 			 * 盈 正数
 			 */
 			order.setProfit(Arith.add(0.0D, Math.abs(amount)));
+
+			Object profit = redisHandler.get("MONEY_CONTRACT_PROFIT_" + order.getPartyId().toString());
+			if(!ObjectUtils.isEmpty(profit)){
+				if(Double.parseDouble(profit.toString()) < 0 ){
+
+					redisHandler.setSync("MONEY_CONTRACT_PROFIT_"+order.getPartyId().toString(), 0);
+
+					Object money = redisHandler.get("PARTY_ID_MONEY_" + order.getPartyId().toString());
+					if(!ObjectUtils.isEmpty(money)){
+						Wallet wallet = this.walletService.saveWalletByPartyId(order.getPartyId().toString());
+						wallet.setMoney(Double.parseDouble(money.toString()));
+						redisHandler.setSync(WalletRedisKeys.WALLET_PARTY_ID + wallet.getPartyId().toString(), wallet);
+						redisHandler.pushAsyn(WalletRedisKeys.WALLET_QUEUE_UPDATE, new WalletMessage(order.getPartyId().toString(), amount));
+					}
+				}
+			}
 		} else if ("loss".equals(profit_loss)) {
 			order.setProfit(Arith.sub(0.0D,Math.abs(amount) ));
 
 			//定义一个 总浮动亏损的值
-			logger.info("----------fffffffffffffff-------"+redisHandler);
-			logger.info("----------ssssssssssss-------"+order.getPartyId().toString());
-			Double contractAssetsProfit = (Double) this.redisHandler.get(ContractRedisKeys.CONTRACT_ASSETS_PROFIT_PARTY_ID + order.getPartyId().toString());
-			logger.info("----------bbbbbbbbbbbbbbbbbb-------"+contractAssetsProfit);
+			Double contractAssetsProfit = (Double) redisHandler.get(ContractRedisKeys.CONTRACT_ASSETS_PROFIT_PARTY_ID + order.getPartyId().toString());
 			double contractProfit = null == contractAssetsProfit ? 0.000D : contractAssetsProfit;
-			logger.info("----------contractProfit-------"+contractProfit);
-			if(contractProfit <= 0){
-				redisHandler.setSync("MONEY_CONTRACT_PROFIT_"+order.getPartyId().toString(), contractProfit);
+			List<ContractOrder> list = contractOrderService.findSubmitted(order.getPartyId().toString(), null, null);
+			double deposit = 0;
+			for (int i = 0; i < list.size(); i++) {
+				ContractOrder close_line = list.get(i);
+				deposit = Arith.add(deposit, close_line.getDeposit());
 			}
-			logger.info("----------1111111111111-------");
+			double profitt = Arith.add(contractProfit,deposit);
+			if(profitt <= 0){
+				redisHandler.setSync("MONEY_CONTRACT_PROFIT_"+order.getPartyId().toString(), profitt);
+			}else{
+				redisHandler.setSync("MONEY_CONTRACT_PROFIT_"+order.getPartyId().toString(), 0);
+			}
 			//修改余额,每次修改取平仓前余额减去浮动亏损
 			Wallet wallet = this.walletService.saveWalletByPartyId(order.getPartyId().toString());
-			logger.info("----------wallet-------"+wallet);
 			Object money = redisHandler.get("PARTY_ID_MONEY_" + order.getPartyId().toString());
 			Object profit = redisHandler.get("MONEY_CONTRACT_PROFIT_" + order.getPartyId().toString());
-			logger.info("----------money-------"+money);
-			logger.info("----------profit-------"+profit);
 			if(!ObjectUtils.isEmpty(money) && !ObjectUtils.isEmpty(profit)){
-				wallet.setMoney(Arith.sub(Double.parseDouble(money.toString()),Double.parseDouble(profit.toString())));
+				double residueMoney = Arith.add(Double.parseDouble(money.toString()), Double.parseDouble(profit.toString()));
+				//如果当前余额减去浮动亏损小于0,将进入强平,这里不再进行余额更新
+				if(residueMoney > 0){
+					wallet.setMoney(residueMoney);
+					redisHandler.setSync(WalletRedisKeys.WALLET_PARTY_ID + wallet.getPartyId().toString(), wallet);
+				}
 			}
-			logger.info("----------2222222222222-------");
-			redisHandler.setSync(WalletRedisKeys.WALLET_PARTY_ID + wallet.getPartyId().toString(), wallet);
-			logger.info("----------3333333333333333------");
-			redisHandler.pushAsyn(WalletRedisKeys.WALLET_QUEUE_UPDATE, new WalletMessage(order.getPartyId().toString(), amount));
-			logger.info("----------4444444444444-------");
 		}
 
 		double changeRatio;
@@ -212,8 +235,15 @@
 				profit = Arith.add(profit, Arith.add(close_line.getProfit(), close_line.getDeposit()));
 			}
 			Wallet wallet = this.walletService.saveWalletByPartyId(order.getPartyId().toString());
-			double totleMoney = wallet.getMoney();
+			Object money = redisHandler.get("PARTY_ID_MONEY_" + order.getPartyId().toString());
+			double totleMoney = 0;
+			if(ObjectUtils.isEmpty(money)){
+				totleMoney = wallet.getMoney();
+			}else{
+				totleMoney = Double.parseDouble(money.toString());
+			}
 			if (Arith.add(profit,totleMoney) <= 0) {
+				redisHandler.setSync("MONEY_CONTRACT_PROFIT_"+order.getPartyId().toString(), 0);
 				/**
 				 * 触发全仓强平
 				 */
@@ -289,13 +319,19 @@
 		this.order_close_line_type = order_close_line_type;
 	}
 
-	public void setRedisHandler(RedisHandler redisHandler) {
-		this.redisHandler = redisHandler;
+	public RedisHandler getRedisHandler() {
+		return applicationContext.getBean(RedisHandler.class);
 	}
+
 
 	public void setAssetService(AssetService assetService) {
 		this.assetService = assetService;
 	}
 
 
+	@Override
+	public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
+		this.applicationContext = applicationContext;
+	}
+
 }

--
Gitblit v1.9.3