Examples with Instrumentor com.navercorp.pinpoint.bootstrap.instrument.Instrumentor used on opensource projects

Example 31 with Instrumentor

use of com.navercorp.pinpoint.bootstrap.instrument.Instrumentor in project pinpoint by naver.

the class ThriftPlugin method addTTransportEditor.

// Common - transports
private void addTTransportEditor(String tTransportFqcn) {
    final String targetClassName = tTransportFqcn;
    transformTemplate.transform(targetClassName, new TransformCallback() {

        @Override
        public byte[] doInTransform(Instrumentor instrumentor, ClassLoader loader, String className, Class<?> classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) throws InstrumentException {
            final InstrumentClass target = instrumentor.getInstrumentClass(loader, className, classfileBuffer);
            target.addField(ThriftConstants.FIELD_ACCESSOR_SOCKET);
            return target.toBytecode();
        }
    });
}

Example 32 with Instrumentor

use of com.navercorp.pinpoint.bootstrap.instrument.Instrumentor in project pinpoint by naver.

the class ThriftPlugin method addTProtocolInterceptors.

private void addTProtocolInterceptors(ThriftPluginConfig config, String tProtocolClassName) {
    final boolean traceThriftClient = config.traceThriftClient();
    final boolean traceThriftProcessor = config.traceThriftProcessor();
    final String targetClassName = tProtocolClassName;
    transformTemplate.transform(targetClassName, new TransformCallback() {

        @Override
        public byte[] doInTransform(Instrumentor instrumentor, ClassLoader loader, String className, Class<?> classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) throws InstrumentException {
            final InstrumentClass target = instrumentor.getInstrumentClass(loader, className, classfileBuffer);
            // client
            if (traceThriftClient) {
                // TProtocol.writeFieldStop()
                final InstrumentMethod writeFieldStop = target.getDeclaredMethod("writeFieldStop");
                if (writeFieldStop != null) {
                    String interceptor = "com.navercorp.pinpoint.plugin.thrift.interceptor.tprotocol.client.TProtocolWriteFieldStopInterceptor";
                    writeFieldStop.addInterceptor(interceptor);
                }
            }
            // processor
            if (traceThriftProcessor) {
                target.addField(ThriftConstants.FIELD_ACCESSOR_SERVER_MARKER_FLAG);
                // TProtocol.readFieldBegin()
                final InstrumentMethod readFieldBegin = target.getDeclaredMethod("readFieldBegin");
                if (readFieldBegin != null) {
                    String interceptor = "com.navercorp.pinpoint.plugin.thrift.interceptor.tprotocol.server.TProtocolReadFieldBeginInterceptor";
                    readFieldBegin.addInterceptor(interceptor);
                }
                // TProtocol.readBool, TProtocol.readBinary, TProtocol.readI16, TProtocol.readI64
                final List<InstrumentMethod> readTTypes = target.getDeclaredMethods(MethodFilters.name("readBool", "readBinary", "readI16", "readI64"));
                if (readTTypes != null) {
                    String tTypeCommonInterceptor = "com.navercorp.pinpoint.plugin.thrift.interceptor.tprotocol.server.TProtocolReadTTypeInterceptor";
                    for (InstrumentMethod readTType : readTTypes) {
                        if (readTType != null) {
                            readTType.addInterceptor(tTypeCommonInterceptor);
                        }
                    }
                }
                // TProtocol.readMessageEnd()
                final InstrumentMethod readMessageEnd = target.getDeclaredMethod("readMessageEnd");
                if (readMessageEnd != null) {
                    String interceptor = "com.navercorp.pinpoint.plugin.thrift.interceptor.tprotocol.server.TProtocolReadMessageEndInterceptor";
                    readMessageEnd.addInterceptor(interceptor);
                }
                // for async processors
                target.addField(ThriftConstants.FIELD_ACCESSOR_ASYNC_MARKER_FLAG);
                // TProtocol.readMessageBegin()
                final InstrumentMethod readMessageBegin = target.getDeclaredMethod("readMessageBegin");
                if (readMessageBegin != null) {
                    String interceptor = "com.navercorp.pinpoint.plugin.thrift.interceptor.tprotocol.server.TProtocolReadMessageBeginInterceptor";
                    readMessageBegin.addInterceptor(interceptor);
                }
            }
            return target.toBytecode();
        }
    });
}

Example 33 with Instrumentor

use of com.navercorp.pinpoint.bootstrap.instrument.Instrumentor in project pinpoint by naver.

the class ThriftPlugin method addTAsyncClientManagerEditor.

private void addTAsyncClientManagerEditor() {
    final String targetClassName = "org.apache.thrift.async.TAsyncClientManager";
    transformTemplate.transform(targetClassName, new TransformCallback() {

        @Override
        public byte[] doInTransform(Instrumentor instrumentor, ClassLoader loader, String className, Class<?> classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) throws InstrumentException {
            final InstrumentClass target = instrumentor.getInstrumentClass(loader, className, classfileBuffer);
            // TAsyncClientManager.call(TAsyncMethodCall)
            final InstrumentMethod call = target.getDeclaredMethod("call", "org.apache.thrift.async.TAsyncMethodCall");
            if (call != null) {
                String interceptor = "com.navercorp.pinpoint.plugin.thrift.interceptor.client.async.TAsyncClientManagerCallInterceptor";
                call.addInterceptor(interceptor);
            }
            return target.toBytecode();
        }
    });
}

Example 34 with Instrumentor

use of com.navercorp.pinpoint.bootstrap.instrument.Instrumentor in project pinpoint by naver.

the class ThriftPlugin method addTServiceClientEditor.

private void addTServiceClientEditor(ThriftPluginConfig config) {
    final boolean traceServiceArgs = config.traceThriftServiceArgs();
    final boolean traceServiceResult = config.traceThriftServiceResult();
    final String targetClassName = "org.apache.thrift.TServiceClient";
    transformTemplate.transform(targetClassName, new TransformCallback() {

        @Override
        public byte[] doInTransform(Instrumentor instrumentor, ClassLoader loader, String className, Class<?> classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) throws InstrumentException {
            final InstrumentClass target = instrumentor.getInstrumentClass(loader, className, classfileBuffer);
            // TServiceClient.sendBase(String, TBase)
            final InstrumentMethod sendBase = target.getDeclaredMethod("sendBase", "java.lang.String", "org.apache.thrift.TBase");
            if (sendBase != null) {
                String interceptor = "com.navercorp.pinpoint.plugin.thrift.interceptor.client.TServiceClientSendBaseInterceptor";
                sendBase.addInterceptor(interceptor, va(traceServiceArgs));
            }
            // TServiceClient.receiveBase(TBase, String)
            final InstrumentMethod receiveBase = target.getDeclaredMethod("receiveBase", "org.apache.thrift.TBase", "java.lang.String");
            if (receiveBase != null) {
                String interceptor = "com.navercorp.pinpoint.plugin.thrift.interceptor.client.TServiceClientReceiveBaseInterceptor";
                receiveBase.addInterceptor(interceptor, va(traceServiceResult));
            }
            return target.toBytecode();
        }
    });
}

Example 35 with Instrumentor

use of com.navercorp.pinpoint.bootstrap.instrument.Instrumentor in project pinpoint by naver.

the class ThriftPlugin method addTBaseProcessorEditor.

private void addTBaseProcessorEditor() {
    final String targetClassName = "org.apache.thrift.TBaseProcessor";
    transformTemplate.transform(targetClassName, new TransformCallback() {

        @Override
        public byte[] doInTransform(Instrumentor instrumentor, ClassLoader loader, String className, Class<?> classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) throws InstrumentException {
            final InstrumentClass target = instrumentor.getInstrumentClass(loader, className, classfileBuffer);
            // TBaseProcessor.process(TProtocol, TProtocol)
            final InstrumentMethod process = target.getDeclaredMethod("process", "org.apache.thrift.protocol.TProtocol", "org.apache.thrift.protocol.TProtocol");
            if (process != null) {
                String interceptor = "com.navercorp.pinpoint.plugin.thrift.interceptor.server.TBaseProcessorProcessInterceptor";
                process.addInterceptor(interceptor);
            }
            return target.toBytecode();
        }
    });
}