diff --git a/src/main/java/com/dreamchaser/depository_manage/controller/PageController.java b/src/main/java/com/dreamchaser/depository_manage/controller/PageController.java index afb78a1f..8fe8c07e 100644 --- a/src/main/java/com/dreamchaser/depository_manage/controller/PageController.java +++ b/src/main/java/com/dreamchaser/depository_manage/controller/PageController.java @@ -526,6 +526,7 @@ public class PageController { } else { mv.addObject("display", "none"); } + mv.addObject("number",userToken.getNumber()); return mv; } @@ -763,6 +764,7 @@ public class PageController { } else { mv.addObject("display", "none"); } + mv.addObject("number",userToken.getNumber()); return mv; } @@ -952,9 +954,15 @@ public class PageController { } @GetMapping("/table_stock") - public ModelAndView table_stock() { + public ModelAndView table_stock(HttpServletRequest request) { + String token = request.getHeader("user-token"); + if (token == null) { + token = (String) request.getSession().getAttribute("userToken"); + } + UserByPort userToken = AuthenticationTokenPool.getUserToken(token); ModelAndView mv = new ModelAndView(); mv.setViewName("pages/depository/table-stock"); + mv.addObject("number",userToken.getNumber()); return mv; } diff --git a/src/main/resources/templates/pages/depository/table-stock.html b/src/main/resources/templates/pages/depository/table-stock.html index 70ab7478..3b3cdeb3 100644 --- a/src/main/resources/templates/pages/depository/table-stock.html +++ b/src/main/resources/templates/pages/depository/table-stock.html @@ -16,6 +16,7 @@