nashorn/test/src/jdk/nashorn/internal/performance/PerformanceWrapper.java

/*
 * Copyright (c) 2010, 2013, 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.performance;

import java.io.OutputStream;
import java.util.concurrent.Callable;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;
import jdk.nashorn.internal.objects.Global;
import jdk.nashorn.internal.runtime.Context;
import jdk.nashorn.internal.runtime.ScriptFunction;
import jdk.nashorn.internal.runtime.ScriptRuntime;

@SuppressWarnings("javadoc")
public class PerformanceWrapper extends jdk.nashorn.tools.Shell {

    int _numberOfIterations;
    int _runsPerIteration;

    protected void runCompileOnlyTest(final String name, final int numberOfIterations, final int runsPerIteration, final String testURL) throws Throwable {
        final String[] args = { name, "--compile-only=true", "-dump-on-error", "--", testURL };

        final long[] times = new long[numberOfIterations + 1];
        times[0] = System.nanoTime(); // Calendar.getInstance().getTimeInMillis();

        for (int iteration = 1; iteration <= numberOfIterations; iteration++) {
            for (int i = 0; i < runsPerIteration; i++) {
                run(System.in, System.out, System.err, args);
            }
            times[iteration] = System.nanoTime();
        }

        for (int i = 0; i < numberOfIterations; i++) {
            System.out.println("Iteration " + (i + 1) + " average time: " + ((times[i + 1] - times[i]) / (float)runsPerIteration) / 1000000.0 + " ms.");
        }
    }

    protected void runExecuteOnlyTest(final String name, final int numberOfIterations, final int runsPerIteration, final String testURL) throws Throwable {
        runExecuteOnlyTest(name, numberOfIterations, runsPerIteration, testURL, System.out, System.err, new String[0]);
    }

    protected void runExecuteOnlyTest(final String name, final int numberOfIterations, final int runsPerIteration, final String testURL, final OutputStream out, final OutputStream err) throws Throwable {
        runExecuteOnlyTest(name, numberOfIterations, runsPerIteration, testURL, out, err, new String[0]);
    }


    protected void runExecuteOnlyTest(final String name, final int numberOfIterations, final int runsPerIteration, final String testURL, final OutputStream out, final OutputStream err, final String[] newargs) throws Throwable {
        final String[] args=new String[newargs.length+1];
        System.arraycopy(newargs, 0, args, 1, newargs.length);
        args[0]=name;

//      for (String s: args)
//          System.out.println(s);

        _numberOfIterations = numberOfIterations;
        _runsPerIteration = runsPerIteration;
        run(System.in, out, err, args);
//      System.out.println("overridableRun finished");
    }

    @Override
    protected Object apply(final ScriptFunction target, final Object self) {
        if (_runsPerIteration == 0 && _numberOfIterations == 0) {
            final Global global = jdk.nashorn.internal.runtime.Context.getGlobal();
            final ScriptFunction _target = target;
            final Object _self = self;

            class MyThread implements Callable<Object> {
                @Override
                public Object call() {
                    Context.setGlobal(global);
                    //just execute and return script is sufficient
                    final Object scriptRuntimeApplyResult = ScriptRuntime.apply(_target, _self);
                    return scriptRuntimeApplyResult;
                }
            }

            final java.util.concurrent.ThreadPoolExecutor executor = new java.util.concurrent.ThreadPoolExecutor(1, 1, 1, TimeUnit.MINUTES, new java.util.concurrent.ArrayBlockingQueue<Runnable>(10));
            final MyThread myThread = new MyThread();
            // executor.execute(myThread);
            Object result;
            Future<?> futureResult = null;

            try {
                futureResult = executor.submit(myThread);
                final String timeout = System.getProperty("timeout.value");
                int tmout = 0;
                if (timeout != null) {
                    try {
                        tmout = Integer.parseInt(timeout);
                    } catch (final Exception e) {
                        e.printStackTrace();
                    }
                }
                if (tmout != 0) {
                    result = futureResult.get(10, TimeUnit.MINUTES);
                } else {
                    result = futureResult.get();
                }
            } catch (final InterruptedException | ExecutionException e) {
                e.printStackTrace();
                return null;
            } catch (final TimeoutException e) {
                System.out.println("timeout while script execution");
                futureResult.cancel(true);
                return null;
            }

            return result;
        }

        final long[] times = new long[_numberOfIterations + 1];
        times[0] = System.nanoTime();
        for (int iteration = 1; iteration <= _numberOfIterations; iteration++) {
            for (int i = 0; i < _runsPerIteration; i++) {
                // empty
            }
            times[iteration] = System.nanoTime();
        }

        for (int i = 0; i < _numberOfIterations; i++) {
            System.out.println("Iteration " + (i + 1) + " average time: " + ((times[i + 1] - times[i]) / (float)_runsPerIteration) / 1000000.0 + " ms.");
        }

        return ScriptRuntime.apply(target, self);
    }
}