Jelajahi Sumber

切换账套cookies信息中增加营业执照号以及企业名称

liusw 7 tahun lalu
induk
melakukan
d8fdfa2892

+ 9 - 3
src/main/java/com/uas/platform/b2c/common/account/controller/AuthenticationController.java

@@ -1,5 +1,7 @@
 package com.uas.platform.b2c.common.account.controller;
 
+import com.uas.platform.b2c.common.account.dao.EnterpriseDao;
+import com.uas.platform.b2c.common.account.model.Enterprise;
 import com.uas.platform.b2c.common.account.model.User;
 import com.uas.platform.b2c.common.account.model.UserInfo;
 import com.uas.platform.b2c.common.account.service.RoleService;
@@ -41,6 +43,8 @@ public class AuthenticationController {
 	@Autowired
 	private RoleService roleService;
 
+	@Autowired
+	private EnterpriseDao enterpriseDao;
 
 	@RequestMapping(method = RequestMethod.GET, headers = "Accept=application/json")
 	@ResponseBody
@@ -101,17 +105,19 @@ public class AuthenticationController {
 	@ResponseBody
 	public void switchEnterprise(@PathVariable("enUU") long enUU, HttpServletRequest request, HttpServletResponse response) {
 		userService.saveLoginTime(enUU);
-		User user = userService.switchEnterprise(enUU);
-		request.getSession().setAttribute("user", user);
-
 		SSOToken token = (SSOToken)SSOHelper.getToken(request);
 		UserAccount tokenUser = FastjsonUtils.fromJson(token.getData(), UserAccount.class);
+		User user = userService.switchEnterprise(enUU);
+		request.getSession().setAttribute("user", user);
 		if (enUU == 0L) {
 			tokenUser.setSpaceUU(null);
 			tokenUser.setBusinessCode(null);
 			tokenUser.setSpaceName(null);
 		} else {
+			Enterprise enterprise = enterpriseDao.findByUu(enUU);
 			tokenUser.setSpaceUU(enUU);
+			tokenUser.setBusinessCode(enterprise.getEnBussinessCode());
+			tokenUser.setSpaceName(enterprise.getEnName());
 		}
 		token.setData(FastjsonUtils.toJson(tokenUser));
 		// 设置cookie