com.kjetland.jackson.jsonSchema.SubclassesResolver Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mbknor-jackson-jsonschema-java Show documentation

mbknor-jackson-jsonSchema

There is a newer version: 1.1.2

Show newest version

package com.kjetland.jackson.jsonSchema;

import io.github.classgraph.ClassGraph;
import io.github.classgraph.ScanResult;
import java.util.List;
import lombok.extern.slf4j.Slf4j;

@Slf4j
public class SubclassesResolver {

    private ScanResult scanResult;
    
    public SubclassesResolver() {
        this(null);
    }
    
    public SubclassesResolver (List packagesToScan, List classesToScan) {
        this(buildClassGraph(packagesToScan, classesToScan));
    }
    
    public SubclassesResolver (ClassGraph classGraph) {
        if (classGraph == null) {
            log.debug("Entire classpath will be scanned because SubclassesResolver is not configured. See "
                    + "https://github.com/mbknor/mbknor-jackson-jsonSchema#subclass-resolving-using-reflection");
            classGraph = new ClassGraph();
        }

        scanResult = classGraph.enableClassInfo().scan();
    }

    public List> getSubclasses(Class clazz) {
        if (clazz.isInterface())
            return scanResult.getClassesImplementing(clazz.getName()).loadClasses();
        else
            return scanResult.getSubclasses(clazz.getName()).loadClasses();
    }
    
    static private ClassGraph buildClassGraph(List packagesToScan, List classesToScan) {
        if (packagesToScan == null && classesToScan == null)
            return null;

        ClassGraph classGraph = new ClassGraph();

        if (packagesToScan != null)
            classGraph.whitelistPackages(packagesToScan.toArray(String[]::new));

        if (classesToScan != null)
            classGraph.whitelistClasses(classesToScan.toArray(String[]::new));

        return classGraph;
    }
}