抢红包活动所用到的一些多线程
public class RedPacketTask {
private final static Logger logger = LoggerFactory.getLogger(RedPacketTask.class);
@Autowired
private ThreadPoolTaskScheduler threadPoolTaskScheduler;
@Autowired
private ActivityRedService activityRedService;
/** 存放所有启动定时任务对象，极其重要 **/
private ConcurrentHashMap<String, ScheduledFuture<?>> scheduleMap = new ConcurrentHashMap<>();
/** 扫描有效活动并开启 **/
public void startActivity() {
ScheduledFuture<?> future = threadPoolTaskScheduler.schedule(new ActivityRunnable(), new CronTrigger("0/1 * * * * *"));
scheduleMap.put("activity", future);
}
private class ActivityRunnable implements Runnable {
@Override
public void run() {
//从redis中获取活动信息
byte[] activityRedByte = RedisService.getRedisService().getByte(RedisKeyConstants.ACTIVITY_RED_INFO);
List<ActivityRed> activityRedList = FstSerializer.getInstance().deserialize(activityRedByte,ActivityRed.class);
List<ActivityRed> activityRedResult = new ArrayList<>();
activityRedList.stream().forEach(activityRed ->{
boolean flag = accept(activityRed);
if(flag){
activityRedResult.add(activityRed);
}
});
activityRedList.removeAll(activityRedResult);
byte[] activityRedBytes = FstSerializer.getInstance().serialize(activityRedList);
RedisService.getRedisService().setByte(RedisKeyConstants.ACTIVITY_RED_INFO, activityRedBytes);
RedisService.getRedisService().persist(RedisKeyConstants.ACTIVITY_RED_INFO);
}
private boolean accept(ActivityRed activityRed) {
int taskInterval = activityRed.getTaskInterval();
long period = taskInterval*60*1000;
Date startTime = activityRed.getStartTime();
Date endTime = activityRed.getEndTime();
Date currentDate = new Date();
//根据ID获取执行任务
ScheduledFuture<?> future = scheduleMap.get(activityRed.getActivityRedId());
//判断是否存在已开始的执行任务
if (DateUtil.compareDate(startTime, currentDate)<=0 && future == null) {
ScheduledFuture<?> addFuture = threadPoolTaskScheduler.scheduleAtFixedRate(
new RedPacketRunnable(activityRed), startTime, period);
scheduleMap.put(activityRed.getActivityRedId(), addFuture);
}
if(DateUtil.compareDate(endTime, currentDate)<=0 ){
//移除过期的活动
if (future != null) {
future.cancel(true);
}
return true;
}
return false;
}
}
private class RedPacketRunnable implements Runnable {
private ActivityRed activityRed;
RedPacketRunnable(ActivityRed activityRed){
this.activityRed = activityRed;
}
@Override
public void run() {
//设置下一次抢红包时间
Calendar calendar = Calendar.getInstance();
calendar.setTime(new Date());
calendar.set(Calendar.MINUTE, calendar.get(Calendar.MINUTE)+activityRed.getTaskInterval());
logger.debug("将【{}】红包金额【{}】转到可用金额【{}】中并设置下一次抢红包时间【{}】",
activityRed.getActivityName(),
activityRed.getAmount(),
activityRed.getAvailableAmount(),
DateUtil.formatDate(calendar.getTime(),"yyyy-MM-dd HH:mm:ss"));
activityRed.setAvailableAmount(activityRed.getAmount()+activityRed.getAvailableAmount());
activityRed.setAmount(new Double(0));
activityRedService.modActivityRed(activityRed);
RedisService.getRedisService().set(activityRed.getActivityRedId()+":"+"nextTime",
DateUtil.formatDate(calendar.getTime(),"yyyy-MM-dd HH:mm:ss"));
RedisService.getRedisService().expire(activityRed.getActivityRedId()+":"+"nextTime",activityRed.getTaskInterval()*60);
//设置抢红时间
RedisService.getRedisService().set(activityRed.getActivityRedId()+":"+"robTime", activityRed.getRobTimes().toString());
RedisService.getRedisService().expire(activityRed.getActivityRedId()+":"+"robTime", activityRed.getRobTimes());
}
}
