package com.yami.trading.admin.task.contract;
|
|
import cn.hutool.core.collection.CollectionUtil;
|
import com.yami.trading.bean.contract.domain.ContractOrder;
|
import com.yami.trading.common.config.ThreadPoolComponent;
|
import com.yami.trading.common.util.ThreadUtils;
|
import com.yami.trading.service.contract.ContractLock;
|
import com.yami.trading.service.contract.ContractOrderCalculationService;
|
import com.yami.trading.service.contract.ContractOrderService;
|
import org.apache.commons.logging.Log;
|
import org.apache.commons.logging.LogFactory;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Component;
|
|
import java.util.List;
|
import java.util.Map;
|
import java.util.Set;
|
import java.util.stream.Collectors;
|
|
@Component
|
public class ContractOrderCalculationJob implements Runnable {
|
private static Log logger = LogFactory.getLog(ContractOrderCalculationJob.class);
|
@Autowired
|
private ContractOrderService contractOrderService;
|
@Autowired
|
private ContractOrderCalculationService contractOrderCalculationService;
|
@Autowired
|
private ThreadPoolComponent threadPoolComponent;
|
public void run() {
|
|
while (true) {
|
try {
|
List<ContractOrder> list = this.contractOrderService.findSubmitted();
|
Map<String, List<ContractOrder>> partyIdContractOrders = list.stream().collect(Collectors.groupingBy(ContractOrder::getPartyId));
|
|
Set<String> partyIds = partyIdContractOrders.keySet();
|
// 按提交者进行分组，避免多次查询
|
for(String partyId : partyIds){
|
threadPoolComponent.getExecutor().execute(()->{
|
List<ContractOrder> l = partyIdContractOrders.get(partyId);
|
for (int i = 0; i < l.size(); i++) {
|
ContractOrder order = l.get(i);
|
|
boolean lock = false;
|
try {
|
if (!ContractLock.add(order.getOrderNo())) {
|
continue;
|
}
|
lock = true;
|
this.contractOrderCalculationService.saveCalculation(order.getOrderNo(), l);
|
|
} catch (Throwable e) {
|
logger.error("error:", e);
|
} finally {
|
if (lock) {
|
ContractLock.remove(order.getOrderNo());
|
}
|
|
}
|
|
}
|
});
|
}
|
|
|
|
|
|
} catch (Throwable e) {
|
e.printStackTrace();
|
logger.error("run fail", e);
|
} finally {
|
/**
|
* 暂停0.1秒
|
*/
|
ThreadUtils.sleep(3000);
|
}
|
}
|
}
|
|
public void start(){
|
|
new Thread(this, "ContractOrderCalculationJob").start();
|
if (logger.isInfoEnabled()) {
|
logger.info("持仓单盈亏计算线程启动！");
|
}
|
|
}
|
|
public void setContractOrderService(ContractOrderService contractOrderService) {
|
this.contractOrderService = contractOrderService;
|
}
|
|
public void setContractOrderCalculationService(ContractOrderCalculationService contractOrderCalculationService) {
|
this.contractOrderCalculationService = contractOrderCalculationService;
|
}
|
|
}
|
