DelayQueue实现延迟任务

1.构建延迟消息体

@Data
public class DelayedMessage implements Delayed {

    private final long delayTime;
    private final long expire;

    public DelayedMessage(long delayTime){
        this.delayTime=delayTime;
        this.expire=delayTime+System.currentTimeMillis();
    }


    @Override
    public long getDelay(TimeUnit unit) {
        return unit.convert(this.expire - System.currentTimeMillis(), TimeUnit.MILLISECONDS);
    }

    @Override
    public int compareTo(Delayed o) {
        if (this.getDelay(TimeUnit.MILLISECONDS) > o.getDelay(TimeUnit.MILLISECONDS)) {
            return 1;
        } else if (this.getDelay(TimeUnit.MILLISECONDS) < o.getDelay(TimeUnit.MILLISECONDS)) {
            return -1;
        } else {
            return 0;
        }
    }
}

2.设置延时时间执行延时任务

@Service
@Slf4j
public class DelayedMessageServiceImpl implements DelayedMessageService {

    @Override
    public void sendDelayedMessage(long delayedTime) {
        try{
            DelayQueue delayQueue =new DelayQueue<>();
            delayQueue.put(new DelayedMessage(delayedTime));
            while(!delayQueue.isEmpty()){
                DelayedMessage DelayedMessage=delayQueue.take();
                //执行延时任务
            }
        }catch(Exception e){
            log.error("_DelayedMessageService_failed_exception={}",e.getMessage());
        }
    }
}

3.异步调用延时任务

CompletableFuture completableFuture =asyncService.runAsync(()->
                    {
                        delayedMessageService.sendDelayedMessage(1000*60*2L);
                    }
            );

你可能感兴趣的:(Java,java,后端)