/* * Copyright (c) 2018-2028, DreamLu All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are met: * * Redistributions of source code must retain the above copyright notice, * this list of conditions and the following disclaimer. * Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * Neither the name of the dreamlu.net developer nor the names of its * contributors may be used to endorse or promote products derived from * this software without specific prior written permission. * Author: DreamLu 卢春梦 (596392912@qq.com) */ package org.springblade.core.metrics.sentinel; import com.alibaba.csp.sentinel.metric.extension.MetricExtension; import com.alibaba.csp.sentinel.slots.block.BlockException; import io.micrometer.core.instrument.Metrics; import io.micrometer.core.instrument.Tags; import org.springblade.core.auto.service.AutoService; import java.util.concurrent.TimeUnit; import java.util.concurrent.atomic.AtomicLong; /** * Sentinel Metrics Extension * * @author L.cm */ @AutoService(MetricExtension.class) public class SentinelMetricsExtension implements MetricExtension { public static final String PASS_REQUESTS_TOTAL = "sentinel_pass_requests_total"; public static final String BLOCK_REQUESTS_TOTAL = "sentinel_block_requests_total"; public static final String SUCCESS_REQUESTS_TOTAL = "sentinel_success_requests_total"; public static final String EXCEPTION_REQUESTS_TOTAL = "sentinel_exception_requests_total"; public static final String REQUESTS_LATENCY_SECONDS = "sentinel_requests_latency_seconds"; public static final String CURRENT_THREADS = "sentinel_current_threads"; public static final String DEFAULT_TAT_NAME = "resource"; private final AtomicLong CURRENT_THREAD_COUNT = new AtomicLong(0); @Override public void addPass(String resource, int n, Object... args) { Metrics.counter(PASS_REQUESTS_TOTAL, DEFAULT_TAT_NAME, resource).increment(n); } @Override public void addBlock(String resource, int n, String origin, BlockException ex, Object... args) { Metrics.counter(BLOCK_REQUESTS_TOTAL, resource, ex.getClass().getSimpleName(), ex.getRuleLimitApp(), origin).increment(n); } @Override public void addSuccess(String resource, int n, Object... args) { Metrics.counter(SUCCESS_REQUESTS_TOTAL, DEFAULT_TAT_NAME, resource).increment(n); } @Override public void addException(String resource, int n, Throwable throwable) { Metrics.counter(EXCEPTION_REQUESTS_TOTAL, DEFAULT_TAT_NAME, resource).increment(n); } @Override public void addRt(String resource, long rt, Object... args) { Metrics.timer(REQUESTS_LATENCY_SECONDS, DEFAULT_TAT_NAME, resource).record(rt, TimeUnit.MICROSECONDS); } @Override public void increaseThreadNum(String resource, Object... args) { Tags tags = Tags.of(DEFAULT_TAT_NAME, resource); Metrics.gauge(CURRENT_THREADS, tags, CURRENT_THREAD_COUNT, AtomicLong::incrementAndGet); } @Override public void decreaseThreadNum(String resource, Object... args) { Tags tags = Tags.of(DEFAULT_TAT_NAME, resource); Metrics.gauge(CURRENT_THREADS, tags, CURRENT_THREAD_COUNT, AtomicLong::decrementAndGet); } }