com.thebund1st.daming.sdk.web.handler.SmsVerificationJwtExceptionHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of sms-verification-sdk Show documentation

Your description of the lib

The newest version!

package com.thebund1st.daming.sdk.web.handler;


import com.thebund1st.daming.sdk.jwt.SmsVerificationJwtException;
import org.springframework.http.HttpStatus;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;
import org.springframework.web.context.request.WebRequest;

import javax.servlet.http.HttpServletResponse;
import java.security.Principal;

@RestControllerAdvice
public class SmsVerificationJwtExceptionHandler {

    @ExceptionHandler({
            SmsVerificationJwtException.class
    })
    public void handle(SmsVerificationJwtException ex,
                       WebRequest request, HttpServletResponse response) throws Exception {
        Principal userPrincipal = request.getUserPrincipal();
        if (userPrincipal == null) {
            response.sendError(HttpStatus.UNAUTHORIZED.value(), ex.getMessage());
        } else {
            response.sendError(HttpStatus.FORBIDDEN.value(), ex.getMessage());
        }
    }
}