1
0

Compare commits

...

4 Commits

Author SHA1 Message Date
d2b3305ca6 fix: 修复价格显示错误 2025-09-16 18:40:44 +08:00
10a0e14024 fix: 修复指标显示错位 2025-09-16 18:21:49 +08:00
a9621a10ac perf: 优化查询效率 2025-09-16 18:21:33 +08:00
b5688bd3ab feat: 查询优化 2025-09-16 18:20:59 +08:00
7 changed files with 97 additions and 78 deletions

View File

@@ -2,8 +2,12 @@ package com.lanyuanxiaoyao.leopard.core.repository;
import com.lanyuanxiaoyao.leopard.core.entity.Daily;
import com.lanyuanxiaoyao.service.template.repository.SimpleRepository;
import com.querydsl.core.types.Predicate;
import java.time.LocalDate;
import java.util.List;
import java.util.Optional;
import org.springframework.data.domain.Sort;
import org.springframework.data.jpa.repository.EntityGraph;
import org.springframework.data.jpa.repository.Query;
import org.springframework.stereotype.Repository;
@@ -14,4 +18,12 @@ public interface DailyRepository extends SimpleRepository<Daily> {
@Query("select distinct daily.tradeDate from Daily daily where daily.stock.id = ?1")
List<LocalDate> findDistinctTradeDateByStockId(Long stockId);
@EntityGraph(attributePaths = {"stock"})
@Override
Optional<Daily> findOne(Predicate predicate);
@EntityGraph(attributePaths = {"stock"})
@Override
List<Daily> findAll(Predicate predicate, Sort sort);
}

View File

@@ -2,8 +2,20 @@ package com.lanyuanxiaoyao.leopard.core.repository;
import com.lanyuanxiaoyao.leopard.core.entity.FinanceIndicator;
import com.lanyuanxiaoyao.service.template.repository.SimpleRepository;
import com.querydsl.core.types.Predicate;
import java.util.List;
import java.util.Optional;
import org.springframework.data.domain.Sort;
import org.springframework.data.jpa.repository.EntityGraph;
import org.springframework.stereotype.Repository;
@Repository
public interface FinanceIndicatorRepository extends SimpleRepository<FinanceIndicator> {
@EntityGraph(attributePaths = {"stock"})
@Override
Optional<FinanceIndicator> findOne(Predicate predicate);
@EntityGraph(attributePaths = {"stock"})
@Override
List<FinanceIndicator> findAll(Predicate predicate, Sort sort);
}

View File

@@ -2,7 +2,10 @@ package com.lanyuanxiaoyao.leopard.core.repository;
import com.lanyuanxiaoyao.leopard.core.entity.Stock;
import com.lanyuanxiaoyao.service.template.repository.SimpleRepository;
import jakarta.transaction.Transactional;
import java.util.Collection;
import java.util.List;
import org.springframework.data.jpa.repository.Modifying;
import org.springframework.data.jpa.repository.Query;
import org.springframework.stereotype.Repository;
@@ -14,4 +17,11 @@ import org.springframework.stereotype.Repository;
public interface StockRepository extends SimpleRepository<Stock> {
@Query("select distinct stock.industry from Stock stock where stock.industry is not null")
List<String> findDistinctIndustries();
@Query("select distinct stock.code from Stock stock")
List<String> findDistinctCodes();
@Modifying
@Transactional(rollbackOn = Throwable.class)
void deleteAllByCodeIn(Collection<String> code);
}

View File

@@ -8,21 +8,20 @@ import com.lanyuanxiaoyao.leopard.core.entity.Stock;
import com.lanyuanxiaoyao.leopard.core.repository.DailyRepository;
import com.lanyuanxiaoyao.leopard.core.repository.StockRepository;
import com.lanyuanxiaoyao.leopard.server.helper.NumberHelper;
import com.lanyuanxiaoyao.leopard.server.service.TaskService;
import com.lanyuanxiaoyao.leopard.server.service.TuShareService;
import com.yomahub.liteflow.annotation.LiteflowComponent;
import com.yomahub.liteflow.core.NodeComponent;
import java.time.LocalDate;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.stream.Collectors;
import lombok.extern.slf4j.Slf4j;
import org.springframework.transaction.support.TransactionTemplate;
@Slf4j
@LiteflowComponent("update_daily")
public class UpdateDailyNode extends TaskNodeComponent {
public class UpdateDailyNode extends NodeComponent {
private final StockRepository stockRepository;
private final DailyRepository dailyRepository;
@@ -30,8 +29,7 @@ public class UpdateDailyNode extends TaskNodeComponent {
private final TransactionTemplate transactionTemplate;
public UpdateDailyNode(TaskService taskService, StockRepository stockRepository, DailyRepository dailyRepository, TuShareService tuShareService, TransactionTemplate transactionTemplate) {
super(taskService);
public UpdateDailyNode(StockRepository stockRepository, DailyRepository dailyRepository, TuShareService tuShareService, TransactionTemplate transactionTemplate) {
this.stockRepository = stockRepository;
this.dailyRepository = dailyRepository;
this.tuShareService = tuShareService;
@@ -51,15 +49,11 @@ public class UpdateDailyNode extends TaskNodeComponent {
}
var existsTradeDates = dailyRepository.findDistinctTradeDate();
var nowDate = LocalDate.now();
var stocks = stockRepository.findAll();
var stocksMap = stocks.stream().collect(Collectors.toMap(Stock::getCode, stock -> stock));
var allTradeDates = tradeDates.stream()
var stocksMap = stockRepository.findAll().stream().collect(Collectors.toMap(Stock::getCode, stock -> stock));
tradeDates.parallelStream()
.filter(date -> date.isBefore(nowDate) || date.isEqual(nowDate))
.filter(date -> !existsTradeDates.contains(date))
.sorted()
.toList();
var total = new AtomicInteger(allTradeDates.size());
allTradeDates.parallelStream()
.filter(date -> date.isAfter(LocalDate.of(2024, 12, 31)))
.forEach(tradeDate -> {
var factorResponse = tuShareService.factorList(tradeDate);
var factorMap = new HashMap<String, Double>();

View File

@@ -126,6 +126,13 @@ public class UpdateFinanceIndicatorNode extends TaskNodeComponent {
(existing, replacement) -> existing
));
var financeIndicatorsMap = financeIndicatorRepository.findAll(QFinanceIndicator.financeIndicator.year.eq(year))
.stream()
.collect(Collectors.toMap(
indicator -> indicator.getStock().getCode(),
indicator -> indicator
));
for (Stock stock : stocks) {
var balance = balancesMap.get(stock.getCode());
var income = incomesMap.get(stock.getCode());
@@ -134,10 +141,7 @@ public class UpdateFinanceIndicatorNode extends TaskNodeComponent {
if (ArrayUtil.<Object>isAllNull(balance, income, cashFlow, finaIndicator)) {
continue;
}
var indicator = financeIndicatorRepository.findOne(
QFinanceIndicator.financeIndicator.stock.id.eq(stock.getId())
.and(QFinanceIndicator.financeIndicator.year.eq(year))
).orElse(new FinanceIndicator());
var indicator = financeIndicatorsMap.getOrDefault(stock.getCode(), new FinanceIndicator());
indicator.setStock(stock);
indicator.setYear(year);
if (ObjectUtil.isNotNull(balance)) {

View File

@@ -8,7 +8,6 @@ import com.yomahub.liteflow.annotation.LiteflowComponent;
import com.yomahub.liteflow.core.NodeComponent;
import jakarta.transaction.Transactional;
import java.time.LocalDate;
import java.util.HashSet;
import java.util.stream.Collectors;
@LiteflowComponent("update_stock")
@@ -25,44 +24,32 @@ public class UpdateStockNode extends NodeComponent {
@Transactional(rollbackOn = Throwable.class)
@Override
public void process() {
var stocks = stockRepository.findAll();
var stocksMap = stocks.stream().collect(Collectors.toMap(Stock::getCode, stock -> stock));
var targetCodes = new HashSet<String>();
tuShareService.stockList()
var existsStockMap = stockRepository.findAll().stream().collect(Collectors.toMap(Stock::getCode, stock -> stock));
var stocks = tuShareService.stockList()
.data()
.items()
.forEach(item -> {
.stream()
.map(item -> {
var code = item.get(0);
var name = item.get(1);
var fullname = item.get(2);
var market = EnumUtil.fromString(Stock.Market.class, item.get(3));
var industry = item.get(4);
var listedDate = LocalDate.parse(item.get(5), TuShareService.TRADE_FORMAT);
if (stocksMap.containsKey(code)) {
var stock = stocksMap.get(code);
stock.setName(name);
stock.setFullname(fullname);
stock.setMarket(market);
stock.setIndustry(industry);
stock.setListedDate(listedDate);
} else {
var stock = new Stock();
stock.setCode(code);
stock.setName(name);
stock.setFullname(fullname);
stock.setMarket(market);
stock.setIndustry(industry);
stock.setListedDate(listedDate);
stocks.add(stock);
}
targetCodes.add(code);
});
var deleteStocks = stocks.stream()
.filter(stock -> !targetCodes.contains(stock.getCode()))
.map(Stock::getId)
var stock = existsStockMap.getOrDefault(code, new Stock());
stock.setCode(code);
stock.setName(name);
stock.setFullname(fullname);
stock.setMarket(market);
stock.setIndustry(industry);
stock.setListedDate(listedDate);
return stock;
})
.toList();
stockRepository.deleteByIds(deleteStocks);
var currentCodes = stocks.stream().map(Stock::getCode).toList();
var existsCodes = stockRepository.findDistinctCodes();
var deleteCodes = existsCodes.stream().filter(code -> !currentCodes.contains(code)).toList();
stockRepository.deleteAllByCodeIn(deleteCodes);
stockRepository.saveAll(stocks);
}
}

View File

@@ -3,6 +3,7 @@ import {useParams} from 'react-router'
import {amisRender, commonInfo, readOnlyDialogOptions, remoteMappings} from '../../util/amis.tsx'
import type {Schema} from 'amis'
import {isNil} from 'es-toolkit'
import {toNumber} from 'es-toolkit/compat'
const formatFinanceNumber = (value: number): string => {
if (isNil(value)) {
@@ -280,8 +281,8 @@ function StockDetail() {
content: '${balanceSheet.currentAssets}',
},
{
label: financePropertyLabel(id, '流动资产占比', 'PERCENTAGE', 'fixedAssetsToTotalAssetsRatio'),
content: '${balanceSheet.fixedAssetsRatio}',
label: financePropertyLabel(id, '流动资产占比', 'PERCENTAGE', 'currentAssetsToTotalAssetsRatio'),
content: '${balanceSheet.currentAssetsRatio}',
},
{
label: financePropertyLabel(id, '流动负债', 'FINANCE', 'currentLiabilities'),
@@ -291,14 +292,14 @@ function StockDetail() {
label: financePropertyLabel(id, '流动负债占比', 'PERCENTAGE', 'currentLiabilitiesToTotalAssetsRatio'),
content: '${balanceSheet.currentLiabilitiesRatio}',
},
{
label: financePropertyLabel(id, '流动资产占比', 'PERCENTAGE', 'currentAssetsToTotalAssetsRatio'),
content: '${balanceSheet.currentAssetsRatio}',
},
{
label: financePropertyLabel(id, '非流动资产', 'FINANCE', 'fixedAssets'),
content: '${balanceSheet.fixedAssets}',
},
{
label: financePropertyLabel(id, '非流动资产占比', 'PERCENTAGE', 'fixedAssetsToTotalAssetsRatio'),
content: '${balanceSheet.fixedAssetsRatio}',
},
{
label: financePropertyLabel(id, '非流动负债', 'FINANCE', 'longTermLiabilities'),
content: '${balanceSheet.longTermLiabilities}',
@@ -409,10 +410,10 @@ function StockDetail() {
],
},
{type: 'divider'},
"100日线数据",
"100日线数据",
{
type: 'chart',
height: 800,
height: 500,
api: `get:${commonInfo.baseUrl}/stock/daily/${id}`,
config: {
backgroundColor: '#fff',
@@ -433,30 +434,28 @@ function StockDetail() {
padding: 12,
formatter: function (params: any) {
const param = params[0]
const open = param.data[0]
const close = param.data[1]
const lowest = param.data[2]
const highest = param.data[3]
const open = toNumber(param.data[1]).toFixed(2)
const close = toNumber(param.data[2]).toFixed(2)
const lowest = toNumber(param.data[3]).toFixed(2)
const highest = toNumber(param.data[4]).toFixed(2)
return [
`<div style="font-weight: bold; margin-bottom: 4px;">${param.name}</div>`,
`<div style="display: flex; justify-content: space-between; margin: 2px 0;">`,
`<span>开盘:</span>`,
`<span style="margin-left: 12px; font-weight: bold;">${open}</span>`,
`</div>`,
`<div style="display: flex; justify-content: space-between; margin: 2px 0;">`,
`<span>收盘:</span>`,
`<span style="margin-left: 12px; font-weight: bold;">${close}</span>`,
`</div>`,
`<div style="display: flex; justify-content: space-between; margin: 2px 0;">`,
`<span>最低:</span>`,
`<span style="margin-left: 12px; font-weight: bold;">${lowest}</span>`,
`</div>`,
`<div style="display: flex; justify-content: space-between; margin: 2px 0;">`,
`<span>最高:</span>`,
`<span style="margin-left: 12px; font-weight: bold;">${highest}</span>`,
`</div>`,
].join('')
return `<div class="text-center font-bold mb-2">${param.name}</div>
<div class="text-center">
<span>开盘:</span>
<span class="font-bold ml-4">${open}</span>
</div>
<div class="text-center">
<span>收盘:</span>
<span class="font-bold ml-4">${close}</span>
</div>
<div class="text-center">
<span>最低:</span>
<span class="font-bold ml-4">${lowest}</span>
</div>
<div class="text-center">
<span>最高:</span>
<span class="font-bold ml-4">${highest}</span>
</div>`
},
},
grid: {
@@ -495,7 +494,7 @@ function StockDetail() {
color: '#666',
fontWeight: 'bold',
formatter: function (value: number) {
return value.toFixed(0)
return value.toFixed(2)
},
},
splitLine: {
@@ -538,6 +537,7 @@ function StockDetail() {
],
},
},
"12月线数据",
],
},
],