|
|
@@ -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
|