|
@@ -2,6 +2,8 @@ package cn.com.qmth.stmms.ms.accesscontrol;
|
|
|
|
|
|
import cn.com.qmth.stmms.ms.accesscontrol.config.LoginConfig;
|
|
|
import cn.com.qmth.stmms.ms.core.domain.user.AbstractUser;
|
|
|
+import cn.com.qmth.stmms.ms.core.domain.user.MarkUser;
|
|
|
+import cn.com.qmth.stmms.ms.core.domain.user.Role;
|
|
|
import cn.com.qmth.stmms.ms.core.repository.MarkUserRepo;
|
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
|
import org.springframework.http.HttpStatus;
|
|
@@ -9,6 +11,7 @@ import org.springframework.web.servlet.handler.HandlerInterceptorAdapter;
|
|
|
|
|
|
import javax.servlet.http.HttpServletRequest;
|
|
|
import javax.servlet.http.HttpServletResponse;
|
|
|
+import java.util.Objects;
|
|
|
|
|
|
public class LoginInterceptor extends HandlerInterceptorAdapter {
|
|
|
|
|
@@ -27,11 +30,16 @@ public class LoginInterceptor extends HandlerInterceptorAdapter {
|
|
|
if (loginConfig.adminLoginConfig().getLoginName().equals(loginName)
|
|
|
|| loginConfig.inspectionLoginConfig().getLoginName().equals(loginName)) {
|
|
|
return true;
|
|
|
- } else if (loginConfig.clientLoginConfig().getLoginName().equals(loginName)) {
|
|
|
+ }
|
|
|
+// else if (loginConfig.clientLoginConfig().getLoginName().equals(loginName)) {
|
|
|
+// response.sendError(HttpStatus.UNAUTHORIZED.value(), "该账户只能在采集端登录");
|
|
|
+// return false;
|
|
|
+// }
|
|
|
+ AbstractUser user = markUserRepo.findByLoginName(loginName);
|
|
|
+ if (Objects.equals(Role.COLLECTOR, ((MarkUser) user).getRole())) {
|
|
|
response.sendError(HttpStatus.UNAUTHORIZED.value(), "该账户只能在采集端登录");
|
|
|
return false;
|
|
|
}
|
|
|
- AbstractUser user = markUserRepo.findByLoginName(loginName);
|
|
|
String sessionId = request.getSession().getId();
|
|
|
if (user.getSessionId().equalsIgnoreCase(sessionId)) {
|
|
|
return true;
|
|
@@ -46,6 +54,7 @@ public class LoginInterceptor extends HandlerInterceptorAdapter {
|
|
|
|
|
|
}
|
|
|
|
|
|
+ @Override
|
|
|
public void afterCompletion(HttpServletRequest request,
|
|
|
HttpServletResponse response, Object handler, Exception ex)
|
|
|
throws Exception {
|