From d3155bce538bc1af811fe083874e7eac2259efda Mon Sep 17 00:00:00 2001
From: xiaoliuhu <xiaoliuhu@juyou88.com>
Date: Mon, 22 Apr 2024 13:57:39 +0800
Subject: [PATCH] =?UTF-8?q?fix:=E4=BF=AE=E6=94=B9=E8=8E=B7=E5=8F=96?=
 =?UTF-8?q?=E8=82=A1=E7=A5=A8=E8=A1=8C=E6=83=85=E5=A2=9E=E5=8A=A0=E8=BF=94?=
 =?UTF-8?q?=E5=9B=9E=E8=82=A1=E7=A5=A8id=E5=AD=97=E6=AE=B5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../domain/basic/repository/RetifiveStockRepository.java | 9 +++++++--
 src/main/java/cn/stock/market/dto/RetifiveStockInfo.java | 3 +++
 .../cn/stock/market/web/MessageRetifiveController.java   | 5 +++++
 3 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/src/main/java/cn/stock/market/domain/basic/repository/RetifiveStockRepository.java b/src/main/java/cn/stock/market/domain/basic/repository/RetifiveStockRepository.java
index 96833cf..074f45e 100644
--- a/src/main/java/cn/stock/market/domain/basic/repository/RetifiveStockRepository.java
+++ b/src/main/java/cn/stock/market/domain/basic/repository/RetifiveStockRepository.java
@@ -5,11 +5,10 @@ import cn.qutaojing.common.domain.respostory.SimplePoConvertEntityRepository;
 import cn.qutaojing.common.utils.SpringUtils;
 import cn.stock.market.domain.basic.convert.RetifiveStockConvert;
 import cn.stock.market.domain.basic.entity.RetifiveStock;
+import cn.stock.market.infrastructure.db.po.QRetifiveStockPO;
 import cn.stock.market.infrastructure.db.po.RetifiveStockPO;
 import cn.stock.market.infrastructure.db.repo.RetifiveStockRepo;
 import com.rp.spring.jpa.GenericJpaRepository;
-import java.lang.Integer;
-import java.lang.Override;
 import lombok.RequiredArgsConstructor;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Repository;
@@ -30,6 +29,8 @@ public class RetifiveStockRepository extends SimplePoConvertEntityRepository<Ret
 
   final RetifiveStockConvert convert;
 
+  final QRetifiveStockPO p = QRetifiveStockPO.retifiveStockPO;
+
   @Override
   public GenericJpaRepository<RetifiveStockPO, Integer> repo() {
     return repo;
@@ -43,4 +44,8 @@ public class RetifiveStockRepository extends SimplePoConvertEntityRepository<Ret
   public static RetifiveStockRepository of() {
     return SpringUtils.getBean(RetifiveStockRepository.class);
   }
+
+  public RetifiveStock findBtStockByCoCode(String coCode) {
+    return findOneIfMutil(p.symbol.eq(coCode));
+  }
 }
diff --git a/src/main/java/cn/stock/market/dto/RetifiveStockInfo.java b/src/main/java/cn/stock/market/dto/RetifiveStockInfo.java
index 73cfb67..4158a3a 100644
--- a/src/main/java/cn/stock/market/dto/RetifiveStockInfo.java
+++ b/src/main/java/cn/stock/market/dto/RetifiveStockInfo.java
@@ -60,5 +60,8 @@ public class RetifiveStockInfo {
 
     @ApiModelProperty("股票当前的交易状态: 1-表示退市Closed, 2-Opening Auction, 3-Continuous Trading, 15-Intra-day Close (scheduled)")
     private String status;
+
+    @ApiModelProperty("股票id")
+    private Integer id;
 }
 
diff --git a/src/main/java/cn/stock/market/web/MessageRetifiveController.java b/src/main/java/cn/stock/market/web/MessageRetifiveController.java
index c5f002d..c296323 100644
--- a/src/main/java/cn/stock/market/web/MessageRetifiveController.java
+++ b/src/main/java/cn/stock/market/web/MessageRetifiveController.java
@@ -180,6 +180,11 @@ public class MessageRetifiveController {
                 RefreshMsg refreshMsg = appClient.getMessageFuture().get(10, TimeUnit.SECONDS);// 设置超时时间，例如10秒
                 if (DataType.DataTypes.FIELD_LIST == refreshMsg.payload().dataType()){
                     RetifiveStockInfo retifiveStockInfo = decode3(refreshMsg.payload().fieldList(), refreshMsg.name());
+                    //获取股票id
+                    RetifiveStock stock = retifiveStockService.repository().findBtStockByCoCode(itemName);
+                    if(stock != null){
+                        retifiveStockInfo.setId(stock.getId());
+                    }
                     return ServerResponse.createBySuccess("操作成功",retifiveStockInfo);
                 }
             } catch (Exception e) {