forgetest5
/
ifffff
mirror of https://gitee.com/maxjhandsome/pig
1
0
Fork 0
Code Issues Releases Wiki Activity

fixed： 异常处理bug

This commit is contained in:
冷冷 2018-02-23 09:13:42 +08:00
parent 742a162f98
commit d071985eca
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pig-auth/src/main/java/com/github/pig/auth/component/PigWebResponseExceptionTranslator.java
View File

@ -5,6 +5,7 @@ import org.springframework.http.ResponseEntity;
import org.springframework.security.authentication.InternalAuthenticationServiceException; import org.springframework.security.authentication.InternalAuthenticationServiceException;
import org.springframework.security.oauth2.common.exceptions.InvalidGrantException; import org.springframework.security.oauth2.common.exceptions.InvalidGrantException;
import org.springframework.security.oauth2.common.exceptions.OAuth2Exception; import org.springframework.security.oauth2.common.exceptions.OAuth2Exception;
import org.springframework.security.oauth2.common.exceptions.UnsupportedResponseTypeException;
import org.springframework.security.oauth2.provider.error.DefaultWebResponseExceptionTranslator; import org.springframework.security.oauth2.provider.error.DefaultWebResponseExceptionTranslator;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
@ -25,6 +26,8 @@ import org.springframework.stereotype.Component;
@Component @Component
public class PigWebResponseExceptionTranslator extends DefaultWebResponseExceptionTranslator { public class PigWebResponseExceptionTranslator extends DefaultWebResponseExceptionTranslator {
public static final String BAD_MSG = "坏的凭证";
/** /**
* @param e spring security内部异常 * @param e spring security内部异常
* @return 经过处理的异常信息 * @return 经过处理的异常信息
@ -33,12 +36,12 @@ public class PigWebResponseExceptionTranslator extends DefaultWebResponseExcepti
@Override @Override
public ResponseEntity<OAuth2Exception> translate(Exception e) throws Exception { public ResponseEntity<OAuth2Exception> translate(Exception e) throws Exception {
OAuth2Exception oAuth2Exception; OAuth2Exception oAuth2Exception;
if (e instanceof InvalidGrantException && StringUtils.equals("坏的凭证",e.getMessage())) { if (e instanceof InvalidGrantException && StringUtils.equals(BAD_MSG, e.getMessage())) {
oAuth2Exception = new InvalidGrantException("密码错误", e); oAuth2Exception = new InvalidGrantException("密码错误", e);
} else if (e instanceof InternalAuthenticationServiceException) { } else if (e instanceof InternalAuthenticationServiceException) {
oAuth2Exception = new InvalidGrantException("用户名不存在", e); oAuth2Exception = new InvalidGrantException("用户名不存在", e);
} else { } else {
oAuth2Exception = (OAuth2Exception) e; oAuth2Exception = new UnsupportedResponseTypeException("服务处理异常", e);
} }
return super.translate(oAuth2Exception); return super.translate(oAuth2Exception);
} }