com.liubs.shadowrpcfly.server.handler.QpsStatHandler Maven / Gradle / Ivy

Go to download
package com.liubs.shadowrpcfly.server.handler;

import com.liubs.shadowrpcfly.logging.Logger;
import io.netty.channel.ChannelDuplexHandler;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelPromise;


import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.atomic.AtomicLong;

/**
 * @author Liubsyy
 * @date 2023/12/29
 */
public class QpsStatHandler extends ChannelDuplexHandler {

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

    //请求数量
    private static AtomicLong activeRequests = new AtomicLong(0);

    //每秒请求量
    private static AtomicInteger perSecondsRequests = new AtomicInteger(0);

    // 定时任务，用于计算QPS
    static {
        ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1);
        scheduler.scheduleAtFixedRate(() -> {
            int qps = perSecondsRequests.getAndSet(0);
            if(qps > 0) {
                logger.info("Current QPS: " + qps);
            }

            // 可以进一步将QPS记录到日志或监控系统
        }, 1, 1, TimeUnit.SECONDS);
    }


    @Override
    public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise) throws Exception {
        ctx.write(msg, promise).addListener((f)->{
            if(f.isSuccess()) {
                perSecondsRequests.incrementAndGet();
                activeRequests.incrementAndGet();
            }
        });
    }
}