top.cutexingluo.tools.security.exception.XTAuthenticationEntryPoint Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of xingtools-pkg-jdk8 Show documentation

xingtools 依赖core包，jdk 分类包，对不同的jdk版本提供兼容性

The newest version!

package top.cutexingluo.tools.security.exception;

import org.springframework.security.core.AuthenticationException;
import org.springframework.security.web.AuthenticationEntryPoint;
import top.cutexingluo.tools.common.Constants;
import top.cutexingluo.tools.common.Result;
import top.cutexingluo.tools.common.base.IResult;
import top.cutexingluo.tools.utils.ee.web.front.WebUtils;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;

/**
 * XT 认证异常拦截默认封装为 Result
 * 

 * SpringSecurity 认证端点 
 * 需要导入 spring-security
 *
 * @author XingTian
 * @version 1.0.1
 * @date 2023/6/17 18:58
 * @updateFrom 1.0.3
 */
public class XTAuthenticationEntryPoint implements AuthenticationEntryPoint {

    /**
     * returnResultSource 包装类
     * 自定义返回值
     *
     * @since 1.0.3
     */
    IResult returnResult;

    public XTAuthenticationEntryPoint(IResult returnResultSource) {
        this.returnResult = returnResultSource;
    }

    /**
     * @param httpServletRequest  http servlet请求
     * @param httpServletResponse http servlet响应
     * @param e                   异常
     * @throws IOException      ioexception
     * @throws ServletException servlet异常
     */
    @Override
    public void commence(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, AuthenticationException e) throws IOException, ServletException {
        // 处理认证异常并封装为合适的响应格式返回给前端
        unauthorized(httpServletResponse, e.getMessage());
    }

    /**
     * update the static off
     *
     * @since 1.0.3
     */
    public  void unauthorized(HttpServletResponse httpServletResponse, String msg) throws IOException {
        IResult error = null;
        if (returnResult == null) {
            error = (IResult) Result.errorBy(Constants.CODE_401, msg);
        } else {
            error = (IResult) returnResult;
        }
        WebUtils.response(httpServletResponse, error, HttpServletResponse.SC_UNAUTHORIZED);
    }


}