|
@@ -1,5 +1,7 @@
|
|
|
package com.uas.platform.b2b.controller;
|
|
package com.uas.platform.b2b.controller;
|
|
|
|
|
|
|
|
|
|
+import javax.servlet.http.HttpServletRequest;
|
|
|
|
|
+import javax.servlet.http.HttpServletResponse;
|
|
|
import javax.servlet.http.HttpSession;
|
|
import javax.servlet.http.HttpSession;
|
|
|
|
|
|
|
|
import org.springframework.http.HttpStatus;
|
|
import org.springframework.http.HttpStatus;
|
|
@@ -13,6 +15,9 @@ import org.springframework.web.bind.annotation.RestController;
|
|
|
import com.uas.platform.b2b.model.User;
|
|
import com.uas.platform.b2b.model.User;
|
|
|
import com.uas.platform.b2b.model.UserInfo;
|
|
import com.uas.platform.b2b.model.UserInfo;
|
|
|
import com.uas.platform.b2b.support.SystemSession;
|
|
import com.uas.platform.b2b.support.SystemSession;
|
|
|
|
|
+import com.uas.platform.core.util.serializer.FlexJsonUtils;
|
|
|
|
|
+import com.uas.sso.SSOHelper;
|
|
|
|
|
+import com.uas.sso.SSOToken;
|
|
|
|
|
|
|
|
@RestController
|
|
@RestController
|
|
|
@RequestMapping("/authentication")
|
|
@RequestMapping("/authentication")
|
|
@@ -35,9 +40,19 @@ public class AuthenticationController {
|
|
|
*/
|
|
*/
|
|
|
@RequestMapping(value = "/{enUU}")
|
|
@RequestMapping(value = "/{enUU}")
|
|
|
@ResponseBody
|
|
@ResponseBody
|
|
|
- public void switchEnterprise(@PathVariable("enUU") long enUU) {
|
|
|
|
|
|
|
+ public void switchEnterprise(HttpServletRequest request, HttpServletResponse response, @PathVariable("enUU") long enUU) {
|
|
|
User user = SystemSession.getUser();
|
|
User user = SystemSession.getUser();
|
|
|
user.setCurrentEnterprise(enUU);
|
|
user.setCurrentEnterprise(enUU);
|
|
|
|
|
+ // 通知到账户中心
|
|
|
|
|
+ SSOToken token = SSOHelper.attrToken(request);
|
|
|
|
|
+ if (null != token) {
|
|
|
|
|
+ com.uas.account.entity.User tokenUser = FlexJsonUtils.fromJson(token.getData(), com.uas.account.entity.User.class);
|
|
|
|
|
+ tokenUser.setSpaceDialectUID(String.valueOf(enUU));
|
|
|
|
|
+ tokenUser.setId(null);
|
|
|
|
|
+ SSOToken newToken = new SSOToken(request, tokenUser.getUid());
|
|
|
|
|
+ newToken.setData(FlexJsonUtils.toJson(tokenUser));
|
|
|
|
|
+ SSOHelper.setSSOCookie(request, response, newToken, true);
|
|
|
|
|
+ }
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
}
|
|
}
|