Jelajahi Sumber

停止生产接口调用 并对前端封装接口

wangmiaomiao 10 bulan lalu
induk
melakukan
0d6683b86b

+ 14 - 0
slibra-admin/src/main/java/com/slibra/web/controller/business/FrontController.java

@@ -347,6 +347,20 @@ public class FrontController extends BaseController {
     }
 
 
+    /**
+     *
+     * 大模型停止生成
+     * @param sessionId
+     * @return
+     */
+    @GetMapping("/bigModel/warning/stopChat/{sessionId}")
+    public AjaxResult stopChat(@PathVariable String sessionId)
+    {
+        log.info("进入了 大模型停止生成  接口");
+        return AjaxResult.success(frontService.stopChat(sessionId));
+    }
+
+
     //--------------------下面是测试的方法--------------------
     /*@GetMapping("/test/data")
     public void testData(){

+ 2 - 2
slibra-admin/src/main/resources/application.yml

@@ -78,13 +78,13 @@ spring:
   # redis 配置
   redis:
     # 地址
-    host: localhost
+    host: 10.0.0.24
     # 端口,默认为6379
     port: 6379
     # 数据库索引
     database: 0
     # 密码
-    password:
+    password: ^YHN&UJM
     # 连接超时时间
     timeout: 10s
     lettuce:

+ 4 - 0
slibra-common/src/main/java/com/slibra/common/constant/MyConstants.java

@@ -8,6 +8,10 @@ public class MyConstants {
     public static final String SUCCESS = "操作成功";
 
     public static final String PERCENT_SYMBOL = "%";
+    public static final String SESSION_ID = "session_id";
+    public static final String STOP_GEN_STREAM = "stop_gen_stream";
+
+
 
     //报警计算使用的比例  默认是1.1倍
     public static final String XIAOSUANYAN_BZK = "15";

+ 10 - 5
slibra-common/src/main/java/com/slibra/common/core/redis/RedisCache.java

@@ -1,12 +1,12 @@
 package com.slibra.common.core.redis;
 
-import java.util.Collection;
-import java.util.Iterator;
-import java.util.List;
-import java.util.Map;
-import java.util.Set;
+import java.util.*;
 import java.util.concurrent.TimeUnit;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.data.redis.connection.stream.MapRecord;
+import org.springframework.data.redis.connection.stream.RecordId;
+import org.springframework.data.redis.connection.stream.StreamOffset;
+import org.springframework.data.redis.connection.stream.StreamRecords;
 import org.springframework.data.redis.core.BoundSetOperations;
 import org.springframework.data.redis.core.HashOperations;
 import org.springframework.data.redis.core.RedisTemplate;
@@ -265,4 +265,9 @@ public class RedisCache
     {
         return redisTemplate.keys(pattern);
     }
+
+
+    public RecordId streamXadd(String key, HashMap map){
+        return redisTemplate.opsForStream().add(key, map);
+    }
 }

+ 1 - 0
slibra-quartz/src/main/java/com/slibra/quartz/task/AsyncTask.java

@@ -1773,6 +1773,7 @@ public class AsyncTask {
         tXinyiBigTableHour.setHyNh31Jqr(tXinyiRobot.getNh31Jqr());
         tXinyiBigTableHour.setHyNh32Jqr(tXinyiRobot.getNh32Jqr());
         tXinyiBigTableHour.setHyNh31Yz(tXinyiRobot.getNh31Yz());
+        tXinyiBigTableHour.setHyNo3Hlc1Yz(tXinyiRobot.getNo3Hlc1Yz());
         tXinyiBigTableHour.setHyNh32Yz(tXinyiRobot.getNh32Yz());
         tXinyiBigTableHour.setHyNo3Hlc2Yz(tXinyiRobot.getNo3Hlc2Yz());
         tXinyiBigTableHour.setHyNo3Hlj1Jqr(tXinyiRobot.getNo3Hlj1Jqr());

+ 2 - 0
slibra-system/src/main/java/com/slibra/business/service/IFrontService.java

@@ -46,4 +46,6 @@ public interface IFrontService
     JSONObject customWorkOrderHandle(WorkOrderReq workOrderReq);
 
     JSONObject customWorkOrderHandleByData(WorkOrderReq workOrderReq, List<WorkOrderRes> workOrderRes);
+
+    String stopChat(String sessionId);
 }

+ 14 - 0
slibra-system/src/main/java/com/slibra/business/service/impl/FrontServiceImpl.java

@@ -1,5 +1,6 @@
 package com.slibra.business.service.impl;
 
+import com.alibaba.fastjson2.JSON;
 import com.alibaba.fastjson2.JSONObject;
 import com.slibra.business.domain.*;
 import com.slibra.business.mapper.*;
@@ -10,6 +11,7 @@ import com.slibra.business.res.XinyiDailyFeeSimple;
 import com.slibra.business.res.XinyiIndustrySimple;
 import com.slibra.business.service.IFrontService;
 import com.slibra.common.DecimalUtils;
+import com.slibra.common.core.redis.RedisCache;
 import com.slibra.common.exception.ServiceException;
 import com.slibra.common.utils.DateUtils;
 import com.slibra.common.utils.SecurityUtils;
@@ -17,6 +19,7 @@ import com.slibra.common.utils.StringUtils;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.BeanUtils;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.data.redis.connection.stream.RecordId;
 import org.springframework.stereotype.Service;
 import org.springframework.util.CollectionUtils;
 
@@ -57,6 +60,8 @@ public class FrontServiceImpl implements IFrontService {
     @Autowired
     private TXinyiChatRecordMapper xinyiChatRecordMapper;
 
+    @Autowired
+    private RedisCache redisCache;
     @Override
     public XinyiIndustrySimple countInfo() {
         //查询基本的工业库数据
@@ -374,6 +379,15 @@ public class FrontServiceImpl implements IFrontService {
         return jsonObject;
     }
 
+    @Override
+    public String stopChat(String sessionId) {
+        HashMap<String, Object> map = new HashMap<>();
+        map.put(SESSION_ID, sessionId);
+        RecordId stopGenStream = redisCache.streamXadd(STOP_GEN_STREAM, map);
+        log.info("返回结果是{}", JSON.toJSONString(stopGenStream));
+        return "操作成功";
+    }
+
     private void buildIndustryData(TXinyiIndustry industry, WorkOrderRes workOrderRes, WorkOrderReq workOrderReq) {
         if(workOrderReq.getJsSlq())
             workOrderRes.setJsSlq(DecimalUtils.getAbsAndScale(industry.getJsSlq(), 2));