org.springframework.security.oauth2.provider.endpoint.TokenEndpoint.postAccessToken（）

package com.wongoing.common.model;/** * @description: 代码枚举 * @author: zheng * @date: Created in 2021/1/26 11:18 * @version: 0.0.1 * @modified By: */public enum CodeEnum {    SUCCESS(0),    ERROR(1);    private Integer code;    CodeEnum(Integer code) {        this.code = code;    }    public Integer getCode() {        return this.code;    }}

package com.wongoing.common.model;import lombok.AllArgsConstructor;import lombok.Data;import lombok.NoArgsConstructor;import java.io.Serializable;/** * @description: Rest API 接口方法返回类型定义 * @author: zheng * @date: Created in 2021/1/26 13:25 * @version: 0.0.1 * @modified By: */@Data@NoArgsConstructor@AllArgsConstructorpublic class Result<T> implements Serializable {    private T data;    private Integer code;    private String msg;    public static <T> Result<T> of(T data, Integer code, String msg) {        return new Result<>(data, code, msg);    }    public static <T> Result<T> succeed(String msg) {        return of(null, CodeEnum.SUCCESS.getCode(), msg);    }    public static <T> Result<T> succeed(T model, String msg) {        return of(model, CodeEnum.SUCCESS.getCode(), msg);    }    public static <T> Result<T> succeed(T model) {        return of(model, CodeEnum.SUCCESS.getCode(), "");    }    public static <T> Result<T> failed(String msg) {        return of(null, CodeEnum.ERROR.getCode(), msg);    }    public static <T> Result<T> failed(T model, String msg) {        return of(model, CodeEnum.ERROR.getCode(), msg);    }}

package com.wongoing.oauth2.filter;import com.wongoing.common.constant.SecurityConstants;import com.wongoing.common.context.TenantContextHolder;import com.wongoing.common.model.Result;import lombok.extern.slf4j.Slf4j;import org.aspectj.lang.ProceedingJoinPoint;import org.aspectj.lang.annotation.Around;import org.aspectj.lang.annotation.Aspect;import org.springframework.http.HttpStatus;import org.springframework.http.ResponseEntity;import org.springframework.security.authentication.InsufficientAuthenticationException;import org.springframework.security.core.Authentication;import org.springframework.security.oauth2.common.OAuth2AccessToken;import org.springframework.security.oauth2.common.util.OAuth2Utils;import org.springframework.security.oauth2.provider.OAuth2Authentication;import org.springframework.stereotype.Component;import java.security.Principal;import java.util.Map;/** * @description: oauth-token拦截器 * 1. 赋值租户 * 2. 统一返回token格式 * * @author: zheng * @date: Created in 2021/7/12 16:25 * @version: 0.0.1 * @modified By: */@Slf4j@Component@Aspectpublic class OauthTokenAspect {    @Around("execution(* org.springframework.security.oauth2.provider.endpoint.TokenEndpoint.postAccessToken(..))")    public Object handleControllerMethod(ProceedingJoinPoint joinPoint) throws Throwable {        try {            Object[] args = joinPoint.getArgs();            Principal principal = (Principal) args[0];            if (!(principal instanceof Authentication)) {                throw new InsufficientAuthenticationException("There is no client authentication. Try adding an appropriate authentication filter.");            }            String clientId = this.getClientId(principal);            Map<String, String> parameters = (Map<String, String>) args[1];            String grantType = parameters.get(OAuth2Utils.GRANT_TYPE);            //保存租户id            TenantContextHolder.setTenant(clientId);            Object proceed = joinPoint.proceed();            if (SecurityConstants.AUTHORIZATION_CODE.equals(grantType)) {                /**                 * 如果使用 @EnableOAuth2Sso 注解不能修改返回格式，否则授权码模式可以统一改                 * 因为本项目的 sso-demo/ss-sso 里面使用了 @EnableOAuth2Sso 注解，所以这里就不修改授权码模式的token返回值了                 */                return proceed;            } else {                ResponseEntity<OAuth2AccessToken> responseEntity = (ResponseEntity<OAuth2AccessToken>) proceed;                OAuth2AccessToken body = responseEntity.getBody();                return ResponseEntity                        .status(HttpStatus.OK)                        .body(Result.succeed(body));            }        } finally {            TenantContextHolder.clear();        }    }    private String getClientId(Principal principal) {        Authentication client = (Authentication) principal;        if (!client.isAuthenticated()) {            throw new InsufficientAuthenticationException("The client is not authenticated.");        }        String clientId = client.getName();        if (client instanceof OAuth2Authentication) {            clientId = ((OAuth2Authentication) client).getOAuth2Request().getClientId();        }        return clientId;    }}

public abstract class OAuth2Utils {	public static final String GRANT_TYPE = "grant_type";}

public interface SecurityConstants {	/**     * 授权码模式     */    String AUTHORIZATION_CODE = "authorization_code";}