org.apache.shiro.biz.authc.DefaultAuthenticationSuccessHandler Maven / Gradle / Ivy
package org.apache.shiro.biz.authc;
import com.alibaba.fastjson2.JSON;
import org.apache.shiro.authc.AuthenticationException;
import org.apache.shiro.authc.AuthenticationToken;
import org.apache.shiro.authc.UsernamePasswordToken;
import org.apache.shiro.biz.ShiroBizMessageSource;
import org.apache.shiro.biz.authc.token.DefaultAuthenticationToken;
import org.apache.shiro.biz.utils.SubjectUtils;
import org.apache.shiro.biz.utils.WebUtils;
import org.apache.shiro.biz.web.servlet.http.HttpStatus;
import org.apache.shiro.subject.Subject;
import org.springframework.context.NoSuchMessageException;
import org.springframework.context.support.MessageSourceAccessor;
import org.springframework.http.MediaType;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
/**
* 认证请求成功后的处理实现
*/
public class DefaultAuthenticationSuccessHandler implements AuthenticationSuccessHandler {
protected MessageSourceAccessor messages = ShiroBizMessageSource.getAccessor();
@Override
public boolean supports(AuthenticationToken token) {
return SubjectUtils.isAssignableFrom(token.getClass(), UsernamePasswordToken.class,
DefaultAuthenticationToken.class);
}
@Override
public void onAuthenticationSuccess(AuthenticationToken token, ServletRequest request, ServletResponse response,
Subject subject) {
try {
//HttpServletRequest httpRequest = WebUtils.toHttp(request);
HttpServletResponse httpResponse = WebUtils.toHttp(response);
httpResponse.setStatus(HttpStatus.SC_OK);
httpResponse.setContentType(MediaType.APPLICATION_JSON_VALUE);
httpResponse.setCharacterEncoding(StandardCharsets.UTF_8.toString());
// Response Authentication status information
JSON.writeTo(response.getOutputStream(), AuthcResponse.success(messages.getMessage(AuthcResponseCode.SC_AUTHC_SUCCESS.getMsgKey())));
} catch (NoSuchMessageException e) {
throw new AuthenticationException(e);
} catch (IOException e) {
throw new AuthenticationException(e);
}
}
@Override
public int getOrder() {
return Integer.MAX_VALUE;
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy