io.deephaven.engine.util.DynamicCompileUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of deephaven-engine-table Show documentation
Engine Table: Implementation and closely-coupled utilities
There is a newer version: 0.37.1
Show newest version
/**
 * Copyright (c) 2016-2022 Deephaven Data Labs and Patent Pending
 */
package io.deephaven.engine.util;

import io.deephaven.engine.context.ExecutionContext;
import io.deephaven.engine.context.QueryCompiler;

import java.util.*;
import java.util.function.Supplier;

/**
 * Utilities for dynamic compilation.
 */
public class DynamicCompileUtils {

    public static  Supplier compileSimpleFunction(final Class resultType, final String code) {
        return compileSimpleFunction(resultType, code, Collections.emptyList(), Collections.emptyList());
    }

    public static  Supplier compileSimpleStatement(final Class resultType, final String code,
            final String... imports) {
        final List importClasses = new ArrayList<>();
        for (final String importString : imports) {
            try {
                importClasses.add(Class.forName(importString));
            } catch (ClassNotFoundException e) {
                throw new RuntimeException("Could not find class", e);
            }
        }

        return compileSimpleFunction(resultType, "return " + code, importClasses, Collections.emptyList());
    }

    public static  Supplier compileSimpleFunction(final Class resultType, final String code,
            final Collection imports, final Collection staticImports) {
        final StringBuilder classBody = new StringBuilder();

        classBody.append("import ").append(resultType.getName()).append(";\n");
        for (final Class im : imports) {
            classBody.append("import ").append(im.getName()).append(";\n");
        }
        for (final Class sim : staticImports) {
            classBody.append("import static ").append(sim.getName()).append(".*;\n");
        }

        classBody.append("public class $CLASSNAME$ implements ").append(Supplier.class.getCanonicalName()).append("<")
                .append(resultType.getCanonicalName()).append(">").append(" ").append("{\n");
        classBody.append("  @Override\n");
        classBody.append("  public ").append(resultType.getCanonicalName()).append(" get() {\n");
        classBody.append(code).append(";\n");
        classBody.append("  }\n");
        classBody.append("}\n");

        final Class partitionClass = ExecutionContext.getContext().getQueryCompiler()
                .compile("Function", classBody.toString(), QueryCompiler.FORMULA_PREFIX);

        try {
            // noinspection unchecked
            return ((Supplier) partitionClass.newInstance());
        } catch (InstantiationException | IllegalAccessException e) {
            throw new RuntimeException("Could not instantiate function.", e);
        }
    }

    public static Class getClassThroughCompilation(final String object) {
        final StringBuilder classBody = new StringBuilder();
        classBody.append("public class $CLASSNAME$ implements ").append(Supplier.class.getCanonicalName())
                .append("{ \n");
        classBody.append("  @Override\n");
        classBody.append("  public Class get() { return ").append(object).append(".class; }\n");
        classBody.append("}\n");

        final Class partitionClass = ExecutionContext.getContext().getQueryCompiler()
                .compile("Function", classBody.toString(), QueryCompiler.FORMULA_PREFIX);

        try {
            // noinspection unchecked
            return ((Supplier) partitionClass.newInstance()).get();
        } catch (InstantiationException | IllegalAccessException e) {
            throw new RuntimeException("Could not instantiate function.", e);
        }
    }
}