Spring Boot集成Spring Security系列七之接口异常处理这是Spring Boot集成Spring

package com.example.springsecurity.common.exception;


import com.example.springsecurity.common.api.IErrorCode;

/**
 * 自定义API异常
 * Created by Lu Fei on 2022/04/20.
 */
public class ApiException extends RuntimeException {
    private String errorCode;
    private String errorMessage;

    public ApiException(IErrorCode errorCode) {
        this.errorCode = errorCode.getCode();
        this.errorMessage = errorCode.getMessage();
    }

    public ApiException(String message) {
        super(message);
    }

    public ApiException(Throwable cause) {
        super(cause);
    }

    public ApiException(String message, Throwable cause) {
        super(message, cause);
    }

    public String getErrorCode() {
        return errorCode;
    }

    public String getErrorMessage() {
        return errorMessage;
    }
}

package com.example.springsecurity.common.exception;

import com.example.springsecurity.common.api.CommonResult;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;

@RestControllerAdvice
public class GlobalExceptionHandler {

    @ExceptionHandler(value = ApiException.class)
    public CommonResult handle(ApiException e) {
        if(e.getErrorCode() != null) {
            return new CommonResult(e.getErrorCode(), e.getErrorMessage());
        }

        return CommonResult.failed();
    }

}

修改UserServiceImpl

定义接口参数断言工具类

package com.example.springsecurity.common.exception;

import com.example.springsecurity.common.api.IErrorCode;
import org.springframework.util.StringUtils;

public class ApiExceptionAsserts {
    public static void hasText(String text, IErrorCode errorCode) {
        if(!StringUtils.hasText(text)) {
            throw new ApiException(errorCode);
        }
    }
}

public String login(String username, String password) {

    ApiExceptionAsserts.hasText(username, UserResultCode.USERNAME_REQUIRED);
    ApiExceptionAsserts.hasText(password, UserResultCode.PASSWORD_REQUIRED);

    String token = null;

    try {
        UserDetails userDetails = userDetailsService.loadUserByUsername(username);
        System.out.println(userDetails.toString());
        if(!passwordEncoder.matches(password, userDetails.getPassword())){
            throw new BadCredentialsException("密码不正确");
        }
        UsernamePasswordAuthenticationToken authentication = new UsernamePasswordAuthenticationToken(userDetails, null, userDetails.getAuthorities());
        SecurityContextHolder.getContext().setAuthentication(authentication);
        token = jwtTokenUtil.generateToken(userDetails);
    } catch (Exception e) {
        e.printStackTrace();
        LOGGER.warn("登录异常: {}", e.getMessage());
    }
    return token;
}