com.github.panhongan.utils.throttling.Throttling Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of lalaluplus-commons Show documentation
lalaluplus java commons
The newest version!
package com.github.panhongan.utils.throttling;

import com.github.panhongan.utils.function.ThrowableFunction;
import java.util.function.Function;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Throttling
 *
 * @author lalalu plus
 * @since 2019.6.5
 */

public class Throttling {

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

    /**
     * @param throttlingCounter ThrottlingCounter
     * @param function Function
     * @param input Input
     * @param throttlingResult Output
     * @param  Input type
     * @param  Output type
     * @return O Output
     */
    public static  O throttling(ThrottlingCounter throttlingCounter,
                                      Function function,
                                      I input,
                                      O throttlingResult) {
        try {
            if (throttlingCounter.tryEnter()) {
                return function.apply(input);
            } else {
                logger.warn("trigger throttling : {}", throttlingCounter.toString());
                return throttlingResult;
            }
        } finally {
            throttlingCounter.leave();
        }
    }

    /**
     * @param throttlingCounter ThrottlingCounter
     * @param function Function
     * @param input Input
     * @param throttlingResult Output
     * @param  Input type
     * @param  Output type
     * @param  Throwable type
     * @return O Output
     * @throws T Throwable object
     */
    public static  O throttling(ThrottlingCounter throttlingCounter,
                                                           ThrowableFunction function,
                                                           I input,
                                                           O throttlingResult) throws T {
        try {
            if (throttlingCounter.tryEnter()) {
                return function.apply(input);
            } else {
                logger.warn("trigger throttling : {}", throttlingCounter.toString());
                return throttlingResult;
            }
        } catch (Throwable t) {
            logger.warn("", t);
            throw t;
        } finally {
            throttlingCounter.leave();
        }
    }
}