diff --git a/kicc-platform/kicc-platform-biz/kicc-common-biz/src/main/java/com/cloud/kicc/commonbiz/controller/PushApplicationController.java b/kicc-platform/kicc-platform-biz/kicc-common-biz/src/main/java/com/cloud/kicc/commonbiz/controller/PushApplicationController.java index ab4aa8f8..e646a422 100644 --- a/kicc-platform/kicc-platform-biz/kicc-common-biz/src/main/java/com/cloud/kicc/commonbiz/controller/PushApplicationController.java +++ b/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 getQueryWrapper(PushApplication pushApplication) { return new LambdaQueryWrapper() @@ -58,6 +64,10 @@ public class PushApplicationController { @ApiOperation("保存") @PostMapping("/save") public R save(@Valid @RequestBody PushApplication pushApplication) { + // 验证企业是否已经认证 + if (iPushThirdPartyService.count(Wrappers.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.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 removeById(@PathVariable String[] ids) { + if (iPushThirdPartyService.count(Wrappers.lambdaQuery().eq(PushThirdParty::getUserId, SecurityUtils.getUser().getId())) == 0){ + throw new CheckedException("当前企业未认证,请认证后在操作!"); + } iPushApplicationService.removeByIds(Arrays.asList(ids)); return R.ok(); }