From 5196e467db4fe892009bdad0559df1e13c0c882c Mon Sep 17 00:00:00 2001 From: Stormeye Wu Date: Fri, 1 Mar 2019 16:25:13 +0800 Subject: [PATCH] =?UTF-8?q?[=E5=BE=AE=E4=BF=A1=E7=AC=AC=E4=B8=89=E6=96=B9?= =?UTF-8?q?=E7=99=BB=E5=BD=95][=E4=BF=AE=E6=94=B9]:=E7=99=BB=E5=BD=95cooki?= =?UTF-8?q?e=EF=BC=8C=E5=90=8E=E7=AB=AF=E8=AE=BE=E7=BD=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../iformall/controller/HomeController.java | 21 ++++++++++++++++--- .../controller/WechatLoginController.java | 2 +- 2 files changed, 19 insertions(+), 4 deletions(-) diff --git a/mallinkAdmin/src/main/java/com/iformall/controller/HomeController.java b/mallinkAdmin/src/main/java/com/iformall/controller/HomeController.java index 93067b4b8..ab898b3cd 100644 --- a/mallinkAdmin/src/main/java/com/iformall/controller/HomeController.java +++ b/mallinkAdmin/src/main/java/com/iformall/controller/HomeController.java @@ -31,6 +31,7 @@ import org.springframework.web.bind.annotation.*; import javax.imageio.ImageIO; import javax.servlet.ServletException; import javax.servlet.ServletOutputStream; +import javax.servlet.http.Cookie; import javax.servlet.http.HttpServletResponse; import java.awt.image.BufferedImage; import java.io.IOException; @@ -81,7 +82,7 @@ public class HomeController extends BaseController { @ApiOperation("登录") @PostMapping("/doLogin") - public ResultData login(@RequestBody MallUserInfo user) { + public ResultData login(@RequestBody MallUserInfo user, HttpServletResponse response) { logger.debug("[" + getIpAddr() + "] HomeController::login"); try { String kaptcha = ShiroUtils.getKaptcha(Constants.KAPTCHA_SESSION_KEY); @@ -114,6 +115,11 @@ public class HomeController extends BaseController { if(menus != null) { info.setMenus(menus); } + + Cookie unameCookie = new Cookie("uname", info.getUsername()); + unameCookie.setMaxAge(60); + response.addCookie(unameCookie); + data.data = info; } catch (Exception e) { return new ResultData(ErrorCode.USER_PASSWD_ERR); @@ -130,7 +136,7 @@ public class HomeController extends BaseController { @ApiOperation("B端登录") @PostMapping("/bHidLogin") - public ResultData bLogin(@RequestBody MallUserInfo user) { + public ResultData bLogin(@RequestBody MallUserInfo user, HttpServletResponse response) { logger.debug("[" + getIpAddr() + "] HomeController::bHidLogin"); ResultData data = new ResultData(); // tenantId is appId @@ -177,6 +183,10 @@ public class HomeController extends BaseController { if(menus != null) { info.setMenus(menus); } + Cookie unameCookie = new Cookie("uname", info.getUsername()); + unameCookie.setMaxAge(60); + response.addCookie(unameCookie); + data.data = info; } catch (Exception e) { return new ResultData(ErrorCode.USER_PASSWD_ERR); @@ -210,7 +220,7 @@ public class HomeController extends BaseController { @ApiOperation(value = "手机验证码登录", notes = "{\"phone\",\"string\",\"code\",\"string\"}") @PostMapping("/doLoginByPhone") - public ResultData doLoginByPhone(@RequestBody Map params) { + public ResultData doLoginByPhone(@RequestBody Map params, HttpServletResponse response) { logger.debug("[" + getIpAddr() + "] HomeController::doLoginByPhone"); // String phone,String code,String pwd String phone = params.get("phone"); @@ -247,6 +257,11 @@ public class HomeController extends BaseController { if(menus != null) { info.setMenus(menus); } + + Cookie unameCookie = new Cookie("uname", info.getUsername()); + unameCookie.setMaxAge(60); + response.addCookie(unameCookie); + return new ResultData(info); } catch (Exception e) { return new ResultData(ErrorCode.USER_PASSWD_ERR); diff --git a/mallinkAdmin/src/main/java/com/iformall/controller/WechatLoginController.java b/mallinkAdmin/src/main/java/com/iformall/controller/WechatLoginController.java index 8d25fd003..e7bc75908 100644 --- a/mallinkAdmin/src/main/java/com/iformall/controller/WechatLoginController.java +++ b/mallinkAdmin/src/main/java/com/iformall/controller/WechatLoginController.java @@ -108,7 +108,7 @@ public class WechatLoginController extends BaseController { } Cookie unameCookie = new Cookie("uname", info.getUsername()); unameCookie.setMaxAge(60); - response.addCookie(openIdCookie); + response.addCookie(unameCookie); response.sendRedirect("https://" + host + "/#/layout"); } catch (Exception e) {