club.zhcs.lina.starter.monitor.metrics.druid.DruidMetrics Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of lina-spring-boot-starter Show documentation
spring-boot 自动配置
There is a newer version: 3.3.4
Show newest version
package club.zhcs.lina.starter.monitor.metrics.druid;

import java.util.Collections;
import java.util.Map;

import com.alibaba.druid.pool.DruidDataSource;
import com.alibaba.druid.stat.JdbcConnectionStat;
import com.alibaba.druid.stat.JdbcDataSourceStat;

import io.micrometer.core.instrument.Gauge;
import io.micrometer.core.instrument.MeterRegistry;
import io.micrometer.core.instrument.Tag;
import io.micrometer.core.instrument.binder.MeterBinder;
import lombok.RequiredArgsConstructor;

/**
 * 
 * @author Kerbores([email protected])
 *
 */
@RequiredArgsConstructor
public class DruidMetrics implements MeterBinder {

    public static final String DRUID_METRIC_NAME_PREFIX = "druid";

    private static final String METRIC_CATEGORY = "name";
    private static final String METRIC_NAME_CONNECT_MAX_TIME = DRUID_METRIC_NAME_PREFIX
                                                               + ".connections.connect.max.time";
    private static final String METRIC_NAME_ALIVE_MAX_TIME = DRUID_METRIC_NAME_PREFIX + ".connections.alive.max.time";
    private static final String METRIC_NAME_ALIVE_MIN_TIME = DRUID_METRIC_NAME_PREFIX + ".connections.alive.min.time";

    private static final String METRIC_NAME_CONNECT_COUNT = DRUID_METRIC_NAME_PREFIX + ".connections.connect.count";
    private static final String METRIC_NAME_ACTIVE_COUNT = DRUID_METRIC_NAME_PREFIX + ".connections.active.count";
    private static final String METRIC_NAME_CLOSE_COUNT = DRUID_METRIC_NAME_PREFIX + ".connections.close.count";
    private static final String METRIC_NAME_ERROR_COUNT = DRUID_METRIC_NAME_PREFIX + ".connections.error.count";
    private static final String METRIC_NAME_CONNECT_ERROR_COUNT = DRUID_METRIC_NAME_PREFIX
                                                                  + ".connections.connect.error.count";
    private static final String METRIC_NAME_COMMIT_COUNT = DRUID_METRIC_NAME_PREFIX + ".connections.commit.count";
    private static final String METRIC_NAME_ROLLBACK_COUNT = DRUID_METRIC_NAME_PREFIX + ".connections.rollback.count";

    private final Map druidDataSourceMap;
    private final Iterable tags;

    public DruidMetrics(Map druidDataSourceMap) {
        this(druidDataSourceMap, Collections.emptyList());
    }

    @Override
    public void bindTo(MeterRegistry meterRegistry) {
        druidDataSourceMap.forEach((name, dataSource) -> {
            JdbcDataSourceStat dsStats = dataSource.getDataSourceStat();
            JdbcConnectionStat connectionStat = dsStats.getConnectionStat();
            // time
            Gauge.builder(METRIC_NAME_CONNECT_MAX_TIME, connectionStat, JdbcConnectionStat::getConnectMillisMax)
                 .description("Connection connect max time")
                 .tags(tags)
                 .tag(METRIC_CATEGORY, name)
                 .baseUnit("ms")
                 .register(meterRegistry);
            Gauge.builder(METRIC_NAME_ALIVE_MAX_TIME, connectionStat, JdbcConnectionStat::getAliveMillisMax)
                 .description("Connection alive max time")
                 .tags(tags)
                 .tag(METRIC_CATEGORY, name)
                 .baseUnit("ms")
                 .register(meterRegistry);
            Gauge.builder(METRIC_NAME_ALIVE_MIN_TIME, connectionStat, JdbcConnectionStat::getAliveMillisMin)
                 .description("Connection alive min time")
                 .tags(tags)
                 .tag(METRIC_CATEGORY, name)
                 .baseUnit("ms")
                 .register(meterRegistry);
            // count
            Gauge.builder(METRIC_NAME_ACTIVE_COUNT, connectionStat, JdbcConnectionStat::getActiveCount)
                 .description("Connection active count")
                 .tags(tags)
                 .tag(METRIC_CATEGORY, name)
                 .register(meterRegistry);
            Gauge.builder(METRIC_NAME_CONNECT_COUNT, connectionStat, JdbcConnectionStat::getConnectCount)
                 .description("Connection connect count")
                 .tags(tags)
                 .tag(METRIC_CATEGORY, name)
                 .register(meterRegistry);
            Gauge.builder(METRIC_NAME_CLOSE_COUNT, connectionStat, JdbcConnectionStat::getCloseCount)
                 .description("Connection close count")
                 .tags(tags)
                 .tag(METRIC_CATEGORY, name)
                 .register(meterRegistry);
            Gauge.builder(METRIC_NAME_ERROR_COUNT, connectionStat, JdbcConnectionStat::getErrorCount)
                 .description("Connection error count")
                 .tags(tags)
                 .tag(METRIC_CATEGORY, name)
                 .register(meterRegistry);
            Gauge.builder(METRIC_NAME_CONNECT_ERROR_COUNT, connectionStat, JdbcConnectionStat::getConnectErrorCount)
                 .description("Connection connect error count")
                 .tags(tags)
                 .tag(METRIC_CATEGORY, name)
                 .register(meterRegistry);
            Gauge.builder(METRIC_NAME_COMMIT_COUNT, connectionStat, JdbcConnectionStat::getCommitCount)
                 .description("Connecting commit count")
                 .tags(tags)
                 .tag(METRIC_CATEGORY, name)
                 .register(meterRegistry);
            Gauge.builder(METRIC_NAME_ROLLBACK_COUNT, connectionStat, JdbcConnectionStat::getRollbackCount)
                 .description("Connection rollback count")
                 .tags(tags)
                 .tag(METRIC_CATEGORY, name)
                 .register(meterRegistry);
        });
    }
}