All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.navercorp.pinpoint.grpc.logging.ThrottledLogger Maven / Gradle / Ivy

There is a newer version: 3.0.0
Show newest version
/*
 * Copyright 2020 NAVER Corp.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.navercorp.pinpoint.grpc.logging;

import org.slf4j.Logger;

import java.util.Objects;
import java.util.concurrent.atomic.AtomicLong;

/**
 * @author Woonduk Kang(emeroad)
 */
public class ThrottledLogger {
    private final Logger logger;
    private final long ratio;
    private final AtomicLong counter = new AtomicLong();

    public static ThrottledLogger getLogger(Logger logger, long ratio) {
        Objects.requireNonNull(logger, "logger");
        return new ThrottledLogger(logger, ratio);
    }

    private ThrottledLogger(Logger logger, long ratio) {
        this.logger = Objects.requireNonNull(logger, "logger");
        this.ratio = ratio;
    }

    private boolean checkLogCounter() {
        if (counter.getAndIncrement() % ratio == 0) {
            return true;
        }
        return false;
    }

    public long getCounter() {
        return counter.get();
    }

    public boolean isInfoEnabled() {
        return logger.isInfoEnabled();
    }

    public void info(String msg) {
        if (!logger.isInfoEnabled()) {
            return;
        }
        if (!checkLogCounter()) {
            return;
        }
        logger.info(msg);
    }

    public void info(String msg, Object arg) {
        if (!logger.isInfoEnabled()) {
            return;
        }
        if (!checkLogCounter()) {
            return;
        }
        logger.info(msg, arg);
    }

    public void info(String msg, Object arg1, Object arg2) {
        if (!logger.isInfoEnabled()) {
            return;
        }
        if (!checkLogCounter()) {
            return;
        }
        logger.info(msg, arg1, arg2);
    }

    public void info(String msg, Throwable t) {
        if (!logger.isInfoEnabled()) {
            return;
        }
        if (!checkLogCounter()) {
            return;
        }
        logger.info(msg, t);
    }

    public void info(String msg, Object... args) {
        if (!logger.isInfoEnabled()) {
            return;
        }
        if (!checkLogCounter()) {
            return;
        }
        logger.info(msg, args);
    }

    // debug -------------

    public boolean isDebugEnabled() {
        return logger.isDebugEnabled();
    }

    public void debug(String msg) {
        if (!logger.isDebugEnabled()) {
            return;
        }
        if (!checkLogCounter()) {
            return;
        }
        logger.debug(msg);
    }

    public void debug(String msg, Object arg) {
        if (!logger.isDebugEnabled()) {
            return;
        }
        if (!checkLogCounter()) {
            return;
        }
        logger.debug(msg, arg);
    }

    public void debug(String msg, Object arg1, Object arg2) {
        if (!logger.isDebugEnabled()) {
            return;
        }
        if (!checkLogCounter()) {
            return;
        }
        logger.debug(msg, arg1, arg2);
    }

    public void debug(String msg, Throwable t) {
        if (!logger.isDebugEnabled()) {
            return;
        }
        if (!checkLogCounter()) {
            return;
        }
        logger.debug(msg, t);
    }

    public void debug(String msg, Object... args) {
        if (!logger.isDebugEnabled()) {
            return;
        }
        if (!checkLogCounter()) {
            return;
        }
        logger.debug(msg, args);
    }

    // warn -------------

    public boolean isWarnEnabled() {
        return logger.isWarnEnabled();
    }

    public void warn(String msg) {
        if (!logger.isWarnEnabled()) {
            return;
        }
        if (!checkLogCounter()) {
            return;
        }
        logger.warn(msg);
    }

    public void warn(String msg, Object arg) {
        if (!logger.isWarnEnabled()) {
            return;
        }
        if (!checkLogCounter()) {
            return;
        }
        logger.warn(msg, arg);
    }

    public void warn(String msg, Object arg1, Object arg2) {
        if (!logger.isWarnEnabled()) {
            return;
        }
        if (!checkLogCounter()) {
            return;
        }
        logger.warn(msg, arg1, arg2);
    }

    public void warn(String msg, Throwable t) {
        if (!logger.isWarnEnabled()) {
            return;
        }
        if (!checkLogCounter()) {
            return;
        }
        logger.warn(msg, t);
    }

    public void warn(String msg, Object... args) {
        if (!logger.isWarnEnabled()) {
            return;
        }
        if (!checkLogCounter()) {
            return;
        }
        logger.warn(msg, args);
    }
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy