feat: 简化交易计算

leopard-core/src/main/java/com/lanyuanxiaoyao/leopard/core/strategy/TradeEngine.java

@@ -3,13 +3,11 @@ package com.lanyuanxiaoyao.leopard.core.strategy;
 import cn.hutool.core.util.ObjectUtil;
 import com.lanyuanxiaoyao.leopard.core.entity.Daily;
 import com.lanyuanxiaoyao.leopard.core.entity.QDaily;
+import com.lanyuanxiaoyao.leopard.core.entity.Stock;
 import com.lanyuanxiaoyao.leopard.core.repository.DailyRepository;
 import java.time.LocalDate;
 import java.util.ArrayList;
-import java.util.HashMap;
 import java.util.List;
-import java.util.Map;
-import java.util.stream.Collectors;
 import lombok.Data;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Service;
@@ -29,9 +27,9 @@ public class TradeEngine {
     this.dailyRepository = dailyRepository;
   }
 
-  public Asset backtest(List<Long> stocks, TradeStrategy strategy, LocalDate startDate, LocalDate endDate) {
+  public Asset backtest(Stock stock, TradeStrategy strategy, LocalDate startDate, LocalDate endDate) {
     var dailies = dailyRepository.findAll(
-      QDaily.daily.stock.id.in(stocks)
+      QDaily.daily.stock.eq(stock)
         .and(QDaily.daily.tradeDate.before(endDate)),
       QDaily.daily.tradeDate.asc()
     );
@@ -45,67 +43,59 @@ public class TradeEngine {
         continue;
       }
       final var currentDate = now;
-      var trades = strategy.trade(
+      var trade = strategy.trade(
         now,
         asset,
         dailies.stream()
           .filter(daily -> daily.getTradeDate().isBefore(currentDate))
-          .collect(Collectors.groupingBy(daily -> daily.getStock().getId()))
+          .toList()
       );
-      for (var trade : trades) {
+      if (trade == 0) {
-        dailies.stream()
+        continue;
-          .filter(daily -> ObjectUtil.equals(daily.getStock().getId(), trade.stockId))
+      }
-          .filter(daily -> ObjectUtil.equals(daily.getTradeDate(), currentDate))
+      var daily = dailies.stream()
+        .filter(d -> ObjectUtil.equals(d.getTradeDate(), currentDate))
         .findFirst()
-          .map(Daily::getHfqClose)
+        .orElseThrow();
-          .ifPresent(close -> {
+      asset.addTrade(now, trade, daily.getHfqClose());
-            if (trade.volume < 0) {
-              asset.setCash(asset.getCash() + Math.abs(trade.volume) * close);
-            } else if (trade.volume > 0) {
-              asset.setCash(asset.getCash() - Math.abs(trade.volume) * close);
-            }
-            asset.getStocks().put(
-              trade.stockId,
-              asset.getStocks().getOrDefault(trade.stockId, 0) + trade.volume
-            );
-          });
-      }
-      asset.getHistories().add(new Asset.History(
-        now,
-        asset.getCash(),
-        asset.getStocks(),
-        trades.stream()
-          .collect(Collectors.groupingBy(trade -> trade.stockId))
-      ));
     }
     return asset;
   }
 
   public interface TradeStrategy {
-    List<Trade> trade(LocalDate now, Asset asset, Map<Long, List<Daily>> dailies);
+    int trade(LocalDate now, Asset asset, List<Daily> dailies);
   }
 
   @Data
   public static final class Asset {
-    private double cash = 0;
+    private List<Trade> trades = new ArrayList<>();
-    private double profit = 0.0;
-    private Map<Long, Integer> stocks = new HashMap<>();
-    private List<History> histories = new ArrayList<>();
 
-    public record History(
+    public void addTrade(LocalDate date, int volume, double price) {
-      LocalDate date,
+      trades.add(new Trade(date, volume, price));
-      double cash,
-      Map<Long, Integer> stocks,
-      Map<Long, List<Trade>> trades
-    ) {
     }
+
+    public int getVolume() {
+      return trades.stream()
+        .mapToInt(Trade::volume)
+        .sum();
+    }
+
+    public double getCash() {
+      return trades.stream()
+        .mapToDouble(trade -> -1 * trade.volume() * trade.price())
+        .sum();
+    }
+
+    public double getPrice() {
+      int volume = getVolume();
+      return volume == 0 ? 0 : getCash() / volume;
     }
 
     public record Trade(
       LocalDate date,
-    Long stockId,
+      int volume,
-    // 用正负数表达买卖
+      double price
-    Integer volume
     ) {
     }
+  }
 }

leopard-strategy/src/main/java/com/lanyuanxiaoyao/leopard/strategy/StrategyApplication.java

@@ -1,7 +1,6 @@
 package com.lanyuanxiaoyao.leopard.strategy;
 
 import cn.hutool.core.lang.Dict;
-import cn.hutool.core.util.ObjectUtil;
 import cn.hutool.core.util.StrUtil;
 import cn.hutool.extra.template.TemplateConfig;
 import cn.hutool.extra.template.TemplateEngine;
@@ -71,33 +70,26 @@ public class StrategyApplication {
       .forEach(code -> {
         var stock = stockRepository.findOne(QStock.stock.code.eq(code)).orElseThrow();
         var asset = tradeEngine.backtest(
-          List.of(stock.getId()),
+          stockRepository.findById(stock.getId()).orElseThrow(),
           (now, currentAsset, dailies) -> {
-            return dailies.entrySet()
+            var stockDailies = dailies
-              .stream()
-              .map(entry -> {
-                var stockId = entry.getKey();
-                var stockDailies = entry.getValue()
               .stream()
               .sorted(Comparator.comparing(Daily::getTradeDate))
               .toList();
             var yesterday = stockDailies.getLast();
             if (yesterday.getHfqClose() > yesterday.getHfqOpen()) {
               log.info("{} Buy for price {} {}", now, yesterday.getHfqOpen(), yesterday.getHfqClose());
-                  return new TradeEngine.Trade(now, stockId, 100);
+              return 100;
             } else if (yesterday.getHfqClose() < yesterday.getHfqOpen()) {
-                  var hold = currentAsset.getStocks().getOrDefault(stockId, 0);
+              var hold = currentAsset.getVolume();
               if (hold > 0) {
                 log.info("{} Sell for price {} {}", now, yesterday.getHfqOpen(), yesterday.getHfqClose());
-                    return new TradeEngine.Trade(now, stockId, -1 * hold);
+                return -1 * hold;
               }
             } else {
               log.info("{} Hold for price {} {}", now, yesterday.getHfqOpen(), yesterday.getHfqClose());
             }
-                return null;
+            return 0;
-              })
-              .filter(ObjectUtil::isNotNull)
-              .toList();
           },
           startDate,
           endDate
@@ -121,7 +113,8 @@ public class StrategyApplication {
           dailyYList.add(List.of(daily.getHfqOpen(), daily.getHfqClose(), daily.getHfqLow(), daily.getHfqHigh()));
           dailyCloseMapping.put(daily.getTradeDate().toString(), daily.getHfqClose());
         }
-        charts.add(Dict.create()
+        charts.add(
+          Dict.create()
             .set("title", code)
             .set(
               "data",
@@ -133,19 +126,15 @@ public class StrategyApplication {
                     .set("yList", dailyYList)
                     .set(
                       "points",
-                    asset.getHistories()
+                      asset.getTrades()
                         .stream()
-                      .filter(history -> ObjectUtil.isNotEmpty(history.trades()))
+                        .map(trade -> {
-                      .filter(history -> history.trades().containsKey(stock.getId()))
-                      .filter(history -> ObjectUtil.isNotEmpty(history.trades().get(stock.getId())))
-                      .map(history -> {
-                          var trade = history.trades().get(stock.getId()).getFirst();
                             return Dict.create()
                               .set("value", trade.volume())
                               .set("itemStyle", Dict.create()
                                 .set("color", trade.volume() > 0 ? "#e5b8b5" : "#b5e2e5")
                               )
-                            .set("coord", List.of(history.date().toString(), dailyCloseMapping.getOrDefault(history.date().toString(), 0.0)));
+                              .set("coord", List.of(trade.date().toString(), dailyCloseMapping.getOrDefault(trade.date().toString(), 0.0)));
                           }
                         )
                         .toList()