模块化拆分
This commit is contained in:
禾几海
@@ -0,0 +1,228 @@
|
||||
package cn.celess.partnersite.serviceimpl;
|
||||
|
||||
import cn.celess.common.enmu.ResponseEnum;
|
||||
import cn.celess.common.entity.PartnerSite;
|
||||
import cn.celess.common.entity.dto.LinkApplyReq;
|
||||
import cn.celess.common.entity.dto.LinkReq;
|
||||
import cn.celess.common.entity.vo.PageData;
|
||||
import cn.celess.common.exception.MyException;
|
||||
import cn.celess.common.mapper.PartnerMapper;
|
||||
import cn.celess.common.service.MailService;
|
||||
import cn.celess.common.service.PartnerSiteService;
|
||||
import cn.celess.common.util.HttpUtil;
|
||||
import cn.celess.common.util.RedisUtil;
|
||||
import cn.celess.common.util.RegexUtil;
|
||||
import com.alibaba.druid.util.StringUtils;
|
||||
import com.fasterxml.jackson.databind.ObjectMapper;
|
||||
import com.github.pagehelper.PageHelper;
|
||||
import com.github.pagehelper.PageInfo;
|
||||
import lombok.SneakyThrows;
|
||||
import org.springframework.beans.BeanUtils;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.mail.SimpleMailMessage;
|
||||
import org.springframework.stereotype.Service;
|
||||
|
||||
import java.util.Date;
|
||||
import java.util.List;
|
||||
import java.util.UUID;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
/**
|
||||
* @author : xiaohai
|
||||
* @date : 2019/05/12 11:43
|
||||
*/
|
||||
@Service
|
||||
public class PartnerSiteServiceImpl implements PartnerSiteService {
|
||||
@Autowired
|
||||
PartnerMapper partnerMapper;
|
||||
@Autowired
|
||||
MailService mailService;
|
||||
@Autowired
|
||||
RedisUtil redisUtil;
|
||||
private static final String SITE_NAME = "小海博客";
|
||||
private static final String SITE_URL = "celess.cn";
|
||||
private static final String SITE_EMAIL = "a@celess.cn";
|
||||
|
||||
@Override
|
||||
public PartnerSite create(LinkReq reqBody) {
|
||||
if (reqBody == null) {
|
||||
throw new MyException(ResponseEnum.PARAMETERS_ERROR);
|
||||
}
|
||||
//判空
|
||||
if (reqBody.getName() == null || reqBody.getUrl() == null) {
|
||||
throw new MyException(ResponseEnum.PARAMETERS_ERROR);
|
||||
}
|
||||
//判空
|
||||
if (reqBody.getName().replaceAll(" ", "").isEmpty() || reqBody.getUrl().replaceAll(" ", "").isEmpty()) {
|
||||
throw new MyException(ResponseEnum.PARAMETERS_ERROR);
|
||||
}
|
||||
//是否存在 同名
|
||||
if (partnerMapper.existsByName(reqBody.getName())) {
|
||||
throw new MyException(ResponseEnum.DATA_HAS_EXIST);
|
||||
}
|
||||
//url是否合法
|
||||
if (!RegexUtil.urlMatch(reqBody.getUrl())) {
|
||||
throw new MyException(ResponseEnum.PARAMETERS_URL_ERROR);
|
||||
}
|
||||
PartnerSite partnerSite = new PartnerSite();
|
||||
reqBody.setId(0);
|
||||
if (!reqBody.getUrl().contains("http://") && !reqBody.getUrl().contains("https://")) {
|
||||
reqBody.setUrl("http://" + reqBody.getUrl());
|
||||
}
|
||||
BeanUtils.copyProperties(reqBody, partnerSite);
|
||||
if (reqBody.getIconPath() == null) {
|
||||
partnerSite.setIconPath("");
|
||||
}
|
||||
if (reqBody.getDesc() == null) {
|
||||
partnerSite.setDesc("");
|
||||
}
|
||||
partnerMapper.insert(partnerSite);
|
||||
return partnerSite;
|
||||
}
|
||||
|
||||
@Override
|
||||
public Boolean del(long id) {
|
||||
//判断数据是否存在
|
||||
if (!partnerMapper.existsById(id)) {
|
||||
throw new MyException(ResponseEnum.DATA_NOT_EXIST);
|
||||
}
|
||||
return partnerMapper.delete(id) == 1;
|
||||
}
|
||||
|
||||
@Override
|
||||
public PartnerSite update(LinkReq reqBody) {
|
||||
PartnerSite partnerSite = partnerMapper.findById(reqBody.getId());
|
||||
if (partnerSite == null) {
|
||||
throw new MyException(ResponseEnum.DATA_NOT_EXIST);
|
||||
}
|
||||
if (partnerMapper.existsByName(reqBody.getName()) && !reqBody.getName().equals(partnerSite.getName())) {
|
||||
throw new MyException(ResponseEnum.DATA_HAS_EXIST);
|
||||
}
|
||||
if (!RegexUtil.urlMatch(reqBody.getUrl())) {
|
||||
throw new MyException(ResponseEnum.PARAMETERS_URL_ERROR);
|
||||
}
|
||||
if (!reqBody.getUrl().contains("http://") && !reqBody.getUrl().contains("https://")) {
|
||||
reqBody.setUrl("http://" + reqBody.getUrl());
|
||||
}
|
||||
if (reqBody.isOpen() != partnerSite.getOpen() && !partnerSite.getNotification() && !StringUtils.isEmpty(partnerSite.getEmail())) {
|
||||
SimpleMailMessage smm = new SimpleMailMessage();
|
||||
smm.setTo(partnerSite.getEmail());
|
||||
smm.setText("您的友链申请，已通过");
|
||||
smm.setSubject("友链申请通过");
|
||||
smm.setSentDate(new Date());
|
||||
mailService.send(smm);
|
||||
partnerSite.setNotification(true);
|
||||
}
|
||||
BeanUtils.copyProperties(reqBody, partnerSite);
|
||||
partnerMapper.update(partnerSite);
|
||||
partnerSite.setName(reqBody.getName());
|
||||
partnerSite.setUrl(reqBody.getUrl());
|
||||
partnerSite.setOpen(reqBody.isOpen());
|
||||
|
||||
return partnerSite;
|
||||
}
|
||||
|
||||
@Override
|
||||
public PageData<PartnerSite> partnerSitePages(int page, int count, Boolean deleted) {
|
||||
PageHelper.startPage(page, count);
|
||||
List<PartnerSite> sitePage = partnerMapper.findAll(deleted);
|
||||
PageInfo<PartnerSite> pageInfo = new PageInfo<PartnerSite>(sitePage);
|
||||
return new PageData<>(pageInfo, sitePage);
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<PartnerSite> findAll() {
|
||||
List<PartnerSite> all = partnerMapper.findAll();
|
||||
all.forEach(partnerSite -> partnerSite.setDelete(null));
|
||||
return all;
|
||||
}
|
||||
|
||||
@SneakyThrows
|
||||
@Override
|
||||
public PartnerSite apply(LinkApplyReq linkApplyReq) {
|
||||
// 空值字段
|
||||
if (StringUtils.isEmpty(linkApplyReq.getName())
|
||||
|| StringUtils.isEmpty(linkApplyReq.getUrl())
|
||||
|| StringUtils.isEmpty(linkApplyReq.getEmail())
|
||||
|| StringUtils.isEmpty(linkApplyReq.getLinkUrl())) {
|
||||
throw new MyException(ResponseEnum.PARAMETERS_ERROR);
|
||||
}
|
||||
// 链接不合法
|
||||
if (!RegexUtil.emailMatch(linkApplyReq.getEmail())) {
|
||||
throw new MyException(ResponseEnum.PARAMETERS_EMAIL_ERROR);
|
||||
}
|
||||
if (!RegexUtil.urlMatch(linkApplyReq.getLinkUrl()) || !RegexUtil.urlMatch(linkApplyReq.getUrl())) {
|
||||
throw new MyException(ResponseEnum.PARAMETERS_URL_ERROR);
|
||||
}
|
||||
if (!StringUtils.isEmpty(linkApplyReq.getIconPath()) && !RegexUtil.urlMatch(linkApplyReq.getIconPath())) {
|
||||
throw new MyException(ResponseEnum.PARAMETERS_URL_ERROR);
|
||||
}
|
||||
// 非强制字段 设置空
|
||||
if (StringUtils.isEmpty(linkApplyReq.getIconPath())) {
|
||||
linkApplyReq.setIconPath("");
|
||||
}
|
||||
// 抓取页面
|
||||
String resp = HttpUtil.getAfterRendering(linkApplyReq.getLinkUrl());
|
||||
if (resp == null) {
|
||||
throw new MyException(ResponseEnum.CANNOT_GET_DATA);
|
||||
}
|
||||
PartnerSite ps = new PartnerSite();
|
||||
if (resp.contains(SITE_URL)) {
|
||||
//包含站点
|
||||
BeanUtils.copyProperties(linkApplyReq, ps);
|
||||
ps.setNotification(false);
|
||||
ps.setOpen(false);
|
||||
boolean exists = partnerMapper.existsByUrl(linkApplyReq.getUrl());
|
||||
if (!exists) {
|
||||
partnerMapper.insert(ps);
|
||||
} else {
|
||||
ps.setId(partnerMapper.findByUrl(linkApplyReq.getUrl()).getId());
|
||||
}
|
||||
SimpleMailMessage smm = new SimpleMailMessage();
|
||||
smm.setSubject("友链申请");
|
||||
smm.setText("有一条友链申请" + (exists ? "，已存在的友链链接" : "") + "，[\n" + linkApplyReq + "\n]");
|
||||
smm.setTo(SITE_EMAIL);
|
||||
smm.setSentDate(new Date());
|
||||
mailService.send(smm);
|
||||
} else {
|
||||
// 不包含站点
|
||||
String uuid;
|
||||
ObjectMapper mapper = new ObjectMapper();
|
||||
if (redisUtil.hasKey(linkApplyReq.getUrl())) {
|
||||
uuid = redisUtil.get(linkApplyReq.getUrl());
|
||||
if (!redisUtil.hasKey(uuid)) {
|
||||
redisUtil.setEx(uuid, mapper.writeValueAsString(linkApplyReq), 10, TimeUnit.MINUTES);
|
||||
}
|
||||
} else {
|
||||
uuid = UUID.randomUUID().toString().replaceAll("-", "");
|
||||
redisUtil.setEx(uuid, mapper.writeValueAsString(linkApplyReq), 10, TimeUnit.MINUTES);
|
||||
redisUtil.setEx(linkApplyReq.getUrl(), uuid, 10, TimeUnit.MINUTES);
|
||||
}
|
||||
throw new MyException(ResponseEnum.APPLY_LINK_NO_ADD_THIS_SITE, null, uuid);
|
||||
}
|
||||
return ps;
|
||||
}
|
||||
|
||||
@SneakyThrows
|
||||
@Override
|
||||
public String reapply(String key) {
|
||||
if (!redisUtil.hasKey(key)) {
|
||||
throw new MyException(ResponseEnum.DATA_EXPIRED);
|
||||
}
|
||||
String s = redisUtil.get(key);
|
||||
ObjectMapper mapper = new ObjectMapper();
|
||||
LinkApplyReq linkApplyReq = mapper.readValue(s, LinkApplyReq.class);
|
||||
if (linkApplyReq == null) {
|
||||
throw new MyException(ResponseEnum.DATA_NOT_EXIST);
|
||||
}
|
||||
SimpleMailMessage smm = new SimpleMailMessage();
|
||||
smm.setSubject("友链申请");
|
||||
smm.setText("有一条未抓取到信息的友链申请，[\n" + linkApplyReq + "\n]");
|
||||
smm.setTo(SITE_EMAIL);
|
||||
smm.setSentDate(new Date());
|
||||
mailService.send(smm);
|
||||
redisUtil.delete(key);
|
||||
redisUtil.delete(linkApplyReq.getUrl());
|
||||
return "success";
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user