From 304d3036527faf7e0751d658098077c4cdef057d Mon Sep 17 00:00:00 2001
From: wangxiang <1827945911@qq.com>
Date: Mon, 20 Feb 2023 18:07:05 +0800
Subject: [PATCH] =?UTF-8?q?:rocket:=20=E4=BC=81=E4=B8=9A=E8=AE=A4=E8=AF=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../commonbiz/controller/PushThirdPartyController.java   | 9 +++++++++
 1 file changed, 9 insertions(+)

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