• Home
• io.github.qsy7.java.aspects.timeout
• plugin
• 0.1.2
• source code
• TimeoutAdvice.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

io.github.qsy7.timeout.plugin.TimeoutAdvice Maven / Gradle / Ivy

package io.github.qsy7.timeout.plugin;

import io.github.qsy7.timeout.TimeConstrainedMethodInvocation;
import io.github.qsy7.timeout.impl.TimeConstrainedMethodCall;
import java.time.Duration;
import net.bytebuddy.asm.Advice;

public class TimeoutAdvice {

  @Advice.OnMethodEnter
  public static TimeConstrainedMethodCall onEnter(@Advice.This Object intercepted) {
    final TimeConstrainedMethodInvocation timeConstrainedMethodInvocation =
        (TimeConstrainedMethodInvocation) intercepted;

    if (timeConstrainedMethodInvocation.getAllowedExecutionDuration() == null) {
      return null;
    }

    final Duration duration = timeConstrainedMethodInvocation.getAllowedExecutionDuration();
    return new TimeConstrainedMethodCall(duration);
  }

  @Advice.OnMethodExit
  public static void onExit(
      @Advice.Enter final TimeConstrainedMethodCall timeConstrainedMethodCall) {
    if (timeConstrainedMethodCall != null) {
      timeConstrainedMethodCall.cancelInterruption();
    }
  }
}