diff --git a/pom.xml b/pom.xml
index 7777b04..806d94d 100644
--- a/pom.xml
+++ b/pom.xml
@@ -7,6 +7,19 @@
com.bwie
test-10.01
1.0-SNAPSHOT
+
+
+
+ org.apache.maven.plugins
+ maven-compiler-plugin
+
+ 8
+ 8
+
+
+
+
+
org.junit.jupiter
@@ -40,11 +53,13 @@
Java-WebSocket
1.3.8
+
com.squareup.okhttp3
okhttp
- 4.10.0
+ 4.9.3
+
org.projectlombok
lombok
@@ -91,6 +106,8 @@
aws-java-sdk-core
1.12.68
+
+
diff --git a/src/main/java/com/bwie/WorkApplication.java b/src/main/java/com/bwie/WorkApplication.java
new file mode 100644
index 0000000..8e80ef3
--- /dev/null
+++ b/src/main/java/com/bwie/WorkApplication.java
@@ -0,0 +1,12 @@
+package com.bwie;
+
+import org.springframework.boot.SpringApplication;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+
+@SpringBootApplication
+public class WorkApplication {
+
+ public static void main(String[] args) {
+ SpringApplication.run(WorkApplication.class);
+ }
+}
diff --git a/src/main/java/com/bwie/common/MsgDTO.java b/src/main/java/com/bwie/common/MsgDTO.java
new file mode 100644
index 0000000..b4a0e03
--- /dev/null
+++ b/src/main/java/com/bwie/common/MsgDTO.java
@@ -0,0 +1,38 @@
+package com.bwie.common;
+
+import com.fasterxml.jackson.annotation.JsonInclude;
+import lombok.AllArgsConstructor;
+import lombok.Builder;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+
+@Data
+@AllArgsConstructor
+@NoArgsConstructor
+@Builder
+@JsonInclude(JsonInclude.Include.NON_NULL)
+public class MsgDTO {
+
+ private String role;
+ /**
+ * 消息内容
+ */
+ private String content;
+ /**
+ * 响应结果字段:结果序号,取值为[0,10]; 当前为保留字段,开发者可忽略
+ */
+ private Integer index;
+
+ public static final String ROLE_USER = "user";
+ public static final String ROLE_ASSISTANT = "assistant";
+
+ public static MsgDTO createUserMsg(String content) {
+ return new MsgDTO(ROLE_USER, content, null);
+ }
+
+ public static MsgDTO createAssistantMsg(String content) {
+ return new MsgDTO(ROLE_ASSISTANT, content, null);
+ }
+
+}
diff --git a/src/main/java/com/bwie/common/LoginRequest.java b/src/main/java/com/bwie/common/request/LoginRequest.java
similarity index 77%
rename from src/main/java/com/bwie/common/LoginRequest.java
rename to src/main/java/com/bwie/common/request/LoginRequest.java
index 64eb688..63f9109 100644
--- a/src/main/java/com/bwie/common/LoginRequest.java
+++ b/src/main/java/com/bwie/common/request/LoginRequest.java
@@ -1,4 +1,4 @@
-package com.bwie.common;
+package com.bwie.common.request;
import lombok.Data;
diff --git a/src/main/java/com/bwie/common/request/RequestDTO.java b/src/main/java/com/bwie/common/request/RequestDTO.java
new file mode 100644
index 0000000..de88b20
--- /dev/null
+++ b/src/main/java/com/bwie/common/request/RequestDTO.java
@@ -0,0 +1,93 @@
+package com.bwie.common.request;
+
+/**
+ * @author 冯凯
+ * @version 1.0
+ * @description: TODO
+ * @date 2023/9/29 13:56
+ */
+
+import com.alibaba.fastjson.annotation.JSONField;
+import com.bwie.common.MsgDTO;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+import java.util.List;
+
+/**
+ * 请求参数
+ * 对应生成的 JSON 结构参考 resources/demo-json/request.json
+ */
+@NoArgsConstructor
+@Data
+public class RequestDTO {
+
+ @JsonProperty("header")
+ private HeaderDTO header;
+ @JsonProperty("parameter")
+ private ParameterDTO parameter;
+ @JsonProperty("payload")
+ private PayloadDTO payload;
+
+ @NoArgsConstructor
+ @Data
+ @AllArgsConstructor
+ public static class HeaderDTO {
+ /**
+ * 应用appid,从开放平台控制台创建的应用中获取
+ */
+ @JSONField(name = "app_id")
+ private String appId;
+ /**
+ * 每个用户的id,用于区分不同用户,最大长度32
+ */
+ @JSONField(name = "uid")
+ private String uid;
+ }
+
+ @NoArgsConstructor
+ @Data
+ @AllArgsConstructor
+ public static class ParameterDTO {
+ private ChatDTO chat;
+
+ @NoArgsConstructor
+ @Data
+ @AllArgsConstructor
+ public static class ChatDTO {
+ /**
+ * 指定访问的领域,general指向V1.5版本 generalv2指向V2版本。注意:不同的取值对应的url也不一样!
+ */
+ @JsonProperty("domain")
+ private String domain;
+ /**
+ * 核采样阈值。用于决定结果随机性,取值越高随机性越强即相同的问题得到的不同答案的可能性越高
+ */
+ @JsonProperty("temperature")
+ private Float temperature;
+ /**
+ * 模型回答的tokens的最大长度
+ */
+ @JSONField(name = "max_tokens")
+ private Integer maxTokens;
+ }
+ }
+
+ @NoArgsConstructor
+ @Data
+ @AllArgsConstructor
+ public static class PayloadDTO {
+ @JsonProperty("message")
+ private MessageDTO message;
+
+ @NoArgsConstructor
+ @Data
+ @AllArgsConstructor
+ public static class MessageDTO {
+ @JsonProperty("text")
+ private List text;
+ }
+ }
+}
diff --git a/src/main/java/com/bwie/common/response/ResponseDTO.java b/src/main/java/com/bwie/common/response/ResponseDTO.java
new file mode 100644
index 0000000..b4436fb
--- /dev/null
+++ b/src/main/java/com/bwie/common/response/ResponseDTO.java
@@ -0,0 +1,112 @@
+package com.bwie.common.response;
+
+import com.bwie.common.MsgDTO;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+import java.util.List;
+
+/**
+ * 返回参数
+ * 对应生成的 JSON 结构参考 resources/demo-json/response.json
+ */
+@NoArgsConstructor
+@Data
+public class ResponseDTO {
+
+ @JsonProperty("header")
+ private HeaderDTO header;
+ @JsonProperty("payload")
+ private PayloadDTO payload;
+
+ @NoArgsConstructor
+ @Data
+ public static class HeaderDTO {
+ /**
+ * 错误码,0表示正常,非0表示出错
+ */
+ @JsonProperty("code")
+ private Integer code;
+ /**
+ * 会话是否成功的描述信息
+ */
+ @JsonProperty("message")
+ private String message;
+ /**
+ * 会话的唯一id,用于讯飞技术人员查询服务端会话日志使用,出现调用错误时建议留存该字段
+ */
+ @JsonProperty("sid")
+ private String sid;
+ /**
+ * 会话状态,取值为[0,1,2];0代表首次结果;1代表中间结果;2代表最后一个结果
+ */
+ @JsonProperty("status")
+ private Integer status;
+ }
+
+ @NoArgsConstructor
+ @Data
+ public static class PayloadDTO {
+ @JsonProperty("choices")
+ private ChoicesDTO choices;
+ /**
+ * 在最后一次结果返回
+ */
+ @JsonProperty("usage")
+ private UsageDTO usage;
+
+ @NoArgsConstructor
+ @Data
+ public static class ChoicesDTO {
+ /**
+ * 文本响应状态,取值为[0,1,2]; 0代表首个文本结果;1代表中间文本结果;2代表最后一个文本结果
+ */
+ @JsonProperty("status")
+ private Integer status;
+ /**
+ * 返回的数据序号,取值为[0,9999999]
+ */
+ @JsonProperty("seq")
+ private Integer seq;
+ /**
+ * 响应文本
+ */
+ @JsonProperty("text")
+ private List text;
+
+ }
+
+ @NoArgsConstructor
+ @Data
+ public static class UsageDTO {
+ @JsonProperty("text")
+ private TextDTO text;
+
+ @NoArgsConstructor
+ @Data
+ public static class TextDTO {
+ /**
+ * 保留字段,可忽略
+ */
+ @JsonProperty("question_tokens")
+ private Integer questionTokens;
+ /**
+ * 包含历史问题的总tokens大小
+ */
+ @JsonProperty("prompt_tokens")
+ private Integer promptTokens;
+ /**
+ * 回答的tokens大小
+ */
+ @JsonProperty("completion_tokens")
+ private Integer completionTokens;
+ /**
+ * prompt_tokens和completion_tokens的和,也是本次交互计费的tokens大小
+ */
+ @JsonProperty("total_tokens")
+ private Integer totalTokens;
+ }
+ }
+ }
+}
diff --git a/src/main/java/com/bwie/component/XfXhStreamClient.java b/src/main/java/com/bwie/component/XfXhStreamClient.java
new file mode 100644
index 0000000..dcdc847
--- /dev/null
+++ b/src/main/java/com/bwie/component/XfXhStreamClient.java
@@ -0,0 +1,153 @@
+package com.bwie.component;
+
+import com.alibaba.fastjson.JSONObject;
+
+import com.bwie.common.MsgDTO;
+import com.bwie.common.request.RequestDTO;
+import com.bwie.config.XfXhConfig;
+import lombok.extern.slf4j.Slf4j;
+import okhttp3.*;
+import org.java_websocket.WebSocket;
+import org.java_websocket.WebSocketListener;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.stereotype.Component;
+
+import javax.annotation.Resource;
+import javax.crypto.Mac;
+import javax.crypto.spec.SecretKeySpec;
+import java.net.URL;
+import java.nio.charset.StandardCharsets;
+import java.text.SimpleDateFormat;
+import java.util.*;
+
+
+@Component
+@Slf4j
+public class XfXhStreamClient {
+
+ /**
+ * @author fst
+ */
+ @Resource
+ private XfXhConfig xfXhConfig;
+
+ @Value("${xfxh.QPS}")
+ private int connectionTokenCount;
+
+ /**
+ * 获取令牌
+ */
+ public static int GET_TOKEN_STATUS = 0;
+ /**
+ * 归还令牌
+ */
+ public static int BACK_TOKEN_STATUS = 1;
+
+ /**
+ * 操作令牌
+ *
+ * @param status 0-获取令牌 1-归还令牌
+ * @return 是否操作成功
+ */
+ public synchronized boolean operateToken(int status) {
+ if (status == GET_TOKEN_STATUS) {
+ // 获取令牌
+ if (connectionTokenCount != 0) {
+ // 说明还有令牌,将令牌数减一
+ connectionTokenCount -= 1;
+ return true;
+ } else {
+ return false;
+ }
+ } else {
+ // 放回令牌
+ connectionTokenCount += 1;
+ return true;
+ }
+ }
+
+ /**
+ * 发送消息
+ *
+ * @param uid 每个用户的id,用于区分不同用户
+ * @param msgList 发送给大模型的消息,可以包含上下文内容
+ * @return 获取websocket连接,以便于我们在获取完整大模型回复后手动关闭连接
+ */
+ public WebSocket sendMsg(String uid, List msgList, WebSocketListener listener) {
+ // 获取鉴权url
+ String authUrl = this.getAuthUrl();
+ // 鉴权方法生成失败,直接返回 null
+ if (authUrl == null) {
+ return null;
+ }
+ OkHttpClient okHttpClient = new OkHttpClient.Builder().build();
+ // 将 https/http 连接替换为 ws/wss 连接
+ String url = authUrl.replace("http://", "ws://").replace("https://", "wss://");
+ Request request = new Request.Builder().url(url).build();
+ // 建立 wss 连接
+ WebSocket webSocket = okHttpClient.newWebSocket(request, listener);
+ // 组装请求参数
+ RequestDTO requestDTO = getRequestParam(uid, msgList);
+ // 发送请求
+ webSocket.send(JSONObject.toJSONString(requestDTO));
+ return webSocket;
+ }
+
+ /**
+ * 生成鉴权方法,具体实现不用关心,这是讯飞官方定义的鉴权方式
+ *
+ * @return 鉴权访问大模型的路径
+ */
+ public String getAuthUrl() {
+ try {
+ URL url = new URL(xfXhConfig.getHostUrl());
+ // 时间
+ SimpleDateFormat format = new SimpleDateFormat("EEE, dd MMM yyyy HH:mm:ss z", Locale.US);
+ format.setTimeZone(TimeZone.getTimeZone("GMT"));
+ String date = format.format(new Date());
+ // 拼接
+ String preStr = "host: " + url.getHost() + "\n" +
+ "date: " + date + "\n" +
+ "GET " + url.getPath() + " HTTP/1.1";
+ // SHA256加密
+ Mac mac = Mac.getInstance("hmacsha256");
+ SecretKeySpec spec = new SecretKeySpec(xfXhConfig.getApiSecret().getBytes(StandardCharsets.UTF_8), "hmacsha256");
+ mac.init(spec);
+
+ byte[] hexDigits = mac.doFinal(preStr.getBytes(StandardCharsets.UTF_8));
+ // Base64加密
+ String sha = Base64.getEncoder().encodeToString(hexDigits);
+ // 拼接
+ String authorizationOrigin = String.format("api_key=\"%s\", algorithm=\"%s\", headers=\"%s\", signature=\"%s\"", xfXhConfig.getApiKey(), "hmac-sha256", "host date request-line", sha);
+ // 拼接地址
+ HttpUrl httpUrl = Objects.requireNonNull(HttpUrl.parse("https://" + url.getHost() + url.getPath())).newBuilder().
+ addQueryParameter("authorization", Base64.getEncoder().encodeToString(authorizationOrigin.getBytes(StandardCharsets.UTF_8))).
+ addQueryParameter("date", date).
+ addQueryParameter("host", url.getHost()).
+ build();
+
+ return httpUrl.toString();
+ } catch (Exception e) {
+ log.error("鉴权方法中发生错误:" + e.getMessage());
+ return null;
+ }
+ }
+
+ /**
+ * 获取请求参数
+ *
+ * @param uid 每个用户的id,用于区分不同用户
+ * @param msgList 发送给大模型的消息,可以包含上下文内容
+ * @return 请求DTO,该 DTO 转 json 字符串后生成的格式参考 resources/demo-json/request.json
+ */
+ public RequestDTO getRequestParam(String uid, List msgList) {
+ RequestDTO requestDTO = new RequestDTO();
+ requestDTO.setHeader(new RequestDTO.HeaderDTO(xfXhConfig.getAppId(), uid));
+ requestDTO.setParameter(new RequestDTO.ParameterDTO(new RequestDTO.ParameterDTO.ChatDTO(xfXhConfig.getDomain(), xfXhConfig.getTemperature(), xfXhConfig.getMaxTokens())));
+ requestDTO.setPayload(new RequestDTO.PayloadDTO(new RequestDTO.PayloadDTO.MessageDTO(msgList)));
+ return requestDTO;
+ }
+
+
+
+}
diff --git a/src/main/java/com/bwie/config/CorsConfig.java b/src/main/java/com/bwie/config/CorsConfig.java
new file mode 100644
index 0000000..35f906c
--- /dev/null
+++ b/src/main/java/com/bwie/config/CorsConfig.java
@@ -0,0 +1,24 @@
+package com.bwie.config;
+
+import org.springframework.context.annotation.Configuration;
+import org.springframework.web.servlet.config.annotation.CorsRegistry;
+import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
+
+
+/**
+ * @author 冯凯
+ * @version 1.0
+ * @description: TODO
+ * @date 2023/10/3 14:43
+ */
+@Configuration
+public class CorsConfig implements WebMvcConfigurer {
+
+ public void addCorsMappings(CorsRegistry registry) {
+ registry.addMapping("/test/sendQuestion")
+ .allowedOrigins("http://localhost:9528")
+ .allowedMethods("GET", "POST")
+ .allowCredentials(true)
+ .maxAge(3600);
+ }
+}
diff --git a/src/main/java/com/bwie/config/XfXhConfig.java b/src/main/java/com/bwie/config/XfXhConfig.java
new file mode 100644
index 0000000..b4afe0a
--- /dev/null
+++ b/src/main/java/com/bwie/config/XfXhConfig.java
@@ -0,0 +1,54 @@
+package com.bwie.config;
+
+import lombok.Data;
+import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.context.annotation.Configuration;
+
+/**
+ * @author 冯凯
+ * @version 1.0
+ * @description: TODO
+ * @date 2023/9/29 13:50
+ */
+@Configuration
+@ConfigurationProperties(prefix = "xfxh")
+@Data
+public class XfXhConfig {
+
+
+ /**
+ * 服务引擎使用 讯飞星火认知大模型V2.0,如果使用 V1.5 需要将 hostUrl 修改为 https://spark-api.xf-yun.com/v1.1/chat
+ */
+ private String hostUrl;
+ /**
+ * 发送请求时指定的访问领域,如果是 V1.5版本 设置为 general,如果是 V2版本 设置为 generalv2
+ */
+ private String domain;
+ /**
+ * 核采样阈值。用于决定结果随机性,取值越高随机性越强即相同的问题得到的不同答案的可能性越高。取值 [0,1]
+ */
+ private Float temperature;
+ /**
+ * 模型回答的tokens的最大长度,V1.5取值为[1,4096],V2.0取值为[1,8192]。
+ */
+ private Integer maxTokens;
+ /**
+ * 大模型回复问题的最大响应时长,单位 s
+ */
+ private Integer maxResponseTime;
+ /**
+ * 用于权限验证,从服务接口认证信息中获取
+ */
+ private String appId;
+ /**
+ * 用于权限验证,从服务接口认证信息中获取
+ */
+ private String apiKey;
+ /**
+ * 用于权限验证,从服务接口认证信息中获取
+ */
+ private String apiSecret;
+
+
+
+}
diff --git a/src/main/java/com/bwie/controller/ChatController.java b/src/main/java/com/bwie/controller/ChatController.java
new file mode 100644
index 0000000..b7b848f
--- /dev/null
+++ b/src/main/java/com/bwie/controller/ChatController.java
@@ -0,0 +1,98 @@
+package com.bwie.controller;
+
+import cn.hutool.core.util.StrUtil;
+
+import com.bwie.common.MsgDTO;
+import com.bwie.component.XfXhStreamClient;
+import com.bwie.config.XfXhConfig;
+import com.bwie.listener.XfXhWebSocketListener;
+import lombok.extern.slf4j.Slf4j;
+import org.java_websocket.WebSocket;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestParam;
+import org.springframework.web.bind.annotation.RestController;
+
+import javax.annotation.Resource;
+import java.util.Collections;
+import java.util.UUID;
+
+/**
+ * @author 冯凯
+ * @version 1.0
+ * @description: TODO
+ * @date 2023/9/29 14:06
+ */
+@RestController
+@RequestMapping("/test")
+@Slf4j
+public class ChatController {
+
+
+ @Resource
+ private XfXhStreamClient xfXhStreamClient;
+
+ @Resource
+ private XfXhConfig xfXhConfig;
+
+
+ /**
+ * 发送问题
+ *
+ * @param question 问题
+ * @return 星火大模型的回答
+ */
+ @GetMapping("/sendQuestion")
+ public String sendQuestion(@RequestParam("question") String question) {
+ // 如果是无效字符串,则不对大模型进行请求
+ if (StrUtil.isBlank(question)) {
+ return "无效问题,请重新输入";
+ }
+ // 获取连接令牌
+ if (!xfXhStreamClient.operateToken(XfXhStreamClient.GET_TOKEN_STATUS)) {
+ return "当前大模型连接数过多,请稍后再试";
+ }
+
+ // 创建消息对象
+ MsgDTO msgDTO = MsgDTO.createUserMsg(question);
+ // 创建监听器
+ XfXhWebSocketListener listener = new XfXhWebSocketListener();
+ // 发送问题给大模型,生成 websocket 连接
+
+ WebSocket webSocket = xfXhStreamClient.sendMsg(UUID.randomUUID().toString().substring(0, 10), Collections.singletonList(msgDTO), listener);
+
+ if (webSocket == null) {
+ // 归还令牌
+ xfXhStreamClient.operateToken(XfXhStreamClient.BACK_TOKEN_STATUS);
+ return "系统内部错误,请联系管理员";
+ }
+ try {
+ int count = 0;
+ // 为了避免死循环,设置循环次数来定义超时时长
+ int maxCount = xfXhConfig.getMaxResponseTime() * 5;
+ while (count <= maxCount) {
+ Thread.sleep(200);
+ if (listener.isWsCloseFlag()) {
+ break;
+ }
+ count++;
+ }
+ if (count > maxCount) {
+ return "大模型响应超时,请联系管理员";
+ }
+ // 响应大模型的答案
+ System.out.println(listener.getAnswer().toString());
+ return listener.getAnswer().toString();
+ } catch (InterruptedException e) {
+ log.error("错误:" + e.getMessage());
+ return "系统内部错误,请联系管理员";
+ } finally {
+ // 关闭 websocket 连接
+ webSocket.close(1000, "");
+ // 归还令牌
+ xfXhStreamClient.operateToken(XfXhStreamClient.BACK_TOKEN_STATUS);
+ }
+ }
+
+
+}
diff --git a/src/main/java/com/bwie/controller/UserController.java b/src/main/java/com/bwie/controller/UserController.java
index bc3892a..f074150 100644
--- a/src/main/java/com/bwie/controller/UserController.java
+++ b/src/main/java/com/bwie/controller/UserController.java
@@ -1,6 +1,6 @@
package com.bwie.controller;
-import com.bwie.common.LoginRequest;
+import com.bwie.common.request.LoginRequest;
import com.bwie.common.Result;
import com.bwie.service.impl.UserServiceImpl;
import org.springframework.beans.factory.annotation.Autowired;
diff --git a/src/main/java/com/bwie/listener/XfXhWebSocketListener.java b/src/main/java/com/bwie/listener/XfXhWebSocketListener.java
new file mode 100644
index 0000000..5435448
--- /dev/null
+++ b/src/main/java/com/bwie/listener/XfXhWebSocketListener.java
@@ -0,0 +1,72 @@
+package com.bwie.listener;
+
+import com.alibaba.fastjson.JSONObject;
+import com.bwie.common.MsgDTO;
+import com.bwie.common.response.ResponseDTO;
+import lombok.extern.slf4j.Slf4j;
+import okhttp3.Response;
+import okhttp3.WebSocket;
+import okhttp3.WebSocketListener;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+/**
+ * @author 冯凯
+ * @version 1.0
+ * @description: TODO
+ * @date 2023/9/29 13:58
+ */
+@Slf4j
+public class XfXhWebSocketListener extends WebSocketListener {
+ private StringBuilder answer = new StringBuilder();
+
+ private boolean wsCloseFlag = false;
+
+ public StringBuilder getAnswer() {
+ return answer;
+ }
+
+ public boolean isWsCloseFlag() {
+ return wsCloseFlag;
+ }
+
+ @Override
+ public void onOpen(@NotNull WebSocket webSocket, @NotNull Response response) {
+ super.onOpen(webSocket, response);
+ }
+
+ @Override
+ public void onMessage(@NotNull WebSocket webSocket, @NotNull String text) {
+ super.onMessage(webSocket, text);
+ // 将大模型回复的 JSON 文本转为 ResponseDTO 对象
+ ResponseDTO responseData = JSONObject.parseObject(text, ResponseDTO.class);
+ // 如果响应数据中的 header 的 code 值不为 0,则表示响应错误
+ if (responseData.getHeader().getCode() != 0) {
+ // 日志记录
+ log.error("发生错误,错误码为:" + responseData.getHeader().getCode() + "; " + "信息:" + responseData.getHeader().getMessage());
+ // 设置回答
+ this.answer = new StringBuilder("大模型响应错误,请稍后再试");
+ // 关闭连接标识
+ wsCloseFlag = true;
+ return;
+ }
+ // 将回答进行拼接
+ for (MsgDTO msgDTO : responseData.getPayload().getChoices().getText()) {
+ this.answer.append(msgDTO.getContent());
+ }
+ // 对最后一个文本结果进行处理
+ if (2 == responseData.getHeader().getStatus()) {
+ wsCloseFlag = true;
+ }
+ }
+
+ @Override
+ public void onFailure(@NotNull WebSocket webSocket, @NotNull Throwable t, @Nullable Response response) {
+ super.onFailure(webSocket, t, response);
+ }
+
+ @Override
+ public void onClosed(@NotNull WebSocket webSocket, int code, @NotNull String reason) {
+ super.onClosed(webSocket, code, reason);
+ }
+}
diff --git a/src/main/java/com/bwie/mapper/UserMapper.java b/src/main/java/com/bwie/mapper/UserMapper.java
index 3c8d556..676879f 100644
--- a/src/main/java/com/bwie/mapper/UserMapper.java
+++ b/src/main/java/com/bwie/mapper/UserMapper.java
@@ -1,6 +1,6 @@
package com.bwie.mapper;
-import com.bwie.common.LoginRequest;
+import com.bwie.common.request.LoginRequest;
import org.apache.ibatis.annotations.Mapper;
import org.springframework.stereotype.Component;
diff --git a/src/main/java/com/bwie/service/UserService.java b/src/main/java/com/bwie/service/UserService.java
index 8cc56db..17fe14d 100644
--- a/src/main/java/com/bwie/service/UserService.java
+++ b/src/main/java/com/bwie/service/UserService.java
@@ -1,6 +1,6 @@
package com.bwie.service;
-import com.bwie.common.LoginRequest;
+import com.bwie.common.request.LoginRequest;
import com.bwie.common.Result;
public interface UserService {
diff --git a/src/main/java/com/bwie/service/impl/UserServiceImpl.java b/src/main/java/com/bwie/service/impl/UserServiceImpl.java
index c262134..3d4c32c 100644
--- a/src/main/java/com/bwie/service/impl/UserServiceImpl.java
+++ b/src/main/java/com/bwie/service/impl/UserServiceImpl.java
@@ -1,7 +1,7 @@
package com.bwie.service.impl;
import com.bwie.annotation.Idempotent;
-import com.bwie.common.LoginRequest;
+import com.bwie.common.request.LoginRequest;
import com.bwie.common.Result;
import com.bwie.mapper.UserMapper;
import com.bwie.service.UserService;
diff --git a/src/main/resources/application.yml b/src/main/resources/application.yml
index 45dcc4d..ec66547 100644
--- a/src/main/resources/application.yml
+++ b/src/main/resources/application.yml
@@ -9,7 +9,7 @@ spring:
password: root
name: defaultDataSource
mybatis:
- mapper-locations: mappers/*xml
+ mapper-locations: mapper/*xml
mybatis:
configuration:
map-underscore-to-camel-case: true
@@ -28,8 +28,8 @@ xfxh:
# 允许同时连接大模型的 websocket 数,如果是普通(免费)用户为 2,超过这个数连接响应会报错,具体参考官网。
QPS: 2
# 用于权限验证,从服务接口认证信息中获取
- appId: 4c5ea24c
+ appId: 3a5er24h
# 用于权限验证,从服务接口认证信息中获取
- apiKey: d53d6f5aae8cfee01420eeff6584057f
+ apiKey: d2hu6f5ace8fdea0982a0eaff658408dh
# 用于权限验证,从服务接口认证信息中获取
- apiSecret: Mzg3YTYxZjYyYTE0MTdiOTMyZTdlMTFi
+ apiSecret: Cst3AJIxEjDyCKQ0MTdiOQIyKIdlLOAi
diff --git a/src/main/resources/mapper/UserMapper.xml b/src/main/resources/mapper/UserMapper.xml
index 4073a41..c5e4ac9 100644
--- a/src/main/resources/mapper/UserMapper.xml
+++ b/src/main/resources/mapper/UserMapper.xml
@@ -4,7 +4,7 @@
"http://mybatis.org/dtd/mybatis-3-mapper.dtd">
-
diff --git a/target/classes/com/bwie/test/Test01.class b/target/classes/com/bwie/test/Test01.class
deleted file mode 100644
index d9f6315..0000000
Binary files a/target/classes/com/bwie/test/Test01.class and /dev/null differ
diff --git a/target/maven-archiver/pom.properties b/target/maven-archiver/pom.properties
deleted file mode 100644
index 0c2efea..0000000
--- a/target/maven-archiver/pom.properties
+++ /dev/null
@@ -1,5 +0,0 @@
-#Generated by Maven
-#Wed Oct 04 22:55:50 CST 2023
-version=1.0-SNAPSHOT
-groupId=com.bwie
-artifactId=test-10.01
diff --git a/target/maven-status/maven-compiler-plugin/compile/default-compile/createdFiles.lst b/target/maven-status/maven-compiler-plugin/compile/default-compile/createdFiles.lst
index 408068a..00ff73f 100644
--- a/target/maven-status/maven-compiler-plugin/compile/default-compile/createdFiles.lst
+++ b/target/maven-status/maven-compiler-plugin/compile/default-compile/createdFiles.lst
@@ -1 +1,4 @@
-com\bwie\test\Test01.class
+com\bwie\common\Result.class
+com\bwie\service\UserService.class
+com\bwie\common\request\LoginRequest.class
+META-INF\spring-configuration-metadata.json
diff --git a/target/maven-status/maven-compiler-plugin/compile/default-compile/inputFiles.lst b/target/maven-status/maven-compiler-plugin/compile/default-compile/inputFiles.lst
index e00223c..c328f80 100644
--- a/target/maven-status/maven-compiler-plugin/compile/default-compile/inputFiles.lst
+++ b/target/maven-status/maven-compiler-plugin/compile/default-compile/inputFiles.lst
@@ -1 +1,17 @@
+D:\lv6\test-10.01\src\main\java\com\bwie\common\response\ResponseDTO.java
+D:\lv6\test-10.01\src\main\java\com\bwie\controller\ChatController.java
+D:\lv6\test-10.01\src\main\java\com\bwie\config\CorsConfig.java
+D:\lv6\test-10.01\src\main\java\com\bwie\service\UserService.java
+D:\lv6\test-10.01\src\main\java\com\bwie\mapper\UserMapper.java
+D:\lv6\test-10.01\src\main\java\com\bwie\common\Result.java
+D:\lv6\test-10.01\src\main\java\com\bwie\controller\UserController.java
+D:\lv6\test-10.01\src\main\java\com\bwie\common\request\LoginRequest.java
+D:\lv6\test-10.01\src\main\java\com\bwie\service\impl\UserServiceImpl.java
+D:\lv6\test-10.01\src\main\java\com\bwie\common\MsgDTO.java
+D:\lv6\test-10.01\src\main\java\com\bwie\annotation\Idempotent.java
+D:\lv6\test-10.01\src\main\java\com\bwie\common\request\RequestDTO.java
+D:\lv6\test-10.01\src\main\java\com\bwie\component\XfXhStreamClient.java
+D:\lv6\test-10.01\src\main\java\com\bwie\aspect\IdempotentAspect.java
+D:\lv6\test-10.01\src\main\java\com\bwie\config\XfXhConfig.java
D:\lv6\test-10.01\src\main\java\com\bwie\test\Test01.java
+D:\lv6\test-10.01\src\main\java\com\bwie\WorkApplication.java
diff --git a/target/maven-status/maven-compiler-plugin/testCompile/default-testCompile/inputFiles.lst b/target/maven-status/maven-compiler-plugin/testCompile/default-testCompile/inputFiles.lst
deleted file mode 100644
index e69de29..0000000
diff --git a/target/test-10.01-1.0-SNAPSHOT.jar b/target/test-10.01-1.0-SNAPSHOT.jar
deleted file mode 100644
index 58af6c8..0000000
Binary files a/target/test-10.01-1.0-SNAPSHOT.jar and /dev/null differ