feat(strategy): 添加股票数据可视化功能

- 在 StrategyApplication 中实现了一个简单的股票数据可视化功能
- 使用 ECharts 和 Amis 渲染股票数据的蜡烛图和均线
- 新增了 TestMarkdown 类，用于测试 Markdown 渲染功能
- 在 application.yml 中添加了 LiteFlow 相关配置
- 更新了 pom.xml，添加了 LiteFlow、CommonMark 等依赖

leopard-server/bin/start.sh

@@ -1,3 +1,3 @@
 #!/bin/bash
 
-nohup /home/ubuntu/jdk-17.0.16+8/bin/java -jar /home/ubuntu/app/leopard-server-1.0.0.jar --spring.profiles.active=build --spring.web.resources.static-locations=file:/home/ubuntu/app/dist --logging.parent=/home/ubuntu/app > /dev/null 2>&1 &
+nohup /home/ubuntu/jdk-21.0.8+9/bin/java -jar /home/ubuntu/app/leopard-server-1.0.0.jar --spring.profiles.active=build --spring.web.resources.static-locations=file:/home/ubuntu/app/dist --logging.parent=/home/ubuntu/app > /dev/null 2>&1 &

leopard-server/bin/stop.sh

@@ -1,10 +1,17 @@
 #!/bin/bash
 
 original_command="$0"
-application_name="/home/ubuntu/app/leopard-server-1.0.0.jar"
+application_name="leopard-server-1.0.0.jar"
 
 function get_pid() {
-  echo $(ps ef | grep "$application_name" | grep -v grep | awk '{ print $1 }')
+  ID=$(ps -ef | grep "$application_name" | grep -v grep | grep -v $original_command | awk '{ print $2 }')
+  if [[ -z "$ID" ]]; then
+    ID=$(ps aux | grep "$application_name" | grep -v grep | grep -v $original_command | awk '{print $2}')
+    if [[ -z "$ID" ]]; then
+      ID=$(/home/ubuntu/jdk-21.0.8+9/bin/jps -lmvV | grep "$application_name" | awk '{print $1}')
+    fi
+  fi
+  echo $ID
 }
 
 pid=$(get_pid)
@@ -31,3 +38,4 @@ while (true); do
   fi
   sleep 1s
 done
+

leopard-server/src/main/java/com/lanyuanxiaoyao/leopard/server/service/task/UpdateFinanceIndicatorNode.java

@@ -138,7 +138,7 @@ public class UpdateFinanceIndicatorNode extends TaskNodeComponent {
         var income = incomesMap.get(stock.getCode());
         var cashFlow = cashFlowsMap.get(stock.getCode());
         var finaIndicator = finaIndicatorsMap.get(stock.getCode());
-        if (ArrayUtil.<Object>isAllNull(balance, income, cashFlow, finaIndicator)) {
+        if (stock.getListedDate().getYear() > year || ArrayUtil.<Object>isAllNull(balance, income, cashFlow, finaIndicator)) {
           continue;
         }
         var indicator = financeIndicatorsMap.getOrDefault(stock.getCode(), new FinanceIndicator());

leopard-strategy/pom.xml

@@ -20,15 +20,27 @@
       <groupId>org.springframework.boot</groupId>
       <artifactId>spring-boot-starter</artifactId>
     </dependency>
+    <dependency>
+      <groupId>com.yomahub</groupId>
+      <artifactId>liteflow-spring-boot-starter</artifactId>
+    </dependency>
     <dependency>
       <groupId>io.github.ralfkonrad.quantlib_for_maven</groupId>
       <artifactId>quantlib</artifactId>
-      <version>1.39.0</version>
     </dependency>
     <dependency>
       <groupId>org.ta4j</groupId>
       <artifactId>ta4j-core</artifactId>
-      <version>0.17</version>
+    </dependency>
+    <dependency>
+      <groupId>org.commonmark</groupId>
+      <artifactId>commonmark</artifactId>
+      <version>0.26.0</version>
+    </dependency>
+    <dependency>
+      <groupId>org.commonmark</groupId>
+      <artifactId>commonmark-ext-gfm-tables</artifactId>
+      <version>0.26.0</version>
     </dependency>
 
     <dependency>

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

@@ -1,28 +1,46 @@
 package com.lanyuanxiaoyao.leopard.strategy;
 
-import cn.hutool.core.lang.Tuple;
+import cn.hutool.core.map.MapUtil;
-import com.lanyuanxiaoyao.leopard.core.entity.Daily;
+import cn.hutool.core.util.StrUtil;
+import com.fasterxml.jackson.databind.ObjectMapper;
 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 com.lanyuanxiaoyao.leopard.core.repository.StockRepository;
 import jakarta.annotation.Resource;
 import jakarta.transaction.Transactional;
-import java.util.stream.Collectors;
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.time.Duration;
+import java.time.ZoneId;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
 import lombok.extern.slf4j.Slf4j;
-import org.apache.commons.math3.stat.descriptive.DescriptiveStatistics;
+import org.commonmark.Extension;
+import org.commonmark.ext.gfm.tables.TablesExtension;
+import org.commonmark.parser.Parser;
+import org.commonmark.renderer.html.HtmlRenderer;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
 import org.springframework.boot.context.event.ApplicationReadyEvent;
 import org.springframework.context.event.EventListener;
 import org.springframework.data.domain.Sort;
 import org.springframework.data.jpa.repository.config.EnableJpaAuditing;
+import org.ta4j.core.BaseBar;
+import org.ta4j.core.BaseBarSeriesBuilder;
+import org.ta4j.core.indicators.SMAIndicator;
+import org.ta4j.core.indicators.helpers.ClosePriceIndicator;
 
 @Slf4j
 @SpringBootApplication(scanBasePackages = "com.lanyuanxiaoyao.leopard")
 @EnableJpaAuditing
 public class StrategyApplication {
+  private static final List<Extension> extensions = List.of(TablesExtension.create());
+  private static final Parser parser = Parser.builder().extensions(extensions).build();
+  private static final HtmlRenderer render = HtmlRenderer.builder().extensions(extensions).build();
+  private static final ObjectMapper mapper = new ObjectMapper();
   @Resource
   private StockRepository stockRepository;
   @Resource
@@ -32,34 +50,316 @@ public class StrategyApplication {
     SpringApplication.run(StrategyApplication.class, args);
   }
 
+  private static void render(Map<String, Object> data) throws IOException {
+    Files.writeString(
+      Path.of("result.html"),
+      StrUtil.format(
+        // language=HTML
+        """
+          <html lang='zh'>
+          <head>
+            <meta charset='utf-8'/>
+            <meta http-equiv="Content-Type" content="text/html; charset=utf-8"/>
+            <meta name='viewport' content='width=device-width, initial-scale=1.0'/>
+            <title>Strategy</title>
+            <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/amis/6.13.0/antd.min.css"/>
+            <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/amis/6.13.0/helper.min.css"/>
+            <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/amis/6.13.0/iconfont.min.css"/>
+            <style>
+              html, body, #root {
+                position: relative;
+                width: 100%;
+                height: 100%;
+                margin: 0;
+                padding: 0;
+              }
+            </style>
+          </head>
+          <body>
+          <div id='root'/>
+          </body>
+          <script src="https://cdnjs.cloudflare.com/ajax/libs/amis/6.13.0/sdk.min.js"></script>
+          <script type='text/javascript'>
+            (function () {
+              const amis = amisRequire('amis/embed')
+              const amisJson = {
+                type: 'page',
+                title: 'Strategy',
+                data: {},
+                body: [
+                  /*{
+                    type: 'table2',
+                    source: '${items}',
+                    columns: [
+                      {
+                        name: 'date',
+                        label: 'Date',
+                        width: 100,
+                      },
+                      {
+                        name: 'code',
+                        label: 'Code',
+                        width: 100,
+                      },
+                      {
+                        name: 'name',
+                        label: 'Name',
+                        width: 100,
+                      },
+                      {
+                        name: 'sma5',
+                        label: 'SMA-5',
+                        width: 100,
+                      },
+                      {
+                        name: 'sma10',
+                        label: 'SMA-10',
+                        width: 100,
+                      },
+                    ],
+                  },*/
+                  {
+                    type: 'chart',
+                    height: 800,
+                    config: {
+                      backgroundColor: '#fff',
+                      animation: true,
+                      animationDuration: 1000,
+                      tooltip: {
+                        trigger: 'axis',
+                        axisPointer: {
+                          type: 'cross',
+                        },
+                        backgroundColor: 'rgba(0, 0, 0, 0.7)',
+                        borderColor: '#333',
+                        borderWidth: 1,
+                        textStyle: {
+                          color: '#fff',
+                          fontSize: 12,
+                        },
+                        padding: 12,
+                        formatter: function (params) {
+                          const param = params[0]
+                          const open = param.data[1].toFixed(2)
+                          const close = param.data[2].toFixed(2)
+                          const lowest = param.data[3].toFixed(2)
+                          const highest = param.data[4].toFixed(2)
+          
+                          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: {
+                        left: '2%',
+                        right: '2%',
+                        top: '5%',
+                        bottom: '12%',
+                        containLabel: true,
+                      },
+                      xAxis: {
+                        data: '${xList || []}',
+                        axisLine: {
+                          lineStyle: {
+                            color: '#e0e0e0',
+                          },
+                        },
+                        axisLabel: {
+                          color: '#666',
+                          fontWeight: 'bold',
+                        },
+                        splitLine: {
+                          show: false,
+                        },
+                        axisTick: {
+                          show: false,
+                        },
+                      },
+                      yAxis: {
+                        scale: true,
+                        axisLine: {
+                          lineStyle: {
+                            color: '#e0e0e0',
+                          },
+                        },
+                        axisLabel: {
+                          color: '#666',
+                          fontWeight: 'bold',
+                          formatter: function (value) {
+                            return value.toFixed(2)
+                          },
+                        },
+                        splitLine: {
+                          lineStyle: {
+                            type: 'dashed',
+                            color: '#f0f0f0',
+                          },
+                        },
+                        axisTick: {
+                          show: false,
+                        },
+                      },
+                      dataZoom: [
+                        {
+                          type: 'inside',
+                          start: 0,
+                          end: 100,
+                        },
+                        {
+                          show: true,
+                          type: 'slider',
+                          top: '90%',
+                          start: 0,
+                          end: 100,
+                        },
+                      ],
+                      series: [
+                        {
+                          type: 'candlestick',
+                          data: '${yList || []}',
+                          itemStyle: {
+                            color: '#eb5454',
+                            color0: '#4aaa93',
+                            borderColor: '#eb5454',
+                            borderColor0: '#4aaa93',
+                            borderWidth: 1,
+                          },
+                        },
+                        {
+                          type: 'line',
+                          data: '${sma10 || []}',
+                          smooth: true,
+                          lineStyle: {
+                            color: '#dcb38a',
+                          },
+                          symbol: 'none',
+                        },
+                        {
+                          type: 'line',
+                          data: '${sma60 || []}',
+                          smooth: true,
+                          lineStyle: {
+                            color: '#6ce3c6',
+                          },
+                          symbol: 'none',
+                        },
+                        {
+                          type: 'line',
+                          data: '${sma120 || []}',
+                          smooth: true,
+                          lineStyle: {
+                            color: '#6cd5e3',
+                          },
+                          symbol: 'none',
+                        },
+                      ],
+                    },
+                  },
+                ],
+              }
+              amis.embed('#root', amisJson, {}, {theme: 'antd'})
+            })()
+          </script>
+          </html>
+          """,
+        mapper.writeValueAsString(data)
+      )
+    );
+  }
+
   @Transactional(rollbackOn = Throwable.class)
   @EventListener(ApplicationReadyEvent.class)
-  public void test() {
+  public void test() throws IOException {
-    var dailies = dailyRepository.findAll(
+    /*var dailies = dailyRepository.findAll(
       QDaily.daily.tradeDate.year().eq(2025),
       Sort.by(Daily_.TRADE_DATE)
     );
-    // log.info("OpenDate: {}, CloseDate: {}", dailies.getFirst().getTradeDate(), dailies.getLast().getTradeDate());
+    var data = dailies.stream()
-    // var closes = dailies.stream().map(Daily::getHfqClose).toList();
-    // var statistics = new DescriptiveStatistics();
-    // closes.forEach(statistics::addValue);
-    // log.info("Ascending: {}", (closes.getLast() - closes.getFirst()) * 100.0 / closes.getFirst());
-    // log.info("STD: {}", statistics.getStandardDeviation());
-    dailies.stream()
       .collect(Collectors.groupingBy(Daily::getStock))
       .entrySet()
-      .stream()
+      .parallelStream()
       .map(entry -> {
         var stock = entry.getKey();
-        var statistics = new DescriptiveStatistics();
+        var dailyList = entry.getValue()
-        entry.getValue()
           .stream()
-          .map(Daily::getHfqClose)
+          .sorted(Comparator.comparing(Daily::getTradeDate))
-          .forEach(statistics::addValue);
+          .toList();
-        var std = statistics.getStandardDeviation();
+        var statistics = new DescriptiveStatistics();
-        return new Tuple(stock, std);
+        dailyList.stream().map(Daily::getHfqClose).forEach(statistics::addValue);
+        var barSeries = new BaseBarSeriesBuilder().build();
+        dailyList.forEach(daily -> barSeries.addBar(new BaseBar(
+          Duration.ofDays(1),
+          daily.getTradeDate().atStartOfDay().atZone(ZoneId.systemDefault()),
+          daily.getHfqOpen(),
+          daily.getHfqHigh(),
+          daily.getHfqLow(),
+          daily.getHfqClose(),
+          daily.getVolume()
+        )));
+        return Map.<String, String>of(
+          "code", stock.getCode(),
+          "name", stock.getName(),
+          "std", NumberUtil.roundStr(statistics.getStandardDeviation(), 2),
+          "increase", NumberUtil.roundStr((dailyList.getLast().getHfqClose() - dailyList.getFirst().getHfqClose()) * 100.0 / dailyList.getFirst().getHfqClose(), 2)
+        );
       })
-      .sorted((t1, t2) -> Double.compare(t2.get(1), t1.get(1)))
+      .sorted((t1, t2) -> Double.compare(Double.parseDouble(t2.get("increase")), Double.parseDouble(t1.get("increase"))))
-      .forEachOrdered(tuple -> log.info("Stock: {}, STD: {}", ((Stock) tuple.get(0)).getCode(), tuple.get(1)));
+      .toList();
+    render(Map.of("items", data));*/
+    var dailies = dailyRepository.findAll(
+      QDaily.daily.stock.code.eq("000002.SZ")
+        .and(QDaily.daily.tradeDate.year().goe(2023)),
+      Sort.by(Daily_.TRADE_DATE)
+    );
+    var barSeries = new BaseBarSeriesBuilder().build();
+    dailies.forEach(daily -> barSeries.addBar(new BaseBar(
+      Duration.ofDays(1),
+      daily.getTradeDate().atStartOfDay().atZone(ZoneId.systemDefault()),
+      daily.getHfqOpen(),
+      daily.getHfqHigh(),
+      daily.getHfqLow(),
+      daily.getHfqClose(),
+      daily.getVolume()
+    )));
+    var sma10 = new SMAIndicator(new ClosePriceIndicator(barSeries), 10);
+    var sma60 = new SMAIndicator(new ClosePriceIndicator(barSeries), 60);
+    var sma120 = new SMAIndicator(new ClosePriceIndicator(barSeries), 120);
+
+    var xList = new ArrayList<String>();
+    var yList = new ArrayList<List<Double>>();
+    var sma10List = new ArrayList<Double>();
+    var sma60List = new ArrayList<Double>();
+    var sma120List = new ArrayList<Double>();
+    for (int index = 0; index < dailies.size(); index++) {
+      var daily = dailies.get(index);
+      xList.add(daily.getTradeDate().toString());
+      yList.add(List.of(daily.getHfqOpen(), daily.getHfqClose(), daily.getHfqLow(), daily.getHfqHigh()));
+      sma10List.add(sma10.getValue(index).doubleValue());
+      sma60List.add(sma60.getValue(index).doubleValue());
+      sma120List.add(sma120.getValue(index).doubleValue());
+    }
+    render(
+      MapUtil.<String, Object>builder()
+        .put("xList", xList)
+        .put("yList", yList)
+        .put("sma10", sma10List)
+        .put("sma60", sma60List)
+        .put("sma120", sma120List)
+        .build()
+    );
   }
 }

leopard-strategy/src/main/resources/application.yml

@@ -2,7 +2,7 @@ spring:
   application:
     name: leopard-strategy
   datasource:
-    url: jdbc:postgresql://81.71.3.24:6785/leopard_dev
+    url: jdbc:postgresql://81.71.3.24:6785/leopard
     username: leopard
     password: '9NEzFzovnddf@PyEP?e*AYAWnCyd7UhYwQK$pJf>7?ccFiN^x4$eKEZ5~E<7<+~X'
     driver-class-name: org.postgresql.Driver
@@ -12,3 +12,6 @@ spring:
     banner-mode: off
 fenix:
   print-banner: false
+liteflow:
+  print-banner: false
+  check-node-exists: false

leopard-strategy/src/test/java/com/lanyuanxiaoyao/leopard/strategy/TestMarkdown.java

@@ -0,0 +1,44 @@
+package com.lanyuanxiaoyao.leopard.strategy;
+
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.util.List;
+import org.commonmark.ext.gfm.tables.TablesExtension;
+import org.commonmark.parser.Parser;
+import org.commonmark.renderer.html.HtmlRenderer;
+
+/**
+ * Markdown Render
+ *
+ * @author lanyuanxiaoyao
+ * @version 20250918
+ */
+public class TestMarkdown {
+  public static void main(String[] args) throws IOException {
+    var extensions = List.of(TablesExtension.create());
+    var parser = Parser.builder()
+      .extensions(extensions)
+      .build();
+    var render = HtmlRenderer.builder()
+      .extensions(extensions)
+      .build();
+    var result = render.render(parser.parse(
+      // language=Markdown
+      """
+        ### Hello
+        
+        ```echarts
+        System.out.println("go");
+        ```
+        
+        > I am iron man
+        
+        |      |    |
+        |------|----|
+        | Tony | 12 |
+        """
+    ));
+    Files.writeString(Path.of("result.html"), result);
+  }
+}