com.newrelic.agent.instrumentation.weaver.extension.ExtensionHolderFactoryImpl Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of newrelic-agent Show documentation

Jar required to run with a java application to monitor performance.

The newest version!

/*
 *
 *  * Copyright 2020 New Relic Corporation. All rights reserved.
 *  * SPDX-License-Identifier: Apache-2.0
 *
 */

package com.newrelic.agent.instrumentation.weaver.extension;

import com.github.benmanes.caffeine.cache.Cache;
import com.github.benmanes.caffeine.cache.Caffeine;
import com.newrelic.agent.bridge.AgentBridge;
import com.newrelic.agent.bridge.ExtensionHolder;
import com.newrelic.agent.bridge.ExtensionHolderFactory;

import java.util.function.Supplier;
import java.util.logging.Level;

/**
 * Implementation for creating an {@link ExtensionHolder}
 * 
 * @see ExtensionHolderImpl
 */
public class ExtensionHolderFactoryImpl implements ExtensionHolderFactory {
    @Override
    public  ExtensionHolder build() {
        return new ExtensionHolderImpl<>();
    }

    /**
     * Uses a caffeine cache to store weaver extension classes.
     */
    public static class ExtensionHolderImpl implements ExtensionHolder {
        // @formatter:off
        private final Cache instanceCache = Caffeine.newBuilder()
                .initialCapacity(32)
                .weakKeys()
                .executor(Runnable::run)
                .build();
        // @formatter:on

        @Override
        public T getAndRemoveExtension(Object instance) {
            return instanceCache.asMap().remove(instance);
        }

        @Override
        public T getExtension(Object instance, Supplier valueLoader) {
            try {
                return instanceCache.get(instance, k -> valueLoader.get());
            } catch (RuntimeException e) {
                AgentBridge.getAgent().getLogger().log(Level.FINE, e, "Unable to load extension class for {0}",
                        instance.getClass().getName());
                throw e;
            }
        }
    }
}