🚀 消息推送模块

kicc-platform/kicc-platform-biz/kicc-common-biz/src/main/java/com/cloud/kicc/commonbiz/controller/PushApplicationController.java

@@ -5,12 +5,17 @@ import cn.hutool.core.lang.UUID;
 import cn.hutool.core.util.StrUtil;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.cloud.kicc.common.core.api.R;
 import com.cloud.kicc.common.core.constant.AppConstants;
+import com.cloud.kicc.common.core.exception.CheckedException;
+import com.cloud.kicc.common.security.util.SecurityUtils;
 import com.cloud.kicc.commonbiz.api.entity.PushMessage;
 import com.cloud.kicc.commonbiz.api.entity.PushApplication;
+import com.cloud.kicc.commonbiz.api.entity.PushThirdParty;
 import com.cloud.kicc.commonbiz.service.IPushApplicationService;
+import com.cloud.kicc.commonbiz.service.IPushThirdPartyService;
 import com.cloud.kicc.system.api.entity.OauthClientDetails;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
@@ -35,6 +40,7 @@ import java.util.Arrays;
 public class PushApplicationController {
 
     private final IPushApplicationService iPushApplicationService;
+    private final IPushThirdPartyService iPushThirdPartyService;
 
     private LambdaQueryWrapper<PushApplication> getQueryWrapper(PushApplication pushApplication) {
         return new LambdaQueryWrapper<PushApplication>()
@@ -58,6 +64,10 @@ public class PushApplicationController {
     @ApiOperation("保存")
     @PostMapping("/save")
     public R save(@Valid @RequestBody PushApplication pushApplication) {
+        // 验证企业是否已经认证
+        if (iPushThirdPartyService.count(Wrappers.<PushThirdParty>lambdaQuery().eq(PushThirdParty::getUserId, SecurityUtils.getUser().getId())) == 0){
+            throw new CheckedException("当前企业未认证，请认证后在操作！");
+        }
         pushApplication.setMessageSecret(UUID.randomUUID().toString().replace("-",""));
         iPushApplicationService.save(pushApplication);
         return R.ok();
@@ -66,6 +76,9 @@ public class PushApplicationController {
     @ApiOperation("修改")
     @PutMapping("/update")
     public R update(@Valid @RequestBody PushApplication pushApplication) {
+        if (iPushThirdPartyService.count(Wrappers.<PushThirdParty>lambdaQuery().eq(PushThirdParty::getUserId, SecurityUtils.getUser().getId())) == 0){
+            throw new CheckedException("当前企业未认证，请认证后在操作！");
+        }
         iPushApplicationService.updateById(pushApplication);
         return R.ok();
     }
@@ -73,6 +86,9 @@ public class PushApplicationController {
     @ApiOperation("删除")
     @DeleteMapping("/remove/{ids:[\\w,]+}")
     public R<Boolean> removeById(@PathVariable String[] ids) {
+        if (iPushThirdPartyService.count(Wrappers.<PushThirdParty>lambdaQuery().eq(PushThirdParty::getUserId, SecurityUtils.getUser().getId())) == 0){
+            throw new CheckedException("当前企业未认证，请认证后在操作！");
+        }
         iPushApplicationService.removeByIds(Arrays.asList(ids));
         return R.ok();
     }