Substitutions needed to run Lucene 8.4.1 on a GraalVM native image

AttributeFactoryHelpers.java

package org.apache.lucene.util;

import java.lang.reflect.Constructor;
import java.lang.reflect.UndeclaredThrowableException;

public final class AttributeFactoryHelpers {
    static final AttributeFactory DEFAULT = new DefaultAttributeFactory();

    static final Constructor<? extends AttributeImpl> findAttributeImplCtor(Class<? extends AttributeImpl> clazz) {
        try {
            return clazz.getConstructor();
        } catch (NoSuchMethodException e) {
            throw new IllegalArgumentException("Cannot lookup accessible no-arg constructor for: " + clazz.getName(), e);
        }
    }

    static final class DefaultAttributeFactory extends AttributeFactory {
        private final ClassValue<Constructor<? extends AttributeImpl>> constructors = new ClassValue<Constructor<? extends AttributeImpl>>() {
            @Override
            protected Constructor<? extends AttributeImpl> computeValue(Class<?> attClass) {
                return AttributeFactoryHelpers.findAttributeImplCtor(findImplClass(attClass.asSubclass(Attribute.class)));
            }
        };

        DefaultAttributeFactory() {}

        @Override
        public AttributeImpl createAttributeInstance(Class<? extends Attribute> attClass) {
            try {
                return constructors.get(attClass).newInstance();
            } catch (Error | RuntimeException e) {
                throw e;
            } catch (Throwable e) {
                throw new UndeclaredThrowableException(e);
            }
        }

        private Class<? extends AttributeImpl> findImplClass(Class<? extends Attribute> attClass) {
            try {
                return Class.forName(attClass.getName() + "Impl", true, attClass.getClassLoader()).asSubclass(AttributeImpl.class);
            } catch (ClassNotFoundException cnfe) {
                throw new IllegalArgumentException("Cannot find implementing class for: " + attClass.getName());
            }
        }
    }

    public static <A extends AttributeImpl> AttributeFactory getStaticImplementation(AttributeFactory delegate, Class<A> clazz) {
        final Constructor<? extends AttributeImpl> constr = AttributeFactoryHelpers.findAttributeImplCtor(clazz);
        return new AttributeFactory.StaticImplementationAttributeFactory<A>(delegate, clazz) {
            @Override
            protected A createInstance() {
                try {
                    return (A) constr.newInstance();
                } catch (Error | RuntimeException e) {
                    throw e;
                } catch (Throwable e) {
                    throw new UndeclaredThrowableException(e);
                }
            }
        };
    }
}

AttributeFactorySubstitutions.java

package org.apache.lucene.util;

import com.oracle.svm.core.annotate.Alias;
import com.oracle.svm.core.annotate.Delete;
import com.oracle.svm.core.annotate.RecomputeFieldValue;
import com.oracle.svm.core.annotate.Substitute;
import com.oracle.svm.core.annotate.TargetClass;

import java.lang.invoke.MethodHandle;
import java.lang.reflect.Constructor;
import java.lang.reflect.UndeclaredThrowableException;
import java.util.Arrays;

@TargetClass(AttributeFactory.class)
final class AttributeFactorySubstitutions {
    @Alias
    @RecomputeFieldValue(kind = RecomputeFieldValue.Kind.FromAlias, isFinal = true)
    public static AttributeFactory DEFAULT_ATTRIBUTE_FACTORY = new AttributeFactory() {
        private final ClassValue<Constructor<? extends AttributeImpl>> constructors = new ClassValue<Constructor<? extends AttributeImpl>>() {
            @Override
            protected Constructor<? extends AttributeImpl> computeValue(Class<?> attClass) {
                return AttributeFactoryHelpers.findAttributeImplCtor(findImplClass(attClass.asSubclass(Attribute.class)));
            }
        };

        @Override
        public AttributeImpl createAttributeInstance(Class<? extends Attribute> attClass) {
            try {
                return constructors.get(attClass).newInstance();
            } catch (Error | RuntimeException e) {
                throw e;
            } catch (Throwable e) {
                throw new UndeclaredThrowableException(e);
            }
        }

        private Class<? extends AttributeImpl> findImplClass(Class<? extends Attribute> attClass) {
            try {
                return Class.forName(attClass.getName() + "Impl", true, attClass.getClassLoader()).asSubclass(AttributeImpl.class);
            } catch (ClassNotFoundException cnfe) {
                throw new IllegalArgumentException("Cannot find implementing class for: " + attClass.getName());
            }
        }
    };

    @Substitute
    public static <A extends AttributeImpl> AttributeFactory getStaticImplementation(AttributeFactory delegate, Class<A> clazz) {
        final Constructor<? extends AttributeImpl> constr = AttributeFactoryHelpers.findAttributeImplCtor(clazz);
        return new AttributeFactory.StaticImplementationAttributeFactory<A>(delegate, clazz) {
            @Override
            protected A createInstance() {
                try {
                    return (A) constr.newInstance();
                } catch (Error | RuntimeException e) {
                    throw e;
                } catch (Throwable e) {
                    throw new UndeclaredThrowableException(e);
                }
            }
        };
    }

    @Delete
    static final MethodHandle findAttributeImplCtor(Class<? extends AttributeImpl> clazz) {
        throw new RuntimeException();
    }
}

MMapDirectorySubstitutions.java

package org.apache.lucene.store;

import com.oracle.svm.core.annotate.Alias;
import com.oracle.svm.core.annotate.RecomputeFieldValue;
import com.oracle.svm.core.annotate.Substitute;
import com.oracle.svm.core.annotate.TargetClass;

@TargetClass(MMapDirectory.class)
final class MMapDirectorySubstitutions {
    @Alias
    @RecomputeFieldValue(kind = RecomputeFieldValue.Kind.FromAlias, isFinal = true)
    private static ByteBufferGuard.BufferCleaner CLEANER = null;

    @Alias
    @RecomputeFieldValue(kind = RecomputeFieldValue.Kind.FromAlias, isFinal = true)
    private static boolean UNMAP_SUPPORTED = false;

    @Substitute
    private static Object unmapHackImpl() {
        return "Manually disabled for graalvm";
    }
}

TokenStreamSubstitutions.java

package org.apache.lucene.analysis;

import com.oracle.svm.core.annotate.Alias;
import com.oracle.svm.core.annotate.RecomputeFieldValue;
import com.oracle.svm.core.annotate.TargetClass;
import org.apache.lucene.analysis.tokenattributes.PackedTokenAttributeImpl;
import org.apache.lucene.util.AttributeFactory;
import org.apache.lucene.util.AttributeFactoryHelpers;

@TargetClass(TokenStream.class)
final class TokenStreamSubstitutions {
    @Alias
    @RecomputeFieldValue(kind = RecomputeFieldValue.Kind.FromAlias, isFinal = true)
    public static AttributeFactory DEFAULT_TOKEN_ATTRIBUTE_FACTORY = AttributeFactoryHelpers.getStaticImplementation(AttributeFactory.DEFAULT_ATTRIBUTE_FACTORY, PackedTokenAttributeImpl.class);
}