更新 'ruoyi-admin/src/main/java/com/ruoyi/app/tableqrcode/TableQrcodeController.java'

1

ruoyi-admin/src/main/java/com/ruoyi/app/tableqrcode/TableQrcodeController.java

@@ -207,29 +207,35 @@ public class TableQrcodeController extends BaseController
         String userId = jwtUtil.getusid(token);
         InfoUser currentUser = infoUserService.selectInfoUserByUserId(Long.valueOf(userId));
 
-        TableQrcode query = new TableQrcode();
-        query.setType(type);
-        query.setStatus(status);
-        query.setUseStatus(useStatus);
+        LambdaQueryWrapper<TableQrcode> query = new LambdaQueryWrapper<>();
+        query.eq(TableQrcode::getType,type);
+        query.eq(TableQrcode::getStatus,status);
+        query.eq(TableQrcode::getUseStatus,useStatus);
 
         if ("4".equals(currentUser.getUserType()))
         {
             // 摊主只能看自己的
-            query.setStoreId(currentUser.getStoreId());
+            query.eq(TableQrcode::getStoreId,currentUser.getStoreId());
         }
         else if ("3".equals(currentUser.getUserType()))
         {
-            query.setNightMarketId(Long.valueOf(userId));
+            query.eq(TableQrcode::getNightMarketId,Long.valueOf(userId));
             // 夜市管理员可按摊位筛选，或查看全部
             if (storeId != null)
             {
-                query.setStoreId(storeId);
+                query.eq(TableQrcode::getStoreId,storeId);
             }
         }
         else if("1".equals(currentUser.getUserType())){
-            query.setStoreId(storeId);
+            List<PosStore> stores=posStoreService.list(new LambdaQueryWrapper<PosStore>().eq(PosStore::getUserId,userId));
+            if(stores.isEmpty()){
+                query.eq(TableQrcode::getId,0L);
+            }else{
+                query.in(TableQrcode::getStoreId,stores.stream().map(PosStore::getId).collect(Collectors.toList()));
+            }
+
         }
-        List<TableQrcode> list = tableQrcodeService.selectTableQrcodeList(query);
+        List<TableQrcode> list = tableQrcodeService.list(query);
         return success(list);
     }