From 7f164cb632b22906ffcac2a107c83b84fc645340 Mon Sep 17 00:00:00 2001 From: 18404992668 <44167514@qq.com> Date: Wed, 4 Oct 2023 19:06:26 +0800 Subject: [PATCH] =?UTF-8?q?=E5=88=9D=E5=A7=8B=E5=8C=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .gitignore | 38 +++++ .idea/.gitignore | 8 + .idea/encodings.xml | 7 + .idea/inspectionProfiles/Project_Default.xml | 5 + .idea/misc.xml | 22 +++ .idea/vcs.xml | 6 + pom.xml | 99 ++++++++++++ .../com/spc/homework/ChatGPT/CorsConfig.java | 16 ++ .../com/spc/homework/ChatGPT/XfXhConfig.java | 45 ++++++ .../homework/ChatGPT/XfXhStreamClient.java | 144 ++++++++++++++++++ .../ChatGPT/XfXhWebSocketListener.java | 66 ++++++++ src/main/java/com/spc/homework/Result.java | 70 +++++++++ .../com/spc/homework/XfXhApplication.java | 11 ++ .../java/com/spc/homework/dto/MsgDTO.java | 36 +++++ .../spc/homework/dto/request/LoginReq.java | 11 ++ .../spc/homework/dto/request/RequestDTO.java | 81 ++++++++++ .../homework/dto/response/ResponseDTO.java | 107 +++++++++++++ .../com/spc/homework/lan/Independent.java | 14 ++ .../spc/homework/lan/IndependentAspect.java | 38 +++++ .../com/spc/homework/mapper/LoginMapper.java | 10 ++ .../spc/homework/service/LoginService.java | 8 + .../homework/service/LoginServiceImpl.java | 20 +++ src/main/java/com/spc/utli/DogGetApi.java | 114 ++++++++++++++ src/main/java/com/spc/utli/DuoJiYunUtil.java | 20 +++ src/main/java/com/spc/utli/S3.java | 41 +++++ src/main/resources/bootstrap.yml | 33 ++++ src/main/resources/mappers/user.xml | 9 ++ 27 files changed, 1079 insertions(+) create mode 100644 .gitignore create mode 100644 .idea/.gitignore create mode 100644 .idea/encodings.xml create mode 100644 .idea/inspectionProfiles/Project_Default.xml create mode 100644 .idea/misc.xml create mode 100644 .idea/vcs.xml create mode 100644 pom.xml create mode 100644 src/main/java/com/spc/homework/ChatGPT/CorsConfig.java create mode 100644 src/main/java/com/spc/homework/ChatGPT/XfXhConfig.java create mode 100644 src/main/java/com/spc/homework/ChatGPT/XfXhStreamClient.java create mode 100644 src/main/java/com/spc/homework/ChatGPT/XfXhWebSocketListener.java create mode 100644 src/main/java/com/spc/homework/Result.java create mode 100644 src/main/java/com/spc/homework/XfXhApplication.java create mode 100644 src/main/java/com/spc/homework/dto/MsgDTO.java create mode 100644 src/main/java/com/spc/homework/dto/request/LoginReq.java create mode 100644 src/main/java/com/spc/homework/dto/request/RequestDTO.java create mode 100644 src/main/java/com/spc/homework/dto/response/ResponseDTO.java create mode 100644 src/main/java/com/spc/homework/lan/Independent.java create mode 100644 src/main/java/com/spc/homework/lan/IndependentAspect.java create mode 100644 src/main/java/com/spc/homework/mapper/LoginMapper.java create mode 100644 src/main/java/com/spc/homework/service/LoginService.java create mode 100644 src/main/java/com/spc/homework/service/LoginServiceImpl.java create mode 100644 src/main/java/com/spc/utli/DogGetApi.java create mode 100644 src/main/java/com/spc/utli/DuoJiYunUtil.java create mode 100644 src/main/java/com/spc/utli/S3.java create mode 100644 src/main/resources/bootstrap.yml create mode 100644 src/main/resources/mappers/user.xml diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..5ff6309 --- /dev/null +++ b/.gitignore @@ -0,0 +1,38 @@ +target/ +!.mvn/wrapper/maven-wrapper.jar +!**/src/main/**/target/ +!**/src/test/**/target/ + +### IntelliJ IDEA ### +.idea/modules.xml +.idea/jarRepositories.xml +.idea/compiler.xml +.idea/libraries/ +*.iws +*.iml +*.ipr + +### Eclipse ### +.apt_generated +.classpath +.factorypath +.project +.settings +.springBeans +.sts4-cache + +### NetBeans ### +/nbproject/private/ +/nbbuild/ +/dist/ +/nbdist/ +/.nb-gradle/ +build/ +!**/src/main/**/build/ +!**/src/test/**/build/ + +### VS Code ### +.vscode/ + +### Mac OS ### +.DS_Store \ No newline at end of file diff --git a/.idea/.gitignore b/.idea/.gitignore new file mode 100644 index 0000000..35410ca --- /dev/null +++ b/.idea/.gitignore @@ -0,0 +1,8 @@ +# 默认忽略的文件 +/shelf/ +/workspace.xml +# 基于编辑器的 HTTP 客户端请求 +/httpRequests/ +# Datasource local storage ignored files +/dataSources/ +/dataSources.local.xml diff --git a/.idea/encodings.xml b/.idea/encodings.xml new file mode 100644 index 0000000..aa00ffa --- /dev/null +++ b/.idea/encodings.xml @@ -0,0 +1,7 @@ + + + + + + + \ No newline at end of file diff --git a/.idea/inspectionProfiles/Project_Default.xml b/.idea/inspectionProfiles/Project_Default.xml new file mode 100644 index 0000000..8d66637 --- /dev/null +++ b/.idea/inspectionProfiles/Project_Default.xml @@ -0,0 +1,5 @@ + + + + \ No newline at end of file diff --git a/.idea/misc.xml b/.idea/misc.xml new file mode 100644 index 0000000..9a301f6 --- /dev/null +++ b/.idea/misc.xml @@ -0,0 +1,22 @@ + + + + + + + + + + + + + + \ No newline at end of file diff --git a/.idea/vcs.xml b/.idea/vcs.xml new file mode 100644 index 0000000..94a25f7 --- /dev/null +++ b/.idea/vcs.xml @@ -0,0 +1,6 @@ + + + + + + \ No newline at end of file diff --git a/pom.xml b/pom.xml new file mode 100644 index 0000000..f250947 --- /dev/null +++ b/pom.xml @@ -0,0 +1,99 @@ + + + 4.0.0 + + com.spc + homework + 1.0-SNAPSHOT + + + 17 + 17 + UTF-8 + + + org.springframework.boot + spring-boot-starter-parent + 2.7.15 + + + + + + org.springframework.boot + spring-boot-configuration-processor + true + + + cn.hutool + hutool-all + 5.8.18 + + + org.springframework.boot + spring-boot-starter-web + + + com.alibaba + fastjson + 1.2.67 + + + org.java-websocket + Java-WebSocket + 1.3.8 + + + com.squareup.okhttp3 + okhttp + 4.10.0 + + + org.projectlombok + lombok + + + org.slf4j + slf4j-log4j12 + + + mysql + mysql-connector-java + 5.1.32 + + + org.mybatis.spring.boot + mybatis-spring-boot-starter + 2.3.1 + + + org.springframework.boot + spring-boot-starter-aop + + + org.json + json + 20210307 + + + commons-codec + commons-codec + 1.15 + + + com.amazonaws + aws-java-sdk-s3 + 1.12.68 + + + + com.amazonaws + aws-java-sdk-core + 1.12.68 + + + + + diff --git a/src/main/java/com/spc/homework/ChatGPT/CorsConfig.java b/src/main/java/com/spc/homework/ChatGPT/CorsConfig.java new file mode 100644 index 0000000..511c01c --- /dev/null +++ b/src/main/java/com/spc/homework/ChatGPT/CorsConfig.java @@ -0,0 +1,16 @@ +package com.spc.homework.ChatGPT; + +import org.springframework.context.annotation.Configuration; +import org.springframework.web.servlet.config.annotation.CorsRegistry; +import org.springframework.web.servlet.config.annotation.WebMvcConfigurer; + +@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/spc/homework/ChatGPT/XfXhConfig.java b/src/main/java/com/spc/homework/ChatGPT/XfXhConfig.java new file mode 100644 index 0000000..81a760d --- /dev/null +++ b/src/main/java/com/spc/homework/ChatGPT/XfXhConfig.java @@ -0,0 +1,45 @@ +package com.spc.homework.ChatGPT; + +import lombok.Data; +import org.springframework.boot.context.properties.ConfigurationProperties; +import org.springframework.context.annotation.Configuration; + +@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/spc/homework/ChatGPT/XfXhStreamClient.java b/src/main/java/com/spc/homework/ChatGPT/XfXhStreamClient.java new file mode 100644 index 0000000..a377421 --- /dev/null +++ b/src/main/java/com/spc/homework/ChatGPT/XfXhStreamClient.java @@ -0,0 +1,144 @@ +package com.spc.homework.ChatGPT; + +import com.alibaba.fastjson.JSONObject; +import com.spc.homework.dto.MsgDTO; +import com.spc.homework.dto.request.RequestDTO; +import lombok.extern.slf4j.Slf4j; +import okhttp3.*; +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 狐狸半面添 + * @create 2023-09-15 1:10 + */ + @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/spc/homework/ChatGPT/XfXhWebSocketListener.java b/src/main/java/com/spc/homework/ChatGPT/XfXhWebSocketListener.java new file mode 100644 index 0000000..850a5d4 --- /dev/null +++ b/src/main/java/com/spc/homework/ChatGPT/XfXhWebSocketListener.java @@ -0,0 +1,66 @@ +package com.spc.homework.ChatGPT; + +import com.alibaba.fastjson.JSONObject; +import com.spc.homework.dto.MsgDTO; +import com.spc.homework.dto.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; + +@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/spc/homework/Result.java b/src/main/java/com/spc/homework/Result.java new file mode 100644 index 0000000..829aac2 --- /dev/null +++ b/src/main/java/com/spc/homework/Result.java @@ -0,0 +1,70 @@ +package com.spc.homework; + +import lombok.Data; + +import java.io.Serializable; + +@Data +public class Result implements Serializable { + private static final long serialVersionUID = 1L; + /** + * 成功 + */ + public static final int SUCCESS = 200; + /** + * 失败 + */ + public static final int FAIL = 500; + /** + * 返回状态码 + */ + private int code; + /** + * 响应信息 + */ + private String msg; + /** + * 响应数据 + */ + private T data; + + public static Result success() { + return restResult(null, SUCCESS, "操作成功"); + } + + public static Result success(T data) { + return restResult(data, SUCCESS, "操作成功"); + } + + public static Result success(T data, String msg) { + return restResult(data, SUCCESS, msg); + } + + public static Result error() { + return restResult(null, FAIL, "操作失败"); + } + + public static Result error(String msg) { + return restResult(null, FAIL, msg); + } + + public static Result error(T data) { + return restResult(data, FAIL, "操作失败"); + } + + public static Result error(T data, String msg) { + return restResult(data, FAIL, msg); + } + + public static Result error(int code, String msg) { + return restResult(null, code, msg); + } + + private static Result restResult(T data, int code, String msg) { + Result apiResult = new Result<>(); + apiResult.setCode(code); + apiResult.setData(data); + apiResult.setMsg(msg); + return apiResult; + } +} diff --git a/src/main/java/com/spc/homework/XfXhApplication.java b/src/main/java/com/spc/homework/XfXhApplication.java new file mode 100644 index 0000000..1cbc4e3 --- /dev/null +++ b/src/main/java/com/spc/homework/XfXhApplication.java @@ -0,0 +1,11 @@ +package com.spc.homework; + +import org.springframework.boot.SpringApplication; +import org.springframework.boot.autoconfigure.SpringBootApplication; + +@SpringBootApplication +public class XfXhApplication { + public static void main(String[] args) { + SpringApplication.run(XfXhApplication.class); + } +} diff --git a/src/main/java/com/spc/homework/dto/MsgDTO.java b/src/main/java/com/spc/homework/dto/MsgDTO.java new file mode 100644 index 0000000..772b818 --- /dev/null +++ b/src/main/java/com/spc/homework/dto/MsgDTO.java @@ -0,0 +1,36 @@ +package com.spc.homework.dto; + +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/spc/homework/dto/request/LoginReq.java b/src/main/java/com/spc/homework/dto/request/LoginReq.java new file mode 100644 index 0000000..5286660 --- /dev/null +++ b/src/main/java/com/spc/homework/dto/request/LoginReq.java @@ -0,0 +1,11 @@ +package com.spc.homework.dto.request; + + +import lombok.Data; + +@Data +public class LoginReq { + private String username; + + private String password; +} diff --git a/src/main/java/com/spc/homework/dto/request/RequestDTO.java b/src/main/java/com/spc/homework/dto/request/RequestDTO.java new file mode 100644 index 0000000..3803f43 --- /dev/null +++ b/src/main/java/com/spc/homework/dto/request/RequestDTO.java @@ -0,0 +1,81 @@ +package com.spc.homework.dto.request; + +import com.alibaba.fastjson.annotation.JSONField; +import com.fasterxml.jackson.annotation.JsonProperty; +import com.spc.homework.dto.MsgDTO; +import lombok.AllArgsConstructor; +import lombok.Data; +import lombok.NoArgsConstructor; + +import java.util.List; + +@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/spc/homework/dto/response/ResponseDTO.java b/src/main/java/com/spc/homework/dto/response/ResponseDTO.java new file mode 100644 index 0000000..72b7bd9 --- /dev/null +++ b/src/main/java/com/spc/homework/dto/response/ResponseDTO.java @@ -0,0 +1,107 @@ +package com.spc.homework.dto.response; + +import com.fasterxml.jackson.annotation.JsonProperty; +import com.spc.homework.dto.MsgDTO; +import lombok.Data; +import lombok.NoArgsConstructor; + +import java.util.List; + +@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/spc/homework/lan/Independent.java b/src/main/java/com/spc/homework/lan/Independent.java new file mode 100644 index 0000000..6a123d7 --- /dev/null +++ b/src/main/java/com/spc/homework/lan/Independent.java @@ -0,0 +1,14 @@ +package com.spc.homework.lan; + +import java.lang.annotation.ElementType; +import java.lang.annotation.Retention; +import java.lang.annotation.RetentionPolicy; +import java.lang.annotation.Target; + + +@Target(ElementType.METHOD) +@Retention(RetentionPolicy.RUNTIME) +public @interface Independent { + + +} diff --git a/src/main/java/com/spc/homework/lan/IndependentAspect.java b/src/main/java/com/spc/homework/lan/IndependentAspect.java new file mode 100644 index 0000000..8cf8a37 --- /dev/null +++ b/src/main/java/com/spc/homework/lan/IndependentAspect.java @@ -0,0 +1,38 @@ +package com.spc.homework.lan; + +import org.aspectj.lang.ProceedingJoinPoint; +import org.aspectj.lang.annotation.Around; +import org.aspectj.lang.annotation.Aspect; +import org.springframework.stereotype.Component; + +import java.util.Map; +import java.util.concurrent.ConcurrentHashMap; + +@Aspect +@Component +public class IndependentAspect { + private Object result=null; + private Map resultMap=new ConcurrentHashMap<>(); + + @Around("@annotation(independent)") + public Object independent(ProceedingJoinPoint joinPoint, Independent independent) throws Throwable { + + String id = generateId(joinPoint.getArgs()); + if (resultMap.containsKey(id)){ + System.out.println("不要重复提交请求"); + return resultMap.get(id); + } + + result= joinPoint.proceed(); + resultMap.put(id,result); + return result; + } + + private String generateId(Object[] args){ + StringBuilder stringBuilder = new StringBuilder(); + for (Object arg : args) { + stringBuilder.append(arg.hashCode()); + } + return stringBuilder.toString(); + } +} diff --git a/src/main/java/com/spc/homework/mapper/LoginMapper.java b/src/main/java/com/spc/homework/mapper/LoginMapper.java new file mode 100644 index 0000000..ded2525 --- /dev/null +++ b/src/main/java/com/spc/homework/mapper/LoginMapper.java @@ -0,0 +1,10 @@ +package com.spc.homework.mapper; + +import com.spc.homework.dto.request.LoginReq; +import org.apache.ibatis.annotations.Mapper; + + +@Mapper +public interface LoginMapper { + LoginReq userLogin(LoginReq loginReq); +} diff --git a/src/main/java/com/spc/homework/service/LoginService.java b/src/main/java/com/spc/homework/service/LoginService.java new file mode 100644 index 0000000..0cf71e1 --- /dev/null +++ b/src/main/java/com/spc/homework/service/LoginService.java @@ -0,0 +1,8 @@ +package com.spc.homework.service; + +import com.spc.homework.Result; +import com.spc.homework.dto.request.LoginReq; + +public interface LoginService { + Result userLogin(LoginReq loginReq); +} diff --git a/src/main/java/com/spc/homework/service/LoginServiceImpl.java b/src/main/java/com/spc/homework/service/LoginServiceImpl.java new file mode 100644 index 0000000..c8c3d04 --- /dev/null +++ b/src/main/java/com/spc/homework/service/LoginServiceImpl.java @@ -0,0 +1,20 @@ +package com.spc.homework.service; + +import com.spc.homework.Result; +import com.spc.homework.dto.request.LoginReq; +import com.spc.homework.mapper.LoginMapper; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Service; + +@Service +public class LoginServiceImpl implements LoginService{ + @Autowired + private LoginMapper loginMapper; + + @Override + public Result userLogin(LoginReq loginReq) { + LoginReq loginReq1=loginMapper.userLogin(loginReq); + + return Result.success(loginReq1); + } +} diff --git a/src/main/java/com/spc/utli/DogGetApi.java b/src/main/java/com/spc/utli/DogGetApi.java new file mode 100644 index 0000000..e9b56b3 --- /dev/null +++ b/src/main/java/com/spc/utli/DogGetApi.java @@ -0,0 +1,114 @@ +package com.spc.utli; + +import com.alibaba.fastjson.JSONObject; +import org.apache.commons.codec.binary.Hex; + +import javax.crypto.Mac; +import javax.crypto.spec.SecretKeySpec; +import java.io.*; +import java.net.HttpURLConnection; +import java.net.URL; +import java.net.URLEncoder; +import java.nio.charset.StandardCharsets; +import java.security.InvalidKeyException; +import java.security.NoSuchAlgorithmException; +import java.util.Map; + +public class DogGetApi { + // 普通 API 请使用这个方法 + public static JSONObject dogeAPIGet(String apiPath, Map params) { + StringBuilder sb = new StringBuilder(); + for(Map.Entry hm : params.entrySet()){ + try { + sb.append(URLEncoder.encode(hm.getKey(), String.valueOf(StandardCharsets.UTF_8))).append('=').append(URLEncoder.encode(hm.getValue(), String.valueOf(StandardCharsets.UTF_8))).append("&"); + } catch (UnsupportedEncodingException e) { + throw new RuntimeException(e); + } + } + String bodyText = sb.toString().replace("&$", ""); + try { + return dogeAPIGet(apiPath, bodyText, false); + } catch (IOException e) { + e.printStackTrace(); + throw new RuntimeException(e.getMessage()); + } + } + + // 要求请求内容 Body 是一个 JSON 的 API，请使用这个方法 + public static JSONObject dogeAPIGet(String apiPath, JSONObject params) { + String bodyText = params.toString(); + try { + return dogeAPIGet(apiPath, bodyText, true); + } catch (IOException e) { + e.printStackTrace(); + throw new RuntimeException(e.getMessage()); + } + } + + // 无参数 API + public static JSONObject dogeAPIGet(String apiPath) { + try { + return dogeAPIGet(apiPath, "", true); + } catch (IOException e) { + e.printStackTrace(); + throw new RuntimeException(e.getMessage()); + } + } + + public static JSONObject dogeAPIGet(String apiPath, String paramsText, Boolean jsonMode) throws IOException{ + // 这里返回值类型是 JSONObject，你也可以根据你的具体情况使用不同的 JSON 库并修改最下方 JSON 处理代码 + + // 这里替换为你的多吉云永久 AccessKey 和 SecretKey，可在用户中心 - 密钥管理中查看 + // 请勿在客户端暴露 AccessKey 和 SecretKey，那样恶意用户将获得账号完全控制权 + String accessKey = "da22704e6e286726"; + String secretKey = "af0e5b9e236bed79e96ad5877f500fba"; + + String signStr = apiPath + "\n" + paramsText; + String sign = ""; + try { + Mac mac = Mac.getInstance("HmacSHA1"); + mac.init(new SecretKeySpec(secretKey.getBytes(), "HmacSHA1")); + sign = new String(new Hex().encode(mac.doFinal(signStr.getBytes())), StandardCharsets.UTF_8); // 这里 Hex 来自 org.apache.commons.codec.binary.Hex + } catch (NoSuchAlgorithmException e) { + e.printStackTrace(); + throw new RuntimeException(e.getMessage()); + } catch (InvalidKeyException e) { + e.printStackTrace(); + throw new RuntimeException(e.getMessage()); + } + String authorization = "TOKEN " + accessKey + ':' + sign; + + URL u = new URL("https://api.dogecloud.com" + apiPath); + HttpURLConnection conn = (HttpURLConnection) u.openConnection(); + conn.setDoOutput(true); + conn.setRequestMethod("POST"); + conn.setRequestProperty( "Content-Type", jsonMode ? "application/json" : "application/x-www-form-urlencoded"); + conn.setRequestProperty( "Authorization", authorization); + conn.setRequestProperty( "Content-Length", String.valueOf(paramsText.length())); + OutputStream os = conn.getOutputStream(); + os.write(paramsText.getBytes()); + os.flush(); + os.close(); + StringBuilder retJSON = new StringBuilder(); + if(conn.getResponseCode() == HttpURLConnection.HTTP_OK){ + String readLine = ""; + try (BufferedReader responseReader=new BufferedReader(new InputStreamReader(conn.getInputStream(), StandardCharsets.UTF_8))) { + while((readLine=responseReader.readLine())!=null){ + retJSON.append(readLine).append("\n"); + } + responseReader.close(); + } + JSONObject ret = new JSONObject(Boolean.parseBoolean(retJSON.toString())); + if (ret.getInt("code") != 200) { + System.err.println("{\"error\":\"API 返回错误：" + ret.getString("msg") + "\"}"); + } else { + JSONObject output = new JSONObject(); + JSONObject data = ret.getJSONObject("data"); + return data; + } + } else { + System.err.println("{\"error\":\"网络错误：" + conn.getResponseCode() + "\"}"); + } + return null; + } +} diff --git a/src/main/java/com/spc/utli/DuoJiYunUtil.java b/src/main/java/com/spc/utli/DuoJiYunUtil.java new file mode 100644 index 0000000..9171215 --- /dev/null +++ b/src/main/java/com/spc/utli/DuoJiYunUtil.java @@ -0,0 +1,20 @@ +package com.spc.utli; + +import com.alibaba.fastjson.JSONObject; + +import static com.spc.utli.DogGetApi.dogeAPIGet; + +public class DuoJiYunUtil { + // 这段代码本质上就是调用多吉云 API 获取临时密钥，你也可以自己实现： +// 该 API 参考文档： https://docs.dogecloud.com/oss/api-tmp-token + + public static JSONObject getCredentials() { + JSONObject body = new JSONObject(); + body.put("channel", "OSS_FULL"); + body.append("scopes", "muyu"); // 要操作的存储空间名称，注意不是 s3Bucket 值，也可以设置为 * + + JSONObject data = dogeAPIGet("/auth/tmp_token.json", body); + JSONObject credentials = data.getJSONObject("Credentials"); + return credentials; + } +} diff --git a/src/main/java/com/spc/utli/S3.java b/src/main/java/com/spc/utli/S3.java new file mode 100644 index 0000000..8d5a6ae --- /dev/null +++ b/src/main/java/com/spc/utli/S3.java @@ -0,0 +1,41 @@ +package com.spc.utli; + +import com.alibaba.fastjson.JSONArray; +import com.alibaba.fastjson.JSONObject; +import com.amazonaws.auth.AWSStaticCredentialsProvider; +import com.amazonaws.auth.BasicSessionCredentials; +import com.amazonaws.client.builder.AwsClientBuilder; +import com.amazonaws.services.s3.AmazonS3; +import com.amazonaws.services.s3.AmazonS3ClientBuilder; +import com.amazonaws.services.s3.model.PutObjectResult; + +import static com.spc.utli.DogGetApi.dogeAPIGet; + +public class S3 { + public static AmazonS3 initS3(JSONObject credentials){ + BasicSessionCredentials awsCredentials = new BasicSessionCredentials(credentials.getString("accessKeyId"), credentials.getString("secretAccessKey"), credentials.getString("sessionToken")); + AmazonS3 s3 = AmazonS3ClientBuilder.standard() + .withCredentials(new AWSStaticCredentialsProvider(awsCredentials)) + .withEndpointConfiguration(new AwsClientBuilder.EndpointConfiguration( + "https://cos.ap-shanghai.myqcloud.com", // 修改为多吉云控制台存储空间 SDK 参数中的 s3Endpoint + "automatic")) + .build(); + return s3; + } + + public static void main(String[] args) throws Exception { + JSONObject data = dogeAPIGet("/oss/bucket/list.json"); + JSONArray buckets = data.getJSONArray("buckets"); + for (int i = 0 ; i < buckets.length(); i++) { + System.out.println(buckets.getJSONObject(i).toString()); + } + + JSONObject credentials = DuoJiYunUtil.getCredentials(); + AmazonS3 s3 = initS3(credentials); + PutObjectResult putObjectResult = s3.putObject("s-sh-7714-muyu-1258813047", "lihao.png", new File("D:\\img\\4~NIBGJKOVQ}7HA$0J~9REW.png")); + String string = putObjectResult.toString(); + System.out.println(string); + + } + +} diff --git a/src/main/resources/bootstrap.yml b/src/main/resources/bootstrap.yml new file mode 100644 index 0000000..11d0fcb --- /dev/null +++ b/src/main/resources/bootstrap.yml @@ -0,0 +1,33 @@ +xfxh: + # 服务引擎使用 讯飞星火认知大模型V2.0，如果使用 V1.5 需要将 hostUrl 修改为 https://spark-api.xf-yun.com/v1.1/chat + hostUrl: https://spark-api.xf-yun.com/v2.1/chat + # 发送请求时指定的访问领域，如果是 V1.5版本 设置为 general，如果是 V2版本 设置为 generalv2 + domain: generalv2 + # 核采样阈值。用于决定结果随机性，取值越高随机性越强即相同的问题得到的不同答案的可能性越高。取值 [0,1] + temperature: 0.5 + # 模型回答的tokens的最大长度，V1.5取值为[1,4096]，V2.0取值为[1,8192]。 + maxTokens: 2048 + # 大模型回复问题的最大响应时长，单位 s + maxResponseTime: 30 + # 允许同时连接大模型的 websocket 数，如果是普通（免费）用户为 2，超过这个数连接响应会报错，具体参考官网。 + QPS: 2 + # 用于权限验证，从服务接口认证信息中获取 + appId: 4c5ea24c + # 用于权限验证，从服务接口认证信息中获取 + apiKey: d53d6f5aae8cfee01420eeff6584057f + # 用于权限验证，从服务接口认证信息中获取 + apiSecret: Mzg3YTYxZjYyYTE0MTdiOTMyZTdlMTFi +server: + port: 8080 +spring: + datasource: + driver-class-name: com.mysql.jdbc.Driver + url: jdbc:mysql://localhost:3306/chat?useUnicode=true&characterEncoding=utf8 + username: root + password: root + name: defaultDataSource +mybatis: + mapper-locations: mappers/*xml + mybatis: + configuration: + map-underscore-to-camel-case: true diff --git a/src/main/resources/mappers/user.xml b/src/main/resources/mappers/user.xml new file mode 100644 index 0000000..e4113ce --- /dev/null +++ b/src/main/resources/mappers/user.xml @@ -0,0 +1,9 @@ + + + + + + +