moliparty/peko-java
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修复重复添加问题

Browse Source
This commit is contained in:
liaozetao
2023-11-23 11:00:58 +08:00
committed by khalil
parent e85ad6d1ff
commit 61256f5022
1 changed files with 20 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
accompany-business/accompany-business-service/src/main/java/com/accompany/business/service/ip/IpRegionWhiteService.java
View File

@@ -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<IpRegionWhiteMapper, IpReg
}
public void addIpRegionWhite(Long uid, String phone, Byte source) {
IpRegionWhite ipRegionWhite = new IpRegionWhite();
ipRegionWhite.setUid(uid);
ipRegionWhite.setPhone(phone);
ipRegionWhite.setSource(source);
ipRegionWhite.setCreateTime(new Date());
save(ipRegionWhite);
List<IpRegionWhite> ipRegionWhites = list(Wrappers.<IpRegionWhite>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);
}
}
}