diff --git a/zlpt-modules/zlpt-equip/src/main/java/com/bonus/zlpt/equip/controller/UserCollectController.java b/zlpt-modules/zlpt-equip/src/main/java/com/bonus/zlpt/equip/controller/UserCollectController.java index 8e0c746..12bc692 100644 --- a/zlpt-modules/zlpt-equip/src/main/java/com/bonus/zlpt/equip/controller/UserCollectController.java +++ b/zlpt-modules/zlpt-equip/src/main/java/com/bonus/zlpt/equip/controller/UserCollectController.java @@ -1,6 +1,7 @@ package com.bonus.zlpt.equip.controller; import java.util.List; +import java.util.Objects; import javax.servlet.http.HttpServletResponse; import cn.hutool.core.util.BooleanUtil; @@ -11,6 +12,8 @@ import com.bonus.zlpt.common.core.web.page.TableDataInfo; import com.bonus.zlpt.common.log.annotation.Log; import com.bonus.zlpt.common.log.enums.BusinessType; import com.bonus.zlpt.common.security.annotation.RequiresPermissions; +import com.bonus.zlpt.common.security.utils.SecurityUtils; +import com.bonus.zlpt.system.api.model.LoginUser; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.GetMapping; import org.springframework.web.bind.annotation.PostMapping; @@ -96,12 +99,17 @@ public class UserCollectController extends BaseController @PutMapping public AjaxResult edit(@RequestBody UserCollect userCollect) { - if (BooleanUtil.isTrue(userCollect.getIsCollect())) { - userCollectService.deleteUserCollectByMaId(userCollect.getMaId()); - return toAjax(userCollectService.insertUserCollect(userCollect)); - } else { - return toAjax(userCollectService.deleteUserCollectByMaId(userCollect.getMaId())); + LoginUser loginUser = SecurityUtils.getLoginUser(); + if (Objects.nonNull(loginUser)) { + userCollect.setUserId(loginUser.getUserid()); + if (BooleanUtil.isTrue(userCollect.getIsCollect())) { + userCollectService.deleteUserCollectByMaId(userCollect.getMaId()); + return toAjax(userCollectService.insertUserCollect(userCollect)); + } else { + return toAjax(userCollectService.deleteUserCollectByMaId(userCollect.getMaId())); + } } + return null; } /**