SystemTimeMillis

Author: tobixlea

src/main/java/software/amazon/lambda/snapstart/ByteCodeIntrospector.java

@@ -37,6 +37,14 @@ public class ByteCodeIntrospector {
 
     private static final String INSTANT_SIGNATURE = "Ljava/time/Instant;";
 
+    private static final String LOCAL_DATE_TIME_SIGNATURE = "Ljava/time/LocalDateTime;";
+
+    private static final String LOCAL_DATE_SIGNATURE = "Ljava/time/LocalDate;";
+
+    private static final String LOCAL_TIME_SIGNATURE = "Ljava/time/LocalTime;";
+
+    private static final String ZONED_DATE_TIME_SIGNATURE = "Ljava/time/ZonedDateTime;";
+
     private static final Map<String, Set<String>> TIMESTAMP_METHODS = new HashMap<String, Set<String>>() {{
         put("java.lang.System", setOf("currentTimeMillis", "nanoTime"));
     }};
@@ -188,6 +196,18 @@ boolean isTimestamp(OpcodeStack stack) {
         if (INSTANT_SIGNATURE.equals(stack.getStackItem(0).getSignature())) {
             return true;
         }
+        if (LOCAL_DATE_TIME_SIGNATURE.equals(stack.getStackItem(0).getSignature())) {
+            return true;
+        }
+        if (LOCAL_DATE_SIGNATURE.equals(stack.getStackItem(0).getSignature())) {
+            return true;
+        }
+        if (LOCAL_TIME_SIGNATURE.equals(stack.getStackItem(0).getSignature())) {
+            return true;
+        }
+        if (ZONED_DATE_TIME_SIGNATURE.equals(stack.getStackItem(0).getSignature())) {
+            return true;
+        }
         XMethod xMethod = stack.getStackItem(0).getReturnValueOf();
         if (xMethod != null) {
             Set<String> methodNames = TIMESTAMP_METHODS.get(xMethod.getClassName());

src/test/java/software/amazon/lambda/snapstart/lambdaexamples/LambdaUsingTs.java

@@ -9,6 +9,10 @@
 import com.amazonaws.services.lambda.runtime.RequestHandler;
 import java.time.Clock;
 import java.time.Instant;
+import java.time.LocalDateTime;
+import java.time.LocalDate;
+import java.time.LocalTime;
+import java.time.ZonedDateTime;
 import java.util.Map;
 
 public class LambdaUsingTs implements RequestHandler<Map<String,String>, String> {
@@ -19,12 +23,21 @@ public class LambdaUsingTs implements RequestHandler<Map<String,String>, String>
     private long tsFromSystemTimeNano;
     private Instant tsFromInstantNow;
     private Instant tsFromClock;
+    private LocalDateTime tsFromLocalDateTimeNow;
+    private LocalDate tsFromLocalDateNow;
+    private LocalTime tsFromLocalTimeNow;
+    private ZonedDateTime tsFromZonedDateTimeNow;
 
     public LambdaUsingTs(Clock clock) {
         tsFromSystemTimeMillis = System.currentTimeMillis(); // This is a bug
         tsFromSystemTimeNano = System.nanoTime(); // This is a bug
         tsFromInstantNow = Instant.now(); // This is a bug
         tsFromClock = clock.instant(); // This is a bug
+        tsFromLocalDateTimeNow = LocalDateTime.now(); // This is a bug
+        tsFromLocalDateNow = LocalDate.now(); // This is a bug
+        tsFromLocalTimeNow = LocalTime.now(); // This is a bug
+        tsFromZonedDateTimeNow = ZonedDateTime.now(); // This is a bug
+
 
         logName = getLogName(); // This is a bug
     }