From c92aa3899e11a355cddfcfeaececcc843d7ad1bb Mon Sep 17 00:00:00 2001
From: xuhy <3313886187@qq.com>
Date: 星期三, 04 九月 2024 19:14:18 +0800
Subject: [PATCH] 修改

---
 ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TBoardController.java |   26 +++++++++++++++++++++-----
 1 files changed, 21 insertions(+), 5 deletions(-)

diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TBoardController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TBoardController.java
index 0fcdc8f..0c00f3c 100644
--- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TBoardController.java
+++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/api/TBoardController.java
@@ -1,10 +1,11 @@
 package com.ruoyi.web.controller.api;
 
 
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.ruoyi.common.core.domain.AjaxResult;
+import com.ruoyi.framework.web.service.TokenService;
 import com.ruoyi.system.domain. TBoard;
 import com.ruoyi.system.service.TBoardService;
-import com.ruoyi.system.service. TBoardService;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -26,10 +27,12 @@
 public class TBoardController {
 
     private final TBoardService boardService;
+    private final TokenService tokenService;
 
     @Autowired
-    public  TBoardController( TBoardService  boardService) {
+    public  TBoardController(TBoardService  boardService, TokenService tokenService) {
         this. boardService =  boardService;
+        this.tokenService = tokenService;
     }
 
     /**
@@ -37,8 +40,10 @@
      */
     @ApiOperation( value = "查询餐桌列表")
     @PostMapping(value = "/list")
-    public AjaxResult<List< TBoard>> list() {
-        return AjaxResult.success( boardService.list());
+    public AjaxResult<List<TBoard>> list() {
+        Long objectId = tokenService.getLoginUser().getObjectId();
+        return AjaxResult.success( boardService.list(Wrappers.lambdaQuery(TBoard.class)
+                .eq(TBoard::getShopId,objectId)));
     }
 
     /**
@@ -47,6 +52,7 @@
     @ApiOperation( value = "添加餐桌")
     @PostMapping(value = "/add")
     public AjaxResult<Boolean> add(@RequestBody  TBoard dto) {
+        dto.setShopId(tokenService.getLoginUser().getObjectId());
         return AjaxResult.success( boardService.save(dto));
     }
 
@@ -60,11 +66,21 @@
     }
 
     /**
+     * 根据店铺id查询餐桌列表
+     */
+    @ApiOperation( value = "根据店铺id查询餐桌列表")
+    @GetMapping(value = "/getBoardByShopId")
+    public AjaxResult<List<TBoard>> getBoardByShopId(@RequestParam("shopId") Long shopId) {
+        return AjaxResult.success( boardService.list(Wrappers.lambdaQuery(TBoard.class)
+                .eq(TBoard::getShopId,shopId)));
+    }
+
+    /**
      * 查看 餐桌详情
      */
     @ApiOperation( value = "查看餐桌详情")
     @GetMapping(value = "/getDetailById")
-    public AjaxResult< TBoard> getDetailById(@RequestParam("id") Long id) {
+    public AjaxResult<TBoard> getDetailById(@RequestParam("id") Long id) {
         return AjaxResult.success( boardService.getById(id));
     }
 

--
Gitblit v1.7.1