将前端传入的token改成后端获取当前用户的token
This commit is contained in:
@@ -117,11 +117,11 @@ public class AiChatController extends BaseController {
|
|||||||
|
|
||||||
// 在主线程中获取当前用户名,避免在异步线程中获取安全上下文
|
// 在主线程中获取当前用户名,避免在异步线程中获取安全上下文
|
||||||
String currentUsername = SecurityUtils.getLoginUser().getUsername();
|
String currentUsername = SecurityUtils.getLoginUser().getUsername();
|
||||||
|
String currentUserToken = SecurityUtils.getLoginUser().getToken();
|
||||||
// 异步执行请求处理,避免阻塞主线程
|
// 异步执行请求处理,避免阻塞主线程
|
||||||
CompletableFuture.runAsync(() -> {
|
CompletableFuture.runAsync(() -> {
|
||||||
try {
|
try {
|
||||||
sendToDifyAndStream(requestData, emitter, currentUsername);
|
sendToDifyAndStream(requestData, emitter, currentUsername, currentUserToken);
|
||||||
} catch (Exception e) {
|
} catch (Exception e) {
|
||||||
e.printStackTrace();
|
e.printStackTrace();
|
||||||
try {
|
try {
|
||||||
@@ -147,7 +147,7 @@ public class AiChatController extends BaseController {
|
|||||||
* @param currentUsername
|
* @param currentUsername
|
||||||
* @throws IOException 当网络请求或IO操作失败时抛出
|
* @throws IOException 当网络请求或IO操作失败时抛出
|
||||||
*/
|
*/
|
||||||
private void sendToDifyAndStream(Map<String, Object> requestData, SseEmitter emitter, String currentUsername)
|
private void sendToDifyAndStream(Map<String, Object> requestData, SseEmitter emitter, String currentUsername,String currentUserToken)
|
||||||
throws IOException {
|
throws IOException {
|
||||||
|
|
||||||
// 构建请求体参数
|
// 构建请求体参数
|
||||||
@@ -184,9 +184,9 @@ public class AiChatController extends BaseController {
|
|||||||
inputs.put("user_name", userName);
|
inputs.put("user_name", userName);
|
||||||
}
|
}
|
||||||
|
|
||||||
Object userToken = requestData.get("user_token");
|
//Object userToken = requestData.get("user_token");
|
||||||
if (userToken != null) {
|
if (currentUserToken != null) {
|
||||||
inputs.put("user_token", userToken);
|
inputs.put("user_token", currentUserToken);
|
||||||
}
|
}
|
||||||
|
|
||||||
Object userRole = requestData.get("user_role");
|
Object userRole = requestData.get("user_role");
|
||||||
|
|||||||
Reference in New Issue
Block a user