SpringBoot @ModelAttribute使用場景分析

Posted on by

前言

項目中遇到這麼一個使用場景，用戶的登錄信息給予token保存，在需要有登錄信息的地方，每次都要去獲取用戶Id,但每次在請求方法中去獲取用戶信息，代碼重復，冗餘，很low於是想到瞭用@ModelAttribute 這個屬性

@ModelAttribute有三種用法：
– 可以標註在方法上；
– 可以標註在方法中的參數上；
– 還可以和@RequestMapping一起標註在方法上；

使用場景

不用@ModelAttribute 時候在需要用戶信息的請求中每次需要單獨獲取用戶信息

String token = request.getAttribute("token").toString();
   User LoginUser = tokenService.decodeToken(token);

代碼重復每次都需要單獨去寫，

於是我想到瞭去優化一下代碼，在需要使用戶信息的controller中寫一個公共方法，每次直接獲取就可以瞭

private User gerUserInfo(HttpServletRequest request){
        String token = request.getAttribute("token").toString();
        User LoginUser = tokenService.decodeToken(token);
        return LoginUser;
    }

這樣寫代碼是簡化瞭一些，但是沒什麼特別大的改觀，還是要在每個需求用戶信息的請求Controller中調用此方法取獲取用用戶信息，如果多個Controller需要獲取用戶信息的話還需要重復寫

也是想到繼承，寫一個公共的controller叫BaseController，每次在需要用戶信息的controller中繼承BaseController 然後在調用就可以瞭

@RestController
public class BaseController {
    @Autowired
    private TokenService tokenService;
   

   private User gerUserInfo(HttpServletRequest request){
        String token = request.getAttribute("token").toString();
        User LoginUser = tokenService.decodeToken(token);
        return LoginUser;
    }
}

這樣看上去似乎比之前兩種做法都簡單便捷很多，在需要使用用戶信息的controller中直接繼承調用就可以啦，但是並沒有根本解決我們的問題，我們還是需要寫重復代碼，在每個controller單獨獲取用戶信息，這是最優嘛？並不是！！！

其實呢springboot提供@ModelAttribute這個註解屬性使用這個通過參數註入就可獲取啦

我們把上面的稍微調整一下如：

@RestController
public class BaseController {
    @Autowired
    private TokenService tokenService;
 

    @ModelAttribute
    public void userInfo(ModelMap modelMap, HttpServletRequest request) {
        String token = request.getAttribute("token").toString();
        User LoginUser = tokenService.decodeToken(token);
    
        modelMap.addAttribute("LoginUser", LoginUser);
        modelMap.addAttribute("userId", LoginUser.getUserId());

    }
}

然後在需要使用用戶信息的controller中進行參數映射就行啦

@ApiOperation(value = "用戶快過期優惠卷信息",tags = "優惠卷接口")
    @GetMapping("/expiredCoupon")
    public List<Coupon> userExpiredCoupon(@ModelAttribute("userId") @ApiParam(hidden = true) String userId){
        return couponService.getUserExpiredCoupon(userId);
    }
@GetMapping("/info")
    @ApiOperation("獲取用戶信息")
    public User getUseInfo(@ModelAttribute("LoginUser") User user) {
        return user;
    }

這樣用戶信息通過形參直接註入到controller中，我們直接在請求中使用就可以啦

@ModelAttribute詳解

  • 被@ModelAttribute註釋的方法會在此controller每個方法執行前被執行
  • 標註在方法上面的註解，將方法返回的對象存儲在model中，該方法在這個控制器其他映射方法執行之前調用@ModelAttribute註釋一個方法的參數 從model中獲取參數@ModelAttribute("LoginUser") User user參數user的值來源於BaseControlleruserInfo()方法中的model屬性

具體更詳細使用參考 @ModelAttribute註解的使用總結

到此這篇關於SpringBoot @ModelAttribute 用法的文章就介紹到這瞭,更多相關SpringBoot @ModelAttribute 內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：