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); + } } }