springMvc参数解析器，让你的方法入参更优雅

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

/**
 * 描述：当前登录用户 </br>
 * 作者：IT学习道场 </br>
 * 时间：2022-11-15 19:07
 */
@Data
@NoArgsConstructor
@AllArgsConstructor
public class CurrentUser {
    private Long userId;
    private String userName;
}
1.
2.
3.
4.
5.
6.
7.
8.
9.
10.
11.
12.
13.
14.
15.
16.

import java.lang.annotation.*;

/**
 * 描述：自定义用户注解 </br>
 * 作者：IT学习道场 </br>
 * 时间：2022-11-15 16:56
 */
@Target(ElementType.PARAMETER)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface WlcUser {
}
1.
2.
3.
4.
5.
6.
7.
8.
9.
10.
11.
12.

import com.example.demo.annotation.WlcUser;
import com.example.demo.po.CurrentUser;
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;

/**
 * 描述：当前用户解析器 </br>
 * 作者：IT学习道场 </br>
 * 时间：2022-11-15 17:01
 */
public class CurrentUserMethodArgumentResolver implements HandlerMethodArgumentResolver {

    @Override
    public boolean supportsParameter(MethodParameter parameter) {
        //方法参数是否带有 WlcUser注解，带的话就解析执行下方 resolveArgument 方法
        return parameter.hasParameterAnnotation(WlcUser.class);
    }

    @Override
    public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer, NativeWebRequest webRequest, WebDataBinderFactory binderFactory) throws Exception {
        //1 这里面可以从httpHeader中获取用户标识或者其他属性值，也可以通过springIoc中bean来执行增删改查，rpc来获取，但是要用spring的上下文来获取bean
        //2 也可以从token中解析出当前登录用户信息，返回也行对象 CurrentUser
        //3 也可以从你自己定义的用户上下文中获取 CurrentUser，前提你要自己写一个用户上下文
        //4 这里调用sprinIoc里管理的bean的话，需要 SpringContextUtil上下文来获取bean，在调用方法
        //5 这里只是自己new了一个当前用户，来直接赋予加了 WlcUser 注解修饰的对象
        CurrentUser currentUser = new CurrentUser(1L,"老王");
        return currentUser;
    }
}
1.
2.
3.
4.
5.
6.
7.
8.
9.
10.
11.
12.
13.
14.
15.
16.
17.
18.
19.
20.
21.
22.
23.
24.
25.
26.
27.
28.
29.
30.
31.
32.

import com.example.demo.resolver.CurrentUserMethodArgumentResolver;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.method.support.HandlerMethodArgumentResolver;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
import java.util.List;

/**
 * 描述：web拦截器，解析器配置类 </br>
 * 作者：IT学习道场 </br>
 * 时间：2022-11-15 19:09
 */
@Configuration
public class WebConfig implements WebMvcConfigurer {
    @Override
    public void addArgumentResolvers(List<HandlerMethodArgumentResolver> resolvers) {
        resolvers.add(new CurrentUserMethodArgumentResolver());
    }
}
1.
2.
3.
4.
5.
6.
7.
8.
9.
10.
11.
12.
13.
14.
15.
16.
17.
18.

import com.example.demo.annotation.WlcUser;
import com.example.demo.po.CurrentUser;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

/**
 * 描述：参数解析解TestController </br>
 * 作者：IT学习道场 </br>
 * 时间：2022-11-16 9:25
 */
@Slf4j
@RestController
@RequestMapping("/currentUserResolver")
public class CurrentUserResolverController {

    @GetMapping("/currentUserResolve")
    public void currentUserResolve(@WlcUser CurrentUser user){
        log.info("参数解析器注入 CurrentUser = {}", user.toString());
    }
}
1.
2.
3.
4.
5.
6.
7.
8.
9.
10.
11.
12.
13.
14.
15.
16.
17.
18.
19.
20.
21.
22.