添加pc端取消收藏、添加收藏——未完成

ruoyi-bussiness/src/main/java/com/ruoyi/cms/controller/cms/CompanyController.java

@@ -1,7 +1,11 @@
 package com.ruoyi.cms.controller.cms;
 
 
+import com.ruoyi.cms.service.IAppUserService;
+import com.ruoyi.cms.service.ICompanyCollectionService;
 import com.ruoyi.cms.util.RoleUtils;
+import com.ruoyi.common.annotation.BussinessLog;
+import com.ruoyi.common.core.domain.entity.AppUser;
 import com.ruoyi.common.core.domain.entity.Company;
 import com.ruoyi.cms.service.ICompanyService;
 import com.ruoyi.common.annotation.Log;
@@ -9,6 +13,7 @@ import com.ruoyi.common.core.controller.BaseController;
 import com.ruoyi.common.core.domain.AjaxResult;
 import com.ruoyi.common.core.page.TableDataInfo;
 import com.ruoyi.common.enums.BusinessType;
+import com.ruoyi.common.utils.SecurityUtils;
 import com.ruoyi.common.utils.StringUtils;
 import com.ruoyi.common.utils.poi.ExcelUtil;
 import io.swagger.annotations.Api;
@@ -33,6 +38,10 @@ public class CompanyController extends BaseController
 {
     @Autowired
     private ICompanyService companyService;
+    @Autowired
+    private ICompanyCollectionService companyCollectionService;
+    @Autowired
+    private IAppUserService appUserService;
 
     /**
      * 查询公司列表
@@ -150,4 +159,50 @@ public class CompanyController extends BaseController
         }
         return AjaxResult.success(companyService.queryCodeCompany(code));
     }
+
+    /**
+     * 用户收藏公司
+     */
+    @BussinessLog(title = "用户收藏公司")
+    @PostMapping("/collection/{companyId}")
+    @ApiOperation("用户收藏公司")
+    public AjaxResult companyCollection(@PathVariable("companyId") Long companyId)
+    {
+        Long userId=null;
+        if(SecurityUtils.isLogin()){
+            String idCard=RoleUtils.getCurrentUseridCard();
+            AppUser appUser=appUserService.selectAppuserByIdcard(idCard);
+            if(appUser==null){
+                return AjaxResult.error("用户信息未完善,请完善身份证信息!");
+            }else{
+                userId=appUser.getUserId();
+            }
+        }else{
+            return AjaxResult.error("用户未登录!");
+        }
+        return toAjax(companyCollectionService.companySysCollection(companyId,userId));
+    }
+
+    /**
+     * 用户取消收藏公司
+     */
+    @BussinessLog(title = "用户取消收藏公司")
+    @DeleteMapping("/collection/{companyId}")
+    @ApiOperation("用户取消收藏公司")
+    public AjaxResult companyCancel(@PathVariable("companyId") Long companyId)
+    {
+        Long userId;
+        if(SecurityUtils.isLogin()){
+            String idCard=RoleUtils.getCurrentUseridCard();
+            AppUser appUser=appUserService.selectAppuserByIdcard(idCard);
+            if(appUser==null){
+                return AjaxResult.error("用户信息未完善,请完善身份证信息!");
+            }else{
+                userId=appUser.getUserId();
+            }
+        }else{
+            return AjaxResult.error("用户未登录!");
+        }
+        return toAjax(companyCollectionService.companySysCancel(companyId,userId));
+    }
 }