控制數(shù)據(jù)采樣頻率工具類

@Component
public class SamplingHelper implements InitializingBean {

    public static final long MAX_COUNT = 100000L;

    private Map<SamplingItem, AtomicInteger> counterMap = new EnumMap<>(SamplingItem.class);

    @Override
    public void afterPropertiesSet() {
        counterMap = Arrays.stream(SamplingItem.values())
                .collect(Collectors.toMap(
                        o -> o,
                        o -> new AtomicInteger(0)
                ));
    }

    public void doSample(SamplingItem samplingItem, Runnable runnable) {
        this.doSample(samplingItem, () -> {
            runnable.run();
            return null;
        });
    }

    public <T> T doSample(SamplingItem samplingItem, Supplier<T> supplier) {
        boolean allowSampling = VivoConfigManager.getBoolean("ALLOW_SAMPLE_COLLECTION", false);
        if (!allowSampling) {
            return null;
        }

        AtomicInteger counter = counterMap.get(samplingItem);
        if (counter == null) {
            return null;
        }

        int count = counter.incrementAndGet();
        try {
            boolean accept = count % samplingItem.samplingRate == 0;
            if (accept) {
                return supplier.get();
            } else {
                return null;
            }
        } finally {
            // 當(dāng)累加到一定數(shù)量的時(shí)候重置，避免溢出
            if (count > MAX_COUNT) {
                counter.set(0);
            }
        }
    }

    public enum SamplingItem {
        /**
         * 采樣類型
         */
        CONNECTION(15),
        TIME_COLLECT(20);

        private final int samplingRate;

        SamplingItem(int samplingRate) {
            this.samplingRate = samplingRate;
        }
    }
}