Commit 7c305fe0 authored by 张鑫's avatar 张鑫 Committed by 吴晟
Browse files

Add witness class for kafka plugin (#1300)

parent 307bd4c2

apm-sniffer/apm-sdk-plugin/kafka-v1-plugin/src/main/java/org/apache/skywalking/apm/plugin/kafka/v1/define/AbstractKafkaInstrumentation.java

0 → 100644
+28 −0
Original line number Diff line number Diff line 
/*

 * Licensed to the Apache Software Foundation (ASF) under one or more

 * contributor license agreements.  See the NOTICE file distributed with

 * this work for additional information regarding copyright ownership.

 * The ASF licenses this file to You under the Apache License, Version 2.0

 * (the "License"); you may not use this file except in compliance with

 * the License.  You may obtain a copy of the License at

 *

 *     http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 *

 */



package org.apache.skywalking.apm.plugin.kafka.v1.define;



import org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.ClassInstanceMethodsEnhancePluginDefine;



public abstract class AbstractKafkaInstrumentation extends ClassInstanceMethodsEnhancePluginDefine {



    @Override protected String[] witnessClasses() {

        return new String[]{"org.apache.kafka.clients.ApiVersions"};

    }

}

apm-sniffer/apm-sdk-plugin/kafka-v1-plugin/src/main/java/org/apache/skywalking/apm/plugin/kafka/v1/define/CallbackInstrumentation.java

+1 −2
Original line number Diff line number Diff line
@@ -22,13 +22,12 @@ import net.bytebuddy.description.method.MethodDescription; 
import net.bytebuddy.matcher.ElementMatcher;

import org.apache.skywalking.apm.agent.core.plugin.interceptor.ConstructorInterceptPoint;

import org.apache.skywalking.apm.agent.core.plugin.interceptor.InstanceMethodsInterceptPoint;

import org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.ClassInstanceMethodsEnhancePluginDefine;

import org.apache.skywalking.apm.agent.core.plugin.match.ClassMatch;



import static net.bytebuddy.matcher.ElementMatchers.named;

import static org.apache.skywalking.apm.agent.core.plugin.match.HierarchyMatch.byHierarchyMatch;



public class CallbackInstrumentation extends ClassInstanceMethodsEnhancePluginDefine {

public class CallbackInstrumentation extends AbstractKafkaInstrumentation {



    public static final String ENHANCE_CLASS = "org.apache.kafka.clients.producer.Callback";

    public static final String ENHANCE_METHOD = "onCompletion";

apm-sniffer/apm-sdk-plugin/kafka-v1-plugin/src/main/java/org/apache/skywalking/apm/plugin/kafka/v1/define/KafkaConsumerInstrumentation.java

+1 −2
Original line number Diff line number Diff line
@@ -22,7 +22,6 @@ import net.bytebuddy.description.method.MethodDescription; 
import net.bytebuddy.matcher.ElementMatcher;

import org.apache.skywalking.apm.agent.core.plugin.interceptor.ConstructorInterceptPoint;

import org.apache.skywalking.apm.agent.core.plugin.interceptor.InstanceMethodsInterceptPoint;

import org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.ClassInstanceMethodsEnhancePluginDefine;

import org.apache.skywalking.apm.agent.core.plugin.match.ClassMatch;



import static net.bytebuddy.matcher.ElementMatchers.named;
@@ -41,7 +40,7 @@ import static org.apache.skywalking.apm.agent.core.plugin.match.NameMatch.byName 
 *

 * @author zhang xin

 */

public class KafkaConsumerInstrumentation extends ClassInstanceMethodsEnhancePluginDefine {

public class KafkaConsumerInstrumentation extends AbstractKafkaInstrumentation {



    public static final String CONSTRUCTOR_INTERCEPT_TYPE = "org.apache.kafka.clients.consumer.ConsumerConfig";

    public static final String CONSTRUCTOR_INTERCEPTOR_CLASS = "org.apache.skywalking.apm.plugin.kafka.v1.ConsumerConstructorInterceptor";

apm-sniffer/apm-sdk-plugin/kafka-v1-plugin/src/main/java/org/apache/skywalking/apm/plugin/kafka/v1/define/KafkaProducerInstrumentation.java

+1 −2
Original line number Diff line number Diff line
@@ -22,7 +22,6 @@ import net.bytebuddy.description.method.MethodDescription; 
import net.bytebuddy.matcher.ElementMatcher;

import org.apache.skywalking.apm.agent.core.plugin.interceptor.ConstructorInterceptPoint;

import org.apache.skywalking.apm.agent.core.plugin.interceptor.InstanceMethodsInterceptPoint;

import org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.ClassInstanceMethodsEnhancePluginDefine;

import org.apache.skywalking.apm.agent.core.plugin.match.ClassMatch;



import static net.bytebuddy.matcher.ElementMatchers.named;
@@ -42,7 +41,7 @@ import static org.apache.skywalking.apm.agent.core.plugin.match.NameMatch.byName 
 *

 * @author zhang xin

 */

public class KafkaProducerInstrumentation extends ClassInstanceMethodsEnhancePluginDefine {

public class KafkaProducerInstrumentation extends AbstractKafkaInstrumentation {



    public static final String INTERCEPTOR_CLASS = "org.apache.skywalking.apm.plugin.kafka.v1.KafkaProducerInterceptor";

    public static final String ENHANCE_CLASS = "org.apache.kafka.clients.producer.KafkaProducer";

apm-sniffer/apm-sdk-plugin/kafka-v1-plugin/src/main/java/org/apache/skywalking/apm/plugin/kafka/v1/define/ProducerRecordInstrumentation.java

+1 −2
Original line number Diff line number Diff line
@@ -22,13 +22,12 @@ import net.bytebuddy.description.method.MethodDescription; 
import net.bytebuddy.matcher.ElementMatcher;

import org.apache.skywalking.apm.agent.core.plugin.interceptor.ConstructorInterceptPoint;

import org.apache.skywalking.apm.agent.core.plugin.interceptor.InstanceMethodsInterceptPoint;

import org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.ClassInstanceMethodsEnhancePluginDefine;

import org.apache.skywalking.apm.agent.core.plugin.match.ClassMatch;



import static net.bytebuddy.matcher.ElementMatchers.takesArguments;

import static org.apache.skywalking.apm.agent.core.plugin.match.NameMatch.byName;



public class ProducerRecordInstrumentation extends ClassInstanceMethodsEnhancePluginDefine {

public class ProducerRecordInstrumentation extends AbstractKafkaInstrumentation {



    public static final String CONSTRUCTOR_INTERCEPTOR_CLASS = "org.apache.skywalking.apm.plugin.kafka.v1.ProducerRecordConstructorInterceptor";

    public static final String ENHANCE_CLASS = "org.apache.kafka.clients.producer.ProducerRecord";