快捷導(dǎo)航

SpringBoot中通過實現(xiàn)WebMvcConfigurer參數(shù)校驗的方法示例

更新時間：2019年11月21日 14:58:00 作者：hhsway

這篇文章主要介紹了SpringBoot中通過實現(xiàn)WebMvcConfigurer參數(shù)校驗的方法示例，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

在Spring5.0和SpringBoot2.0中廢棄了WebMvcConfigurerAdapter類。

現(xiàn)有兩種解決方案

1 直接實現(xiàn)WebMvcConfigurer （官方推薦）
2 直接繼承WebMvcConfigurationSupport

本篇文章討論下使用第一種方式完成參數(shù)校驗。

首先附上代碼。

@Slf4j
@Controller
@RequestMapping("/goods")
public class GoodsController {
 
  @Autowired
  MiaoshaUserService miaoshaUserService;
 
 
//  @GetMapping("/to_list")
//  public String toList(Model model, MiaoshaUser miaoshaUser) {
//    model.addAttribute("miaoshaUser",miaoshaUser);
//    return "list";
//  }
 
 
  @GetMapping("/to_list")
  public String toList(Model model,
             HttpServletResponse response,
             @CookieValue(value = MiaoshaUserServiceImpl.COOKIE_NAME_TOKEN,required = false)String cookieToken,
             @RequestParam(value = MiaoshaUserServiceImpl.COOKIE_NAME_TOKEN,required = false)String paramToken) {
    if(StringUtils.isEmpty(cookieToken) && StringUtils.isEmpty(paramToken)){
      return "login";
    }
    String token = StringUtils.isEmpty(paramToken)?cookieToken:paramToken;
    MiaoshaUser miaoshaUser = miaoshaUserService.getByToken(response,token);
    model.addAttribute("miaoshaUser",miaoshaUser);
    return "list";
  }
 
}

toList需要實現(xiàn)的功能是在cookie或requestParam中獲取到token，通過token去redis中獲取詳細(xì)的用戶信息，再將用戶信息放到頁面進(jìn)行展示。上面?zhèn)鹘y(tǒng)的方式入?yún)⒂蟹浅６啵詈笮枰氖悄玫組iaohshaUser對象，在程序的實際編寫過程中，會有許多方法需要經(jīng)過這樣的過程獲取到該對象，如何精簡下代碼，在入?yún)⒅挥行枰@取的對象的時候，經(jīng)過一系列的操作從cookie拿到token以及redis獲取具體對象呢？

下面通過實現(xiàn)WebMvcConfigurer接口來實現(xiàn)。

/**
 * @author hsw
 * @Date 10:12 2018/5/8
 */
@Configuration
public class WebConfig implements WebMvcConfigurer {
 
  @Autowired
  UserArgumentResolver userArgumentResolver;
 
  @Override
  public void addArgumentResolvers(List<HandlerMethodArgumentResolver> argumentResolvers) {
    argumentResolvers.add(userArgumentResolver);
  }
 
}

新建一個WebConfig類實現(xiàn)WebMvcConfigurer接口，WebMvcConfigurer是一個攔截器，里面有非常多的方法，我們用到的是addArgumentResolvers方法。需要注意的是類頭需要加上@Configuration注解將該類交給spring管理。

addArgumentResolvers具體內(nèi)容后文會提到。

新建UserArgumentResolver類實現(xiàn)HandlerMethodArgumentResolver接口

實現(xiàn)HandlerMethodArgumentResolver接口需要重寫supportsParameter方法和resolveArgument方法。

resolveArgument中從cookie或requestparam中獲取token值，并從redis中查詢到具體的MiaoshaUser信息并返回。

UserArgumentResolver類編寫完成后，在上面的WebConfig類中的addArgumentResolvers方法中將UserArgumentResolver類的實例添加進(jìn)argumentResolvers參數(shù)中即可。

這兩個類完成后，controller類中的toList方法如第一段注釋代碼所示。

/**
 * @author hsw
 * @Date 10:25 2018/5/8
 */
@Service
public class UserArgumentResolver implements HandlerMethodArgumentResolver {
 
  @Autowired
  MiaoshaUserService userService;
 
  /*
   * Miaoshauser類做參數(shù)校驗
   * @author hsw
   * @date 2018/5/8 10:49
   * @param [methodParameter]
   * @return boolean
   */
  @Override
  public boolean supportsParameter(MethodParameter methodParameter) {
    Class<?> clazz = methodParameter.getParameterType();//如若不是MiaoshaUser類不進(jìn)行接下來操作
    return clazz == MiaoshaUser.class;
  }
 
  @Override
  public Object resolveArgument(MethodParameter methodParameter,
                 ModelAndViewContainer modelAndViewContainer,
                 NativeWebRequest nativeWebRequest,
                 WebDataBinderFactory webDataBinderFactory) throws Exception {
    HttpServletRequest request =nativeWebRequest.getNativeRequest(HttpServletRequest.class);//獲取HttpServletRequest
    HttpServletResponse response =nativeWebRequest.getNativeResponse(HttpServletResponse.class);//獲取HttpServletResponse 
 
    String paramToken = request.getParameter(MiaoshaUserServiceImpl.COOKIE_NAME_TOKEN);
    String cookieToken = getCookieValue(request,MiaoshaUserServiceImpl.COOKIE_NAME_TOKEN);
    if(StringUtils.isEmpty(cookieToken) && StringUtils.isEmpty(paramToken)){
      return null;
    }
 
    String token = StringUtils.isEmpty(paramToken)?cookieToken:paramToken;
    return userService.getByToken(response,token);//通過token查詢詳細(xì)信息并返回，返回類為MiaoshaUser
  }
 
  private String getCookieValue(HttpServletRequest request, String cookieName) {
    Cookie[] cookies = request.getCookies();
    for(Cookie cookie:cookies){
      if(cookie.getName().equals(cookieName)){
        return cookie.getValue();
      }
    }
    return null;
  }
}

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: