From 61256f50220cf5e96848da1ea3ca7f83e5f81d31 Mon Sep 17 00:00:00 2001 From: liaozetao <1107136310@qq.com> Date: Thu, 23 Nov 2023 11:00:58 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E9=87=8D=E5=A4=8D=E6=B7=BB?= =?UTF-8?q?=E5=8A=A0=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../service/ip/IpRegionWhiteService.java | 26 ++++++++++++++----- 1 file changed, 20 insertions(+), 6 deletions(-) diff --git a/accompany-business/accompany-business-service/src/main/java/com/accompany/business/service/ip/IpRegionWhiteService.java b/accompany-business/accompany-business-service/src/main/java/com/accompany/business/service/ip/IpRegionWhiteService.java index 6ee123d24..7905784f1 100644 --- a/accompany-business/accompany-business-service/src/main/java/com/accompany/business/service/ip/IpRegionWhiteService.java +++ b/accompany-business/accompany-business-service/src/main/java/com/accompany/business/service/ip/IpRegionWhiteService.java @@ -1,16 +1,20 @@ package com.accompany.business.service.ip; +import cn.hutool.core.collection.CollectionUtil; +import cn.hutool.core.util.StrUtil; import com.accompany.business.model.ip.IpRegionWhite; import com.accompany.business.mybatismapper.ip.IpRegionWhiteMapper; import com.accompany.common.utils.CommonUtil; import com.accompany.core.base.SpringContextHolder; import com.accompany.core.model.Account; +import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import lombok.extern.slf4j.Slf4j; import org.springframework.scheduling.annotation.Async; import org.springframework.stereotype.Service; import java.util.Date; +import java.util.List; @Slf4j @Service @@ -40,12 +44,22 @@ public class IpRegionWhiteService extends ServiceImpl ipRegionWhites = list(Wrappers.lambdaQuery() + .eq(uid != null, IpRegionWhite::getUid, uid) + .eq(StrUtil.isNotEmpty(phone), IpRegionWhite::getPhone, phone)); + if (CollectionUtil.isNotEmpty(ipRegionWhites)) { + IpRegionWhite ipRegionWhite = ipRegionWhites.get(0); + ipRegionWhite.setSource(source); + ipRegionWhite.setCreateTime(new Date()); + updateById(ipRegionWhite); + } else { + IpRegionWhite ipRegionWhite = new IpRegionWhite(); + ipRegionWhite.setUid(uid); + ipRegionWhite.setPhone(phone); + ipRegionWhite.setSource(source); + ipRegionWhite.setCreateTime(new Date()); + save(ipRegionWhite); + } } }