From 90afd612f0933d733d804c21a56b313ef7a548a6 Mon Sep 17 00:00:00 2001
From: Helius <wangdoubleone@gmail.com>
Date: Mon, 22 Aug 2022 16:19:15 +0800
Subject: [PATCH] fix

---
 src/main/java/com/xcong/farmer/cms/modules/system/controller/CmsFriendlyLinkController.java |    7 ++++---
 1 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/main/java/com/xcong/farmer/cms/modules/system/controller/CmsFriendlyLinkController.java b/src/main/java/com/xcong/farmer/cms/modules/system/controller/CmsFriendlyLinkController.java
index 803e0eb..d0f3009 100644
--- a/src/main/java/com/xcong/farmer/cms/modules/system/controller/CmsFriendlyLinkController.java
+++ b/src/main/java/com/xcong/farmer/cms/modules/system/controller/CmsFriendlyLinkController.java
@@ -40,11 +40,11 @@
     @PostMapping(value = "/list")
     public Result list(@RequestBody @Valid FriendlyInfoListDto friendlyInfoListDto) {
         UserEntity user = LoginUserUtil.getLoginUser();
-        QueryWrapper<CmsAdInfoEntity> query = new QueryWrapper<>();
-        query.eq("company_id", user.getCompanyId());
 
+        CmsFriendlyLinkEntity friendlyLink = new CmsFriendlyLinkEntity();
+        friendlyLink.setCompanyId(user.getCompanyId());
         Page<CmsFriendlyLinkEntity> page = new Page<>(friendlyInfoListDto.getPageNum(), friendlyInfoListDto.getPageSize());
-        return Result.ok("获取成功", cmsFriendlyLinkService.page(page));
+        return Result.ok("获取成功", cmsFriendlyLinkService.findInPage(page, friendlyLink));
     }
 
     @ApiOperation(value = "根据ID获取信息")
@@ -61,6 +61,7 @@
         CmsFriendlyLinkEntity friendlyLink = CmsFriendlyLinkConversion.INSTANCE.addDtoToEntity(addFriendlyLinkDto);
         friendlyLink.setCreateBy(user.getNickname());
         friendlyLink.setUpdateBy(user.getNickname());
+        friendlyLink.setCompanyId(user.getCompanyId());
         cmsFriendlyLinkService.save(friendlyLink);
         return Result.ok("添加成功");
     }

--
Gitblit v1.9.1