🚀 企业认证

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.<PushThirdParty>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();
     }