lanyuanxiaoyao/leopard
1
0
Fork 0
Code Activity

feat: 增加额外得分数据保存

Browse Source
This commit is contained in:
兰缘小妖
2025-10-11 15:04:16 +08:00
parent 5390a879e7
commit 49a03adf21
6 changed files with 228 additions and 182 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
leopard-core/src/main/java/com/lanyuanxiaoyao/leopard/core/entity/StockScore.java
View File

@@ -2,10 +2,13 @@ package com.lanyuanxiaoyao.leopard.core.entity;
import com.lanyuanxiaoyao.leopard.core.Constants;
import com.lanyuanxiaoyao.service.template.entity.SimpleEntity;
import jakarta.persistence.ElementCollection;
import jakarta.persistence.Entity;
import jakarta.persistence.EntityListeners;
import jakarta.persistence.JoinTable;
import jakarta.persistence.ManyToOne;
import jakarta.persistence.Table;
import java.util.Map;
import lombok.Getter;
import lombok.Setter;
import lombok.ToString;
@@ -28,5 +31,8 @@ public class StockScore extends SimpleEntity {
private Stock stock;
@ManyToOne
private StockCollection collection;
@ElementCollection
@JoinTable(name = Constants.DATABASE_PREFIX + "stock_score_extra")
private Map<String, String> extra;
private Double score;
}

65
leopard-core/src/main/java/com/lanyuanxiaoyao/leopard/core/service/selector/PyramidStockSelector.java
View File

@@ -4,9 +4,11 @@ import cn.hutool.core.util.ArrayUtil;
import cn.hutool.core.util.ObjectUtil;
import com.lanyuanxiaoyao.leopard.core.entity.FinanceIndicator;
import com.lanyuanxiaoyao.leopard.core.entity.QStock;
import com.lanyuanxiaoyao.leopard.core.entity.Stock;
import com.lanyuanxiaoyao.leopard.core.helper.NumberHelper;
import com.lanyuanxiaoyao.leopard.core.repository.StockRepository;
import java.time.LocalDate;
import java.util.Comparator;
import java.util.HashMap;
import java.util.Set;
import java.util.stream.Collectors;
import lombok.extern.slf4j.Slf4j;
@@ -34,9 +36,12 @@ public class PyramidStockSelector implements StockSelector<PyramidStockSelector.
public Set<Candidate> select(Request request) {
// 选择至少有最近5年财报的股票
// 有点奇怪，001400.SZ有近5年的财报但资料显示是2025年才上市的
var stocks = stockRepository.findAll(QStock.stock.listedDate.before(LocalDate.of(request.year(), 1, 1)));
var scores = stocks.stream().collect(Collectors.toMap(stock -> stock, code -> 0));
for (Stock stock : stocks) {
return stockRepository.findAll(QStock.stock.listedDate.before(LocalDate.of(request.year(), 1, 1)))
.stream()
.map(stock -> {
var extra = new HashMap<String, String>();
var score = 0;
var recentIndicators = stock.getIndicators()
.stream()
.filter(indicator -> indicator.getYear() < request.year())
@@ -44,7 +49,7 @@ public class PyramidStockSelector implements StockSelector<PyramidStockSelector.
.limit(5)
.toList();
if (recentIndicators.size() < 5) {
continue;
return null;
}
var latestIndicator = recentIndicators.getFirst();
@@ -69,8 +74,10 @@ public class PyramidStockSelector implements StockSelector<PyramidStockSelector.
} else if (averageRoe >= 10) {
roeScore = 300;
}
extra.put("平均ROE", NumberHelper.formatPriceDouble(averageRoe));
}
scores.put(stock, scores.get(stock) + roeScore);
extra.put("平均ROE得分", NumberHelper.formatPriceDouble(roeScore));
score += roeScore;
var roaScore = 0;
if (recentIndicators.stream().noneMatch(indicator -> indicator.getReturnOnAssets() == null)) {
@@ -86,8 +93,10 @@ public class PyramidStockSelector implements StockSelector<PyramidStockSelector.
} else if (averageRoa >= 7) {
roaScore = 50;
}
extra.put("平均ROA", NumberHelper.formatPriceDouble(averageRoa));
}
scores.put(stock, scores.get(stock) + roaScore);
extra.put("平均ROA得分", NumberHelper.formatPriceDouble(roaScore));
score += roaScore;
var netProfitScore = 0;
if (recentIndicators.stream().noneMatch(indicator -> indicator.getNetProfit() == null)) {
@@ -101,8 +110,10 @@ public class PyramidStockSelector implements StockSelector<PyramidStockSelector.
} else if (averageNetProfit >= 1000.0 * 10000000) {
netProfitScore = 100;
}
extra.put("平均净利润", NumberHelper.formatPriceDouble(averageNetProfit));
}
scores.put(stock, scores.get(stock) + netProfitScore);
extra.put("平均净利润得分", NumberHelper.formatPriceDouble(netProfitScore));
score += netProfitScore;
var cashScore = 0;
if (
@@ -114,19 +125,22 @@ public class PyramidStockSelector implements StockSelector<PyramidStockSelector.
) {
cashScore = 50;
}
scores.put(stock, scores.get(stock) + cashScore);
extra.put("现金流得分", NumberHelper.formatPriceDouble(cashScore));
score += cashScore;
if (ObjectUtil.isNotNull(latestIndicator.getDaysAccountsReceivableTurnover()) && latestIndicator.getDaysAccountsReceivableTurnover() <= 30) {
scores.put(stock, scores.get(stock) + 20);
extra.put("应收账款周转天数得分", "20");
score += 20;
}
if (ObjectUtil.isNotNull(latestIndicator.getDaysInventoryTurnover()) && latestIndicator.getDaysInventoryTurnover() <= 30) {
scores.put(stock, scores.get(stock) + 20);
extra.put("存货周转天数得分", "20");
score += 20;
}
if (ArrayUtil.isAllNotNull(latestIndicator.getDaysAccountsReceivableTurnover(), latestIndicator.getDaysInventoryTurnover())) {
if (latestIndicator.getDaysAccountsReceivableTurnover() + latestIndicator.getDaysInventoryTurnover() <= 40) {
scores.put(stock, scores.get(stock) + 20);
score += 20;
} else if (latestIndicator.getDaysAccountsReceivableTurnover() + latestIndicator.getDaysInventoryTurnover() <= 60) {
scores.put(stock, scores.get(stock) + 10);
score += 10;
}
}
@@ -137,7 +151,8 @@ public class PyramidStockSelector implements StockSelector<PyramidStockSelector.
.mapToDouble(Double::doubleValue)
.forEach(stat::addValue);
if (stat.getStandardDeviation() <= 0.3) {
scores.put(stock, scores.get(stock) + 50);
extra.put("毛利率标准差得分", "50");
score += 50;
}
}
@@ -150,8 +165,10 @@ public class PyramidStockSelector implements StockSelector<PyramidStockSelector.
} else if (latestIndicator.getOperatingSafetyMarginRatio() >= 30) {
operatingSafeMarginScore = 10;
}
extra.put("安全边际比率", NumberHelper.formatPriceDouble(latestIndicator.getOperatingSafetyMarginRatio()));
}
scores.put(stock, scores.get(stock) + operatingSafeMarginScore);
extra.put("安全边际比率得分", NumberHelper.formatPriceDouble(operatingSafeMarginScore));
score += operatingSafeMarginScore;
var netProfitAscendingScore = 0;
if (recentIndicators.stream().noneMatch(indicator -> indicator.getNetProfit() == null)) {
@@ -178,7 +195,8 @@ public class PyramidStockSelector implements StockSelector<PyramidStockSelector.
netProfitAscendingScore -= 15;
}
}
scores.put(stock, scores.get(stock) + netProfitAscendingScore);
extra.put("近五年净利润得分", NumberHelper.formatPriceDouble(netProfitAscendingScore));
score += netProfitAscendingScore;
var cashAscendingScore = 0;
if (recentIndicators.stream().noneMatch(indicator -> indicator.getCashAndCashEquivalents() == null)) {
@@ -206,13 +224,14 @@ public class PyramidStockSelector implements StockSelector<PyramidStockSelector.
cashAscendingScore -= 15;
}
}
scores.put(stock, scores.get(stock) + cashAscendingScore);
}
return scores.entrySet()
.stream()
.sorted((e1, e2) -> e2.getValue() - e1.getValue())
.limit(request.limit())
.map(entry -> new Candidate(entry.getKey(), entry.getValue()))
extra.put("近五年现金流得分", NumberHelper.formatPriceDouble(cashAscendingScore));
score += cashAscendingScore;
return new Candidate(stock, score, extra);
})
.filter(ObjectUtil::isNotNull)
.sorted(Comparator.comparingDouble(Candidate::score).reversed())
.limit(request.limit)
.collect(Collectors.toSet());
}

2
leopard-core/src/main/java/com/lanyuanxiaoyao/leopard/core/service/selector/StockSelector.java
View File

@@ -13,7 +13,7 @@ import java.util.Set;
public interface StockSelector<T> {
Set<Candidate> select(T request);
record Candidate(Stock stock, double score, Map<String, Object> extra) {
record Candidate(Stock stock, double score, Map<String, String> extra) {
public Candidate(Stock stock, double score) {
this(stock, score, Map.of());
}

1
leopard-core/src/main/java/com/lanyuanxiaoyao/leopard/core/task/PyramidSelect.java
View File

@@ -45,6 +45,7 @@ public class PyramidSelect extends TaskRunner {
var score = new StockScore();
score.setStock(candidate.stock());
score.setScore(candidate.score());
score.setExtra(candidate.extra());
score.setCollection(collection);
return score;
})

13
leopard-server/src/main/java/com/lanyuanxiaoyao/leopard/server/entity/StockScoreVo.java
View File

@@ -3,6 +3,8 @@ package com.lanyuanxiaoyao.leopard.server.entity;
import com.lanyuanxiaoyao.leopard.core.entity.Stock;
import com.lanyuanxiaoyao.leopard.core.entity.StockScore;
import java.time.LocalDate;
import java.util.Map;
import java.util.stream.Collectors;
/**
* @author lanyuanxiaoyao
@@ -16,7 +18,8 @@ public record StockScoreVo(
Stock.Market market,
String industry,
LocalDate listedDate,
Double score
Double score,
String extra
) {
public static StockScoreVo of(StockScore score) {
return new StockScoreVo(
@@ -27,7 +30,13 @@ public record StockScoreVo(
score.getStock().getMarket(),
score.getStock().getIndustry(),
score.getStock().getListedDate(),
score.getScore()
score.getScore(),
score.getExtra()
.entrySet()
.stream()
.sorted(Map.Entry.comparingByKey())
.map(entry -> entry.getKey() + ": " + entry.getValue())
.collect(Collectors.joining("<br>"))
);
}
}

11
leopard-web/src/pages/stock/StockCollectionDetail.tsx
View File

@@ -21,10 +21,21 @@ function StockCollectionDetail() {
undefined,
[
{
className: 'white-space-pre',
name: 'score',
label: '得分',
width: 50,
align: 'center',
type: 'tpl',
tpl: '${score}',
popOver: {
trigger: 'click',
showIcon: false,
body: {
type: 'tpl',
tpl: '${extra|raw}',
},
},
},
],
),