Snapshot VM startup time

Author: fangerer

graalpython/com.oracle.graal.python.shell/src/com/oracle/graal/python/shell/GraalPythonMain.java

@@ -62,6 +62,14 @@
 
 public class GraalPythonMain extends AbstractLanguageLauncher {
     public static void main(String[] args) {
+        long startupNanoTime = System.nanoTime();
+        long startupWallClockTime = System.currentTimeMillis();
+        if (GraalPythonMain.startupNanoTime == -1) {
+            GraalPythonMain.startupNanoTime = startupNanoTime;
+        }
+        if (GraalPythonMain.startupWallClockTime == -1) {
+            GraalPythonMain.startupWallClockTime = startupWallClockTime;
+        }
         new GraalPythonMain().launch(args);
     }
 
@@ -70,6 +78,9 @@ public static void main(String[] args) {
 
     // provided by GraalVM bash launchers, ignored in native image mode
     private static final String BASH_LAUNCHER_EXEC_NAME = System.getProperty("org.graalvm.launcher.executablename");
+    
+    private static long startupWallClockTime = -1;
+    private static long startupNanoTime = -1;
 
     private ArrayList<String> programArgs = null;
     private String commandString = null;
@@ -84,6 +95,7 @@ public static void main(String[] args) {
     private final boolean stdinIsInteractive = System.console() != null;
     private boolean unbufferedIO = false;
     private boolean multiContext = false;
+    private boolean snaptshotStartup = false;
     private VersionAction versionAction = VersionAction.None;
     private List<String> givenArguments;
     private List<String> relaunchArgs;
@@ -247,6 +259,13 @@ protected List<String> preprocessArguments(List<String> givenArgs, Map<String, S
                     i += 1;
                     checkHashPycsMode = arguments.get(i);
                     break;
+                case "-snapshot-startup":
+                    if (wantsExperimental) {
+                        snaptshotStartup = true;
+                    } else {
+                        unrecognized.add(arg);
+                    }
+                    break;
                 default:
                     if (!arg.startsWith("-")) {
                         inputFile = arg;
@@ -473,6 +492,10 @@ protected void launch(Builder contextBuilder) {
         int rc = 1;
         try (Context context = contextBuilder.build()) {
             runVersionAction(versionAction, context.getEngine());
+            
+            if (snaptshotStartup) {
+                evalInternal(context, "__graalpython__.startup_wall_clock_ts = " + startupWallClockTime + "; __graalpython__.startup_nano = " + startupNanoTime);
+            }
 
             if (!quietFlag && (verboseFlag || (commandString == null && inputFile == null && stdinIsInteractive))) {
                 print("Python " + evalInternal(context, "import sys; sys.version + ' on ' + sys.platform").asString());

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/modules/GraalPythonModuleBuiltins.java

@@ -145,6 +145,8 @@ public void initialize(PythonCore core) {
         }
         this.builtinConstants.put("stdio_encoding", standardStreamEncoding);
         this.builtinConstants.put("stdio_error", standardStreamError);
+        this.builtinConstants.put("startup_wall_clock_ts", -1L);
+        this.builtinConstants.put("startup_nano", -1L);
         // we need these during core initialization, they are re-set in postInitialize
         postInitialize(core);
     }
@@ -510,4 +512,14 @@ boolean typeCheck(Object instance, Object cls,
             return lib.typeCheck(instance, cls);
         }
     }
+
+    @Builtin(name = "time_millis", minNumOfPositionalArgs = 0, maxNumOfPositionalArgs = 1, doc = "Like time.time() but in milliseconds resolution.")
+    @GenerateNodeFactory
+    public abstract static class TimeMillis extends PythonUnaryBuiltinNode {
+        @Specialization
+        @TruffleBoundary
+        static long doIt(@SuppressWarnings("unused") Object dummy) {
+            return System.currentTimeMillis();
+        }
+    }
 }

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/modules/TimeModuleBuiltins.java

@@ -230,13 +230,13 @@ public double time() {
     }
 
     // time.monotonic_ns()
-    @Builtin(name = "monotonic_ns", minNumOfPositionalArgs = 0, doc = "Similar to monotonic(), but return time as nanoseconds.")
+    @Builtin(name = "monotonic_ns", minNumOfPositionalArgs = 0, maxNumOfPositionalArgs = 1, doc = "Similar to monotonic(), but return time as nanoseconds.")
     @GenerateNodeFactory
-    public abstract static class PythonMonotonicNsNode extends PythonBuiltinNode {
+    public abstract static class PythonMonotonicNsNode extends PythonUnaryBuiltinNode {
 
         @Specialization
         @TruffleBoundary
-        public long time() {
+        static long time(@SuppressWarnings("unused") Object dummy) {
             return System.nanoTime();
         }
     }