org.killbill.billing.util.metrics.JCacheGaugeFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of killbill-util Show documentation
There is a newer version: 0.24.11
/*
 * Copyright 2020-2022 Equinix, Inc
 * Copyright 2014-2022 The Billing Project, LLC
 *
 * The Billing Project licenses this file to you 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 org.killbill.billing.util.metrics;

import java.lang.management.ManagementFactory;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.Set;

import javax.cache.management.CacheStatisticsMXBean;
import javax.management.MalformedObjectNameException;
import javax.management.ObjectInstance;
import javax.management.ObjectName;

import org.killbill.commons.metrics.api.Gauge;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

// Initially forked from Dropwizard Metrics (Apache License 2.0).
// Copyright (c) 2010-2013 Coda Hale, Yammer.com, 2014-2021 Dropwizard Team
public class JCacheGaugeFactory {

    private static final String PROP_METRIC_REG_JCACHE_STATISTICS = "jcache.statistics.";

    private static final String M_BEAN_COORDINATES = "javax.cache:type=CacheStatistics,CacheManager=*,Cache=";

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

    public static Map> forCache(final String cacheName) {
        final Set cacheBeans = getCacheBeans(cacheName);
        if (cacheBeans == null || cacheBeans.isEmpty()) {
            logger.warn("Unable to retrieve statis for cache {}. Are JCache statistics enabled?", cacheName);
            return Collections.emptyMap();
        }
        if (cacheBeans.size() > 1) {
            logger.warn("Multiple CacheManager detected for cache {}", cacheName);
        }
        final ObjectInstance cacheBean = cacheBeans.iterator().next();
        final ObjectName objectName = cacheBean.getObjectName();

        final List availableStatsNames = retrieveStatsNames();

        final Map> gauges = new HashMap<>(cacheBeans.size() * availableStatsNames.size());
        for (final String statsName : availableStatsNames) {
            final Gauge