diff --git a/kicc-platform/kicc-platform-biz/kicc-common-biz/src/main/java/com/cloud/kicc/commonbiz/controller/PushThirdPartyController.java b/kicc-platform/kicc-platform-biz/kicc-common-biz/src/main/java/com/cloud/kicc/commonbiz/controller/PushThirdPartyController.java index b976e8b1..f9007217 100644 --- a/kicc-platform/kicc-platform-biz/kicc-common-biz/src/main/java/com/cloud/kicc/commonbiz/controller/PushThirdPartyController.java +++ b/kicc-platform/kicc-platform-biz/kicc-common-biz/src/main/java/com/cloud/kicc/commonbiz/controller/PushThirdPartyController.java @@ -4,9 +4,11 @@ package com.cloud.kicc.commonbiz.controller; 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.security.util.SecurityUtils; import com.cloud.kicc.commonbiz.api.entity.PushThirdParty; import com.cloud.kicc.commonbiz.service.IPushThirdPartyService; import io.swagger.annotations.Api; @@ -52,9 +54,16 @@ public class PushThirdPartyController { return R.ok(iPushThirdPartyService.getById(id)); } + @ApiOperation("根据用户ID查询") + @GetMapping("/getByUserId/{id:\\w+}") + public R getByUserId(@PathVariable("id") String id) { + return R.ok(iPushThirdPartyService.getOne(Wrappers.lambdaQuery().eq(PushThirdParty::getUserId, id))); + } + @ApiOperation("保存") @PostMapping("/save") public R save(@Valid @RequestBody PushThirdParty pushThirdParty) { + if (StrUtil.isBlank(pushThirdParty.getUserId())) pushThirdParty.setUserId(SecurityUtils.getUser().getId()); iPushThirdPartyService.save(pushThirdParty); return R.ok(); }