/*
 * Copyright The WildFly Authors
 * SPDX-License-Identifier: Apache-2.0
 */

package org.jboss.as.host.controller;

import java.util.concurrent.TimeUnit;

/**
 * Policy used to automatically try to reconnect to a crashed master HC.
 *
 * @author Emanuel Muckenhuber
 */
interface ReconnectPolicy {

    /**
     * Respawn a process.
     *
     * @param count
     */
    void wait(int count) throws InterruptedException;

    ReconnectPolicy CONNECT = new ReconnectPolicy() {

        @Override
        public void wait(int count) throws InterruptedException {
            final int waitPeriod;
            if (count < 5) {
                waitPeriod = 1;
            } else if (count >= 5 && count < 10) {
                waitPeriod = 3;
            } else if (count >= 10 && count < 15) {
                waitPeriod = 10;
            } else {
                waitPeriod = 20;
            }
            TimeUnit.SECONDS.sleep(waitPeriod);
        }
    };

    ReconnectPolicy RECONNECT = new ReconnectPolicy() {

        private static final int MAX_WAIT = 15;

        @Override
        public void wait(final int count) throws InterruptedException {
            final int waitPeriod = Math.min((count * count), MAX_WAIT);
            TimeUnit.SECONDS.sleep(waitPeriod);
        }
    };

}
