StoredScript.java

/*
 * Copyright (c) 2010, 2014, Oracle and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.  Oracle designates this
 * particular file as subject to the "Classpath" exception as provided
 * by Oracle in the LICENSE file that accompanied this code.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */

package jdk.nashorn.internal.runtime;

import java.io.Serializable;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

/**
 * Class representing a persistent compiled script.
 */
public final class StoredScript implements Serializable {

    /** Compilation id */
    private final int compilationId;

    /** Main class name. */
    private final String mainClassName;

    /** Map of class names to class bytes. */
    private final Map<String, byte[]> classBytes;

    /** Constants array. */
    private final Object[] constants;

    /** Function initializers */
    private final Map<Integer, FunctionInitializer> initializers;

    private static final long serialVersionUID = 2958227232195298340L;

    /**
     * Constructor.
     *
     * @param compilationId compilation id
     * @param mainClassName main class name
     * @param classBytes map of class names to class bytes
     * @param initializers initializer map, id -> FunctionInitializer
     * @param constants constants array
     */
    public StoredScript(final int compilationId, final String mainClassName, final Map<String, byte[]> classBytes, final Map<Integer, FunctionInitializer> initializers, final Object[] constants) {
        this.compilationId = compilationId;
        this.mainClassName = mainClassName;
        this.classBytes = classBytes;
        this.constants = constants;
        this.initializers = initializers;
    }

    /**
     * Get the compilation id for this StoredScript
     * @return compilation id
     */
    public int getCompilationId() {
        return compilationId;
    }

    private Map<String, Class<?>> installClasses(final Source source, final CodeInstaller installer) {
        final Map<String, Class<?>> installedClasses = new HashMap<>();
        final byte[]   mainClassBytes = classBytes.get(mainClassName);
        final Class<?> mainClass      = installer.install(mainClassName, mainClassBytes);

        installedClasses.put(mainClassName, mainClass);

        for (final Map.Entry<String, byte[]> entry : classBytes.entrySet()) {
            final String className = entry.getKey();

            if (!className.equals(mainClassName)) {
                installedClasses.put(className, installer.install(className, entry.getValue()));
            }
        }

        installer.initialize(installedClasses.values(), source, constants);
        return installedClasses;
    }

    FunctionInitializer installFunction(final RecompilableScriptFunctionData data, final CodeInstaller installer) {
        final Map<String, Class<?>> installedClasses = installClasses(data.getSource(), installer);

        assert initializers != null;
        assert initializers.size() == 1;
        final FunctionInitializer initializer = initializers.values().iterator().next();

        for (int i = 0; i < constants.length; i++) {
            if (constants[i] instanceof RecompilableScriptFunctionData) {
                // replace deserialized function data with the ones we already have
                final RecompilableScriptFunctionData newData = data.getScriptFunctionData(((RecompilableScriptFunctionData) constants[i]).getFunctionNodeId());
                assert newData != null;
                newData.initTransients(data.getSource(), installer);
                constants[i] = newData;
            }
        }

        initializer.setCode(installedClasses.get(initializer.getClassName()));
        return initializer;
    }

    /**
     * Install as script.
     *
     * @param source the source
     * @param installer the installer
     * @return main script class
     */
    Class<?> installScript(final Source source, final CodeInstaller installer) {

        final Map<String, Class<?>> installedClasses = installClasses(source, installer);

        for (final Object constant : constants) {
            if (constant instanceof RecompilableScriptFunctionData) {
                final RecompilableScriptFunctionData data = (RecompilableScriptFunctionData) constant;
                data.initTransients(source, installer);
                final FunctionInitializer initializer = initializers.get(data.getFunctionNodeId());
                if (initializer != null) {
                    initializer.setCode(installedClasses.get(initializer.getClassName()));
                    data.initializeCode(initializer);
                }
            }
        }

        return installedClasses.get(mainClassName);
    }

    @Override
    public int hashCode() {
        int hash = mainClassName.hashCode();
        hash = 31 * hash + classBytes.hashCode();
        hash = 31 * hash + Arrays.hashCode(constants);
        return hash;
    }

    @Override
    public boolean equals(final Object obj) {
        if (obj == this) {
            return true;
        }
        if (!(obj instanceof StoredScript)) {
            return false;
        }

        final StoredScript cs = (StoredScript) obj;
        return mainClassName.equals(cs.mainClassName)
                && classBytes.equals(cs.classBytes)
                && Arrays.equals(constants, cs.constants);
    }
}