kanela.agent.util.classloader.ClassLoaderNameMatcher Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of kanela-agent Show documentation
The Kamon Instrumentation Agent
There is a newer version: 1.0.18
/*
 * =========================================================================================
 * Copyright © 2013-2018 the kamon project 
 *
 * Licensed 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 kanela.agent.util.classloader;

import io.vavr.control.Option;
import kanela.agent.api.instrumentation.classloader.ClassLoaderRefiner;
import kanela.agent.util.collection.ConcurrentReferenceHashMap;
import lombok.EqualsAndHashCode;
import lombok.Value;
import net.bytebuddy.matcher.ElementMatcher;
import net.bytebuddy.matcher.ElementMatchers;

import java.util.stream.Collectors;

@Value
@EqualsAndHashCode(callSuper = false)
public class ClassLoaderNameMatcher extends ElementMatcher.Junction.AbstractBase {

    String name;

    private ClassLoaderNameMatcher(String name) {
        this.name = name;
    }

    public static ElementMatcher.Junction.AbstractBase withName(String name) {
        return new ClassLoaderNameMatcher(name);
    }

    public static ElementMatcher.Junction.AbstractBase isReflectionClassLoader() {
        return new ClassLoaderNameMatcher("sun.reflect.DelegatingClassLoader");
    }

    public static ElementMatcher.Junction.AbstractBase isGroovyClassLoader() {
        return new ClassLoaderNameMatcher("org.codehaus.groovy.runtime.callsite.CallSiteClassLoader");
    }

    public static ElementMatcher.Junction.AbstractBase isSBTClassLoader() {
        return new ClassLoaderNameMatcher("xsbt.boot.LibraryClassLoader");
    }

    public static ElementMatcher.Junction.AbstractBase isSBTPluginClassLoader() {
        return new ClassLoaderNameMatcher("sbt.internal.PluginManagement$PluginClassLoader");
    }

    public static ElementMatcher.Junction.AbstractBase isSBTCompilerClassLoader() {
        return new ScalaCompilerClassLoaderMatcher();
    }

    public static ElementMatcher.Junction.AbstractBase isKanelaClassLoader() {
        return new ClassLoaderNameMatcher(ChildFirstURLClassLoader.class.getName());
    }

    @Override
    public boolean matches(ClassLoader target) {
        return target != null && name.equals(target.getClass().getName());
    }


    @Value
    @EqualsAndHashCode(callSuper = false)
    public static class RefinedClassLoaderMatcher extends ElementMatcher.Junction.AbstractBase {

        ConcurrentReferenceHashMap cache = new ConcurrentReferenceHashMap<>();
        ClassLoaderRefiner refiner;

        public static ElementMatcher from(Option classLoaderRefiner){
            return classLoaderRefiner
                    .map(refiner -> (ElementMatcher) new RefinedClassLoaderMatcher(refiner))
                    .getOrElse(ElementMatchers::any);
        }

        @Override
        public boolean matches(final ClassLoader classLoader) {
            if (classLoader == null) return false;
            return cache.computeIfAbsent(classLoader, (key) -> compute(refiner, classLoader));
        }

        private boolean compute(ClassLoaderRefiner classRefiner, ClassLoader classLoader) {
            return !classRefiner
                    .refiners()
                    .stream()
                    .map(refiner -> AnalyzedClass.from(refiner, classLoader).match())
                    .collect(Collectors.toList())
                    .contains(false);
        }
    }
}