feat: support powerjob

src/main/java/io/github/artlibs/autotrace4j/interceptor/impl/PowerJobScheduleTaskInterceptor.java

@@ -0,0 +1,60 @@
+package io.github.artlibs.autotrace4j.interceptor.impl;
+
+import io.github.artlibs.autotrace4j.context.AutoTraceCtx;
+import io.github.artlibs.autotrace4j.interceptor.base.AbstractInstanceInterceptor;
+import net.bytebuddy.description.method.MethodDescription;
+import net.bytebuddy.description.type.TypeDescription;
+import net.bytebuddy.matcher.ElementMatcher;
+
+import java.lang.reflect.Method;
+
+import static net.bytebuddy.matcher.ElementMatchers.*;
+
+/**
+ * PowerJob Interceptor
+ *
+ * @author orangewest
+ * @since 2024-07-30
+ * <p>
+ */
+public class PowerJobScheduleTaskInterceptor extends AbstractInstanceInterceptor {
+
+    /**
+     * {@inheritDoc}
+     */
+    @Override
+    public ElementMatcher<? super TypeDescription> typeMatcher() {
+        return hasSuperType(named("tech.powerjob.worker.core.processor.sdk.BasicProcessor"))
+                .and(not(isInterface())).and(not(isAbstract()));
+    }
+
+    /**
+     * {@inheritDoc}
+     */
+    @Override
+    public ElementMatcher<? super MethodDescription> methodMatcher() {
+        return named("process")
+                .and(takesArgument(0, named("tech.powerjob.worker.core.processor.TaskContext")));
+    }
+
+    /**
+     * {@inheritDoc}
+     */
+    @Override
+    public void onMethodEnter(Object obj, Object[] allArgs, Method originMethod) throws Exception {
+        AutoTraceCtx.setTraceId(AutoTraceCtx.generate());
+        AutoTraceCtx.setSpanId(AutoTraceCtx.generate());
+        // There will be no parent span as this is a startup context
+        AutoTraceCtx.setParentSpanId(null);
+    }
+
+    /**
+     * {@inheritDoc}
+     */
+    @Override
+    public Object onMethodExit(Object obj, Object[] allArgs, Object result, Method originMethod) throws Exception {
+        AutoTraceCtx.removeAll();
+        return result;
+    }
+
+}