87 lines
3.9 KiB
Java
87 lines
3.9 KiB
Java
package com.peanut.modules.job.task;
|
||
|
||
import cn.hutool.core.date.DateUtil;
|
||
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
||
import com.peanut.common.utils.MailUtil;
|
||
import com.peanut.config.DelayQueueConfig;
|
||
import com.peanut.modules.common.entity.MyUserEntity;
|
||
import com.peanut.modules.common.entity.UserVip;
|
||
import com.peanut.modules.common.service.MyUserService;
|
||
import com.peanut.modules.master.service.UserVipService;
|
||
import org.apache.commons.lang.StringUtils;
|
||
import org.apache.commons.lang.time.DateUtils;
|
||
import org.slf4j.Logger;
|
||
import org.slf4j.LoggerFactory;
|
||
import org.springframework.amqp.core.MessagePostProcessor;
|
||
import org.springframework.amqp.rabbit.core.RabbitTemplate;
|
||
import org.springframework.beans.factory.annotation.Autowired;
|
||
import org.springframework.stereotype.Component;
|
||
|
||
import java.util.Date;
|
||
import java.util.List;
|
||
|
||
@Component("userVipTask")
|
||
public class UserVipTask implements ITask{
|
||
private Logger logger = LoggerFactory.getLogger(getClass());
|
||
|
||
@Autowired
|
||
private UserVipService vipService;
|
||
@Autowired
|
||
private MyUserService userService;
|
||
@Autowired
|
||
private RabbitTemplate rabbitTemplate;
|
||
|
||
@Override
|
||
public void run(String params) {
|
||
LambdaQueryWrapper<UserVip> wrapper = new LambdaQueryWrapper();
|
||
wrapper.eq(UserVip::getState,0);
|
||
List<UserVip> list = vipService.list(wrapper);
|
||
if (list.size() > 0) {
|
||
for (UserVip userVip : list) {
|
||
MyUserEntity user = userService.getById(userVip.getUserId());
|
||
if (userVip.getEndTime().getTime()<new Date().getTime()){
|
||
userVip.setState(1);
|
||
vipService.saveOrUpdate(userVip);
|
||
user.setVip("0");
|
||
userService.saveOrUpdate(user);
|
||
// }else if (DateUtils.isSameDay(DateUtils.addDays(new Date(),60),userVip.getEndTime())) {
|
||
// try {
|
||
// if (StringUtils.isNotEmpty(user.getTel())){
|
||
// userService.sendCodeForRegister(user.getTel(),"VIPWillExpire",86);
|
||
// }else if (StringUtils.isNotEmpty(user.getEmail())) {
|
||
// MailUtil.sendMail("太湖超V提醒","超V还有2月到期",user.getEmail());
|
||
// }
|
||
// }catch (Exception e) {
|
||
// logger.error("userVipTask定时任务超V提醒出错，错误为：{}", e.getMessage());
|
||
// }
|
||
// }else if (DateUtils.isSameDay(DateUtils.addDays(new Date(),30),userVip.getEndTime())) {
|
||
// try {
|
||
// if (StringUtils.isNotEmpty(user.getTel())){
|
||
// userService.sendCodeForRegister(user.getTel(),"VIPWillExpire",86);
|
||
// }else if (StringUtils.isNotEmpty(user.getEmail())) {
|
||
// MailUtil.sendMail("太湖超V提醒","超V还有一月到期",user.getEmail());
|
||
// }
|
||
// }catch (Exception e) {
|
||
// logger.error("userVipTask定时任务超V提醒出错，错误为：{}", e.getMessage());
|
||
// }
|
||
}else if (DateUtils.isSameDay(DateUtils.addDays(new Date(),2),userVip.getEndTime())) {
|
||
rabbitTemplate.convertAndSend(
|
||
DelayQueueConfig.USERVIP_EXCHANGE,
|
||
DelayQueueConfig.USERVIP_ROUTING_KEY,
|
||
userVip.getId(),
|
||
messagePostProcessor(userVip.getEndTime().getTime()-new Date().getTime())
|
||
);
|
||
System.out.println("-------------------------会员还剩2天，加入队列");
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
private MessagePostProcessor messagePostProcessor(long date) {
|
||
return message -> {
|
||
message.getMessageProperties().setDelay((int)date);
|
||
return message;
|
||
};
|
||
}
|
||
}
|