From 55e607b91e8058a7fe713fa07cd4c829b5593aa5 Mon Sep 17 00:00:00 2001 From: leutu <leutu@qq.com> Date: 星期二, 27 八月 2024 09:30:57 +0800 Subject: [PATCH] 更新 --- src/main/java/com/yb/util/OkHttpUtil.java | 38 ++++++++++++++++++++++++++++---------- 1 files changed, 28 insertions(+), 10 deletions(-) diff --git a/src/main/java/com/yb/util/OkHttpUtil.java b/src/main/java/com/yb/util/OkHttpUtil.java index 94d050e..9cf62e2 100644 --- a/src/main/java/com/yb/util/OkHttpUtil.java +++ b/src/main/java/com/yb/util/OkHttpUtil.java @@ -3,6 +3,7 @@ import com.alibaba.fastjson.JSON; import okhttp3.*; import org.springframework.beans.factory.annotation.Value; +import org.springframework.scheduling.annotation.Async; import org.springframework.stereotype.Component; import org.springframework.web.bind.annotation.RestController; @@ -10,6 +11,7 @@ import java.util.ArrayList; import java.util.HashMap; import java.util.List; +import java.util.concurrent.TimeUnit; @Component @RestController @@ -26,8 +28,13 @@ // get璇锋眰 public String getMessage(String url) { try { - OkHttpClient client = new OkHttpClient(); + OkHttpClient client = new OkHttpClient.Builder() + .connectTimeout(5, TimeUnit.SECONDS) + .readTimeout(5, TimeUnit.SECONDS) + .writeTimeout(5, TimeUnit.SECONDS) + .build(); Request request = new Request.Builder() + .url(url) .build(); @@ -78,12 +85,14 @@ return null ; } + @Async public String requesFast(String message){ - OkHttpClient client = new OkHttpClient(); + System.out.println("fastgpt:"+message) ; + OkHttpClient client = new OkHttpClient(); Response response = null ; // 鍒涘缓璇锋眰澶达紙Headers锛� okhttp3.Headers headers = new okhttp3.Headers.Builder() - .add("Authorization", "Bearer fastgpt-rML09d7nyx8Bn2Al3UipZia3Q8FNuXdekU1nyMlgCfHNwS8NN1giSPNZ1y6Wmt") + .add("Authorization", "Bearer "+apikey) .build(); /* @@ -105,7 +114,7 @@ // 鏋勫缓璇锋眰浣擄紙RequestBody锛夛紝杩欓噷浣跨敤鐨勬槸瀛楃涓茬被鍨嬬殑璇锋眰鏁版嵁 HashMap<String,Object> hashMap = new HashMap<>() ; - hashMap.put("chatId",""+Math.random()*100000); + hashMap.put("chatId","1111"); hashMap.put("stream","false"); hashMap.put("detail","false"); List<HashMap<String,String>> list = new ArrayList<>(); @@ -130,7 +139,15 @@ if (response.isSuccessful()) { //System.out.println("Success: " + response.body().string()); String rep = response.body().string(); - return rep ; + if( rep.indexOf("flowResponses") > 0) { + int index = rep.indexOf("flowResponses"); + int index_s = rep.indexOf("data", index); + String data = rep.substring(index_s + 5); + System.out.println("data is ok...."); + return data; + }else{ + return rep ; + } } else { System.out.println("Error: " + response.code() + " " + response.message()); } @@ -139,15 +156,15 @@ } - return response != null?response.body().toString():"null" ; + return null ; } - public String requesFast_kl(String parentId){ + public String requesFast_kl(){ OkHttpClient client = new OkHttpClient(); Response response = null ; // 鍒涘缓璇锋眰澶达紙Headers锛� okhttp3.Headers headers = new okhttp3.Headers.Builder() - .add("Authorization", "Bearer fastgpt-rML09d7nyx8Bn2Al3UipZia3Q8FNuXdekU1nyMlgCfHNwS8NN1giSPNZ1y6Wmt") + .add("Authorization", "Bearer "+apikey) .build(); /* @@ -159,7 +176,7 @@ // 鍒涘缓璇锋眰锛圧equest锛� - if( parentId != null ) fastgpt_kl = fastgpt_kl+"锛焢arentId="+parentId; + okhttp3.Request.Builder builder = new okhttp3.Request.Builder() .url(fastgpt_kl) .headers(headers) ; @@ -171,7 +188,8 @@ response = client.newCall(request).execute(); if (response.isSuccessful()) { //System.out.println("Success: " + response.body().string()); - String rep = response.body().string(); + String rep = response.body().toString(); + return rep ; } else { System.out.println("Error: " + response.code() + " " + response.message()); -- Gitblit v1.9.3