|
@@ -1,20 +1,19 @@
|
|
|
package com.ym.mec.auth.core.handler;
|
|
|
|
|
|
-import java.io.IOException;
|
|
|
-import java.util.Base64;
|
|
|
-import java.util.Date;
|
|
|
-import java.util.HashMap;
|
|
|
-import java.util.Map;
|
|
|
-
|
|
|
-import javax.servlet.http.HttpServletRequest;
|
|
|
-import javax.servlet.http.HttpServletResponse;
|
|
|
-
|
|
|
+import com.fasterxml.jackson.databind.ObjectMapper;
|
|
|
+import com.ym.mec.auth.api.entity.SysUser;
|
|
|
+import com.ym.mec.auth.api.entity.SysUserLogin;
|
|
|
+import com.ym.mec.auth.api.entity.SysUserLoginLog;
|
|
|
+import com.ym.mec.auth.api.enums.SysUserType;
|
|
|
+import com.ym.mec.auth.service.SysUserLoginLogService;
|
|
|
+import com.ym.mec.auth.service.SysUserLoginService;
|
|
|
+import com.ym.mec.auth.service.SysUserService;
|
|
|
+import com.ym.mec.common.entity.HttpResponseResult;
|
|
|
import com.ym.mec.common.entity.ImResult;
|
|
|
import com.ym.mec.common.entity.ImUserModel;
|
|
|
+import com.ym.mec.common.security.SecurityConstants;
|
|
|
import com.ym.mec.im.ImFeignService;
|
|
|
import com.ym.mec.web.WebFeignService;
|
|
|
-
|
|
|
-import org.apache.commons.collections.MapUtils;
|
|
|
import org.apache.commons.lang3.StringUtils;
|
|
|
import org.slf4j.Logger;
|
|
|
import org.slf4j.LoggerFactory;
|
|
@@ -36,15 +35,15 @@ import org.springframework.security.oauth2.provider.token.AuthorizationServerTok
|
|
|
import org.springframework.security.web.authentication.SavedRequestAwareAuthenticationSuccessHandler;
|
|
|
import org.springframework.stereotype.Component;
|
|
|
|
|
|
-import com.fasterxml.jackson.databind.ObjectMapper;
|
|
|
-import com.ym.mec.auth.api.entity.SysUser;
|
|
|
-import com.ym.mec.auth.api.entity.SysUserLogin;
|
|
|
-import com.ym.mec.auth.api.entity.SysUserLoginLog;
|
|
|
-import com.ym.mec.auth.service.SysUserLoginLogService;
|
|
|
-import com.ym.mec.auth.service.SysUserLoginService;
|
|
|
-import com.ym.mec.auth.service.SysUserService;
|
|
|
-import com.ym.mec.common.entity.HttpResponseResult;
|
|
|
-import com.ym.mec.common.security.SecurityConstants;
|
|
|
+import javax.servlet.http.HttpServletRequest;
|
|
|
+import javax.servlet.http.HttpServletResponse;
|
|
|
+import java.io.IOException;
|
|
|
+import java.util.Arrays;
|
|
|
+import java.util.Base64;
|
|
|
+import java.util.Date;
|
|
|
+import java.util.HashMap;
|
|
|
+import java.util.Map;
|
|
|
+import java.util.stream.Collectors;
|
|
|
|
|
|
@Component
|
|
|
public class BaseAuthenticationSuccessEventHandler extends SavedRequestAwareAuthenticationSuccessHandler {
|
|
@@ -156,6 +155,16 @@ public class BaseAuthenticationSuccessEventHandler extends SavedRequestAwareAuth
|
|
|
map.put("tenantId", tenantId);
|
|
|
map.put("companyId", sysUser.getTenantId());
|
|
|
|
|
|
+ // 设置登录管户端类型
|
|
|
+ if (clientId.toUpperCase().equals(SysUserType.EDUCATION.getCode())) {
|
|
|
+
|
|
|
+ String collect = Arrays.stream(sysUser.getUserType().split(","))
|
|
|
+ .filter(x -> x.equals(SysUserType.SYSTEM.getCode()) || x.equals(SysUserType.SCHOOL.getCode()))
|
|
|
+ .collect(Collectors.joining(","));
|
|
|
+ // 登录客户端类型
|
|
|
+ map.put("clientType", collect);
|
|
|
+ }
|
|
|
+
|
|
|
response.setContentType("application/json; charset=utf-8");
|
|
|
HttpResponseResult result = new HttpResponseResult(true, HttpStatus.OK.value(), map, "");
|
|
|
response.getWriter().write(objectMapper.writeValueAsString(result));
|