ru.taskurotta.server.config.expiration.impl.TimeoutWithFixedRetryPolicy Maven / Gradle / Ivy

Go to download
package ru.taskurotta.server.config.expiration.impl;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.Map;
import java.util.Properties;
import java.util.UUID;
import java.util.concurrent.ConcurrentHashMap;


public class TimeoutWithFixedRetryPolicy extends TimeoutPolicy {

    private static final Logger logger = LoggerFactory.getLogger(TimeoutWithFixedRetryPolicy.class);

    public static final String RETRY = "retry";
    private Map expirations = new ConcurrentHashMap();

    private int retry = -1;

    public TimeoutWithFixedRetryPolicy(Properties props) {
        super(props);
        if (props!=null && !props.isEmpty()) {
            if (props.containsKey(RETRY)) {
                this.retry = Integer.valueOf(props.get(RETRY).toString());
            }
        }
        logger.debug("TimeoutWithFixedRetryPolicy created. retry[{}], timeout[{}], timeUnit[{}]", this.retry, this.timeout, this.timeUnit);
    }

    @Override
    public boolean readyToRecover(UUID taskId, UUID processId) {
        boolean result = true;
        if (retry > 0) {
            Integer taskRetry = expirations.get(taskId);

            if (taskRetry == null || taskRetry < retry) {
                expirations.put(taskId, taskRetry == null ? 1 : taskRetry + 1);
            } else {
                result = false;
                logger.error("Task[{}] expiration policy commit failed: Task has been already retried for [{}]/[{}] times", taskId, taskRetry, retry);
            }

        }
        return result;

    }

    @Override
    public long getExpirationTime(UUID taskId, UUID processId, long forTime) {
        //forTime + fixed timeout
        return forTime + timeUnit.toMillis(timeout);
    }

}