diff --git a/src/main/java/com/profiler/modifier/AbstractModifier.java b/src/main/java/com/profiler/modifier/AbstractModifier.java index 3d8da49da..fbf817f22 100644 --- a/src/main/java/com/profiler/modifier/AbstractModifier.java +++ b/src/main/java/com/profiler/modifier/AbstractModifier.java @@ -3,6 +3,8 @@ package com.profiler.modifier; import java.util.logging.Level; import java.util.logging.Logger; +import com.profiler.common.ServiceType; +import com.profiler.interceptor.ServiceTypeSupport; import javassist.ClassPool; import com.profiler.Agent; @@ -41,4 +43,10 @@ public abstract class AbstractModifier implements Modifier { } } + public void setServiceType(Interceptor interceptor, ServiceType serviceType) { + if (interceptor instanceof ServiceTypeSupport) { + ((ServiceTypeSupport) interceptor).setServiceType(serviceType); + } + } + }