From f7a0bea019f495409a2fdc23a06656fdaed99487 Mon Sep 17 00:00:00 2001
From: zhangqinbin <181961702@qq.com>
Date: Wed, 15 Nov 2023 16:39:57 +0800
Subject: [PATCH] =?UTF-8?q?wps=E4=BF=9D=E5=AD=98=E5=9B=9E=E8=B0=83?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../ebtp/extend/wps/controller/WpsController.java     | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/src/main/java/com/chinaunicom/mall/ebtp/extend/wps/controller/WpsController.java b/src/main/java/com/chinaunicom/mall/ebtp/extend/wps/controller/WpsController.java
index 305f83d..83ef2bb 100644
--- a/src/main/java/com/chinaunicom/mall/ebtp/extend/wps/controller/WpsController.java
+++ b/src/main/java/com/chinaunicom/mall/ebtp/extend/wps/controller/WpsController.java
@@ -31,6 +31,7 @@ import org.springframework.web.context.request.RequestContextHolder;
 import org.springframework.web.context.request.ServletRequestAttributes;
 
 import javax.annotation.Resource;
+import javax.servlet.http.Cookie;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 import java.io.*;
@@ -135,7 +136,8 @@ public class WpsController {
 	public void getWpsUrlByGet(HttpServletRequest request, HttpServletResponse response,
 											   @RequestParam("fileId") String fileId,
 											   @RequestParam("fileName") String fileName,
-											   @RequestParam("userToken") String userToken
+											   @RequestParam("userToken") String userToken,
+							                   @RequestHeader(name = "Authorization") String token
 											   ) {
 		WpsClientInVo in = new WpsClientInVo();
 		in.setFileId(fileId);
@@ -145,8 +147,13 @@ public class WpsController {
 		try {
 			response.setContentType("text/html;charset=utf-8");
 			PrintWriter out = response.getWriter();
+			response.setHeader("Cookie","mall3_token="+token);
+			Cookie theC = new Cookie("mall3_token",token);
+			theC.setMaxAge(-1);
+			theC.setPath("/");
+			response.addCookie(theC);
 			out.println(str);
-			;
+
 		}catch (Exception e){
 			e.printStackTrace();
 		}