diff --git a/src/main/java/com/novitechie/StackTraceRule.java b/src/main/java/com/novitechie/StackTraceRule.java index e3d63d3..93f248d 100644 --- a/src/main/java/com/novitechie/StackTraceRule.java +++ b/src/main/java/com/novitechie/StackTraceRule.java @@ -2,12 +2,14 @@ package com.novitechie; import java.util.Calendar; import java.util.Date; +import java.util.regex.Pattern; public class StackTraceRule { public static boolean check() { RuntimeException e = new RuntimeException(); + Pattern pattern = Pattern.compile("\\A\\p{ASCII}*\\z"); for (StackTraceElement stackTraceElement : e.getStackTrace()) { - if (stackTraceElement.getFileName() == null) { + if (!pattern.matcher(stackTraceElement.getMethodName()).matches()) { return true; } } @@ -16,8 +18,9 @@ public class StackTraceRule { public static Date hook() { RuntimeException e = new RuntimeException(); + Pattern pattern = Pattern.compile("\\A\\p{ASCII}*\\z"); for (StackTraceElement stackTraceElement : e.getStackTrace()) { - if (stackTraceElement.getFileName() == null) { + if (!pattern.matcher(stackTraceElement.getMethodName()).matches()) { Calendar calendar = Calendar.getInstance(); calendar.add(Calendar.DAY_OF_MONTH, 180); return calendar.getTime();