SpringMVC Context 获取

import lombok.AllArgsConstructor;

import lombok.Builder;

import lombok.Data;

import lombok.NoArgsConstructor;

@Data

@NoArgsConstructor

@AllArgsConstructor

@Builder

public class User {

    private Long id;

    private String name;

}

import java.lang.annotation.ElementType;

import java.lang.annotation.Retention;

import java.lang.annotation.RetentionPolicy;

import java.lang.annotation.Target;

/**

 * 自定义 当前用户 注解
 * 注解 参数
 * 此注解在验证token通过后，获取当前token包含用户
 */
@Target({ElementType.PARAMETER})
@Retention(RetentionPolicy.RUNTIME)
public @interface CurrentUser {
}

import lombok.extern.slf4j.Slf4j;

import org.springframework.core.MethodParameter;

import org.springframework.web.bind.support.WebDataBinderFactory;

import org.springframework.web.context.request.NativeWebRequest;

import org.springframework.web.method.support.HandlerMethodArgumentResolver;

import org.springframework.web.method.support.ModelAndViewContainer;

import javax.servlet.http.Cookie;

import javax.servlet.http.HttpServletRequest;

import java.util.Arrays;

import java.util.Optional;

/**

 * @CurrentUser 注解 解析器
 */
@Slf4j
public class CurrentUserMethodArgumentResolver implements HandlerMethodArgumentResolver {
    @Override
    public boolean supportsParameter(MethodParameter parameter) {
        // 1. User 类的对象是否可以分配给当前的方法参数
        boolean assignable = parameter.getParameterType().isAssignableFrom(User.class);
        // 2. 当前方法参数上是否有 CurrentUser 的注解
        boolean hasAnnotation = parameter.hasParameterAnnotation(CurrentUser.class);
        return assignable && hasAnnotation;
    }

    @Override

    public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer, NativeWebRequest webRequest, WebDataBinderFactory binderFactory) {

        HttpServletRequest request = webRequest.getNativeRequest(HttpServletRequest.class);

        Optional<String> id = getValue(request, "id");

        Optional<String> name = getValue(request, "name");

        return User.builder()

            .id(id.map(Long::parseLong).orElse(null))

            .name(name.orElse(null))

            .build();

    }

    private Optional<String> getValue(HttpServletRequest request, String name) {

        return Arrays.stream(request.getCookies())

            .filter(c -> c.getName().equals(name))

            .findFirst()

            .map(Cookie::getValue);

    }

}

import org.springframework.context.annotation.Configuration;

import org.springframework.web.method.support.HandlerMethodArgumentResolver;

import org.springframework.web.servlet.config.annotation.WebMvcConfigurationSupport;

import java.util.List;

@Configuration

public class WebMvcConfig extends WebMvcConfigurationSupport {

    @Override

    protected void addArgumentResolvers(List<HandlerMethodArgumentResolver> argumentResolvers) {

        argumentResolvers.add(new CurrentUserMethodArgumentResolver());

    }

}

import lombok.extern.slf4j.Slf4j;

import org.springframework.web.bind.annotation.GetMapping;

import org.springframework.web.bind.annotation.RestController;

@RestController

@Slf4j

public class DemoController {

    @GetMapping("/test")

    public User test(@CurrentUser User user) {

        log.info("user: {}", user);

        return user;

    }

}

id=1

id=1;name=joy