zoujiajian 1 жил өмнө
parent
commit
e6164867f1

+ 2 - 2
netflix-service/src/main/java/com/cyksj/service/chatgpt/impl/ChatGptAccountServiceImpl.java

@@ -130,8 +130,8 @@ public class ChatGptAccountServiceImpl implements ChatGptAccountService {
 
     @Override
     public void clearGptNum(Long relationId) throws Exception {
-        HttpRequest request = new HttpRequest("http://154.198.213.52:9611/cleanUserTokenLimit?relationId=" + relationId);
-        HttpResponse execute = request.method(Method.GET).execute();
+        HttpRequest request = new HttpRequest("https://saas.galaxyva.com/cleanUserTokenLimit?relationId=" + relationId);
+        HttpResponse execute = request.method(Method.GET).setConnectionTimeout(3000).timeout(3000).execute();
         String body = execute.body();
         JSONObject re = Jsons.parseObject(body, JSONObject.class);
         if (!body.contains("成功")) {

+ 2 - 0
netflix-service/src/main/java/com/cyksj/service/claude/ClaudeService.java

@@ -31,4 +31,6 @@ public interface ClaudeService {
     Long getConversationCount(String userToken, Long limitTime);
 
     Long carLimit(String carId);
+
+    void clearNumByRelationId(Long relationId) throws Exception;
 }

+ 17 - 0
netflix-service/src/main/java/com/cyksj/service/claude/impl/ClaudeServiceImpl.java

@@ -4,11 +4,15 @@ import cn.hutool.core.date.DateTime;
 import cn.hutool.core.lang.UUID;
 import cn.hutool.core.util.RandomUtil;
 import cn.hutool.core.util.StrUtil;
+import cn.hutool.http.HttpRequest;
+import cn.hutool.http.HttpResponse;
 import cn.hutool.http.HttpUtil;
+import cn.hutool.http.Method;
 import cn.hutool.json.JSONObject;
 import cn.hutool.json.JSONUtil;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.cyksj.common.exception.BusinessRuntimeException;
+import com.cyksj.common.util.Jsons;
 import com.cyksj.common.util.StringUtil;
 import com.cyksj.mapper.*;
 import com.cyksj.model.entity.*;
@@ -234,6 +238,19 @@ public class ClaudeServiceImpl implements ClaudeService {
         return 0L;
     }
 
+    @Override
+    public void clearNumByRelationId(Long relationId) throws Exception {
+        ClaudeUser claudeUser = claudeUserMapper.selectOne(Wrappers.lambdaQuery(ClaudeUser.class).eq(ClaudeUser::getRelationId, relationId));
+
+        HttpRequest request = new HttpRequest("http://147.79.20.183:9611/claude/cleanUserTokenLimit?userToken=" + claudeUser.getUserToken());
+        HttpResponse execute = request.method(Method.GET).setConnectionTimeout(3000).timeout(3000).execute();
+        String body = execute.body();
+        JSONObject re = Jsons.parseObject(body, JSONObject.class);
+        if (!body.contains("清除成功")) {
+            throw BusinessRuntimeException.getInstance(re.getStr("message"));
+        }
+    }
+
 
     /**
      * 获取车位信息

+ 12 - 0
netflix-web/src/main/java/com/cyksj/web/controller/manage/corp/WxCorpController.java

@@ -33,6 +33,7 @@ import com.cyksj.model.request.corp.CorpCommonReq;
 import com.cyksj.model.views.*;
 import com.cyksj.redis.RedisService;
 import com.cyksj.service.chatgpt.ChatGptAccountService;
+import com.cyksj.service.claude.ClaudeService;
 import com.cyksj.service.corp.CorpService;
 import com.cyksj.service.corp.CorpUserService;
 import com.cyksj.service.mange.CmsOrderDonService;
@@ -95,6 +96,8 @@ public class WxCorpController {
 
 	private final ChatGptAccountService chatGptAccountService;
 
+	private final ClaudeService claudeService;
+
 	private final OrderDonMapper orderDonMapper;
 
 	private final static String CORP_UUID = "corp-uuid";
@@ -554,6 +557,15 @@ public class WxCorpController {
 		return GatewayResponse.SUCCESS.newBuilder().toResult();
 	}
 
+	/**
+	 * 清除claude次数
+	 */
+	@PutMapping("/clear/claude/num")
+	public Result<String> clearClaudeNum(Long relationId) throws Exception {
+		claudeService.clearNumByRelationId(relationId);
+		return GatewayResponse.SUCCESS.newBuilder().toResult();
+	}
+
 	/**
 	 * 获取手机临时验证码
 	 */