Examples with StatementEnhanceInfos org.apache.skywalking.apm.plugin.jdbc.define.StatementEnhanceInfos used on opensource projects

Example 1 with StatementEnhanceInfos

use of org.apache.skywalking.apm.plugin.jdbc.define.StatementEnhanceInfos in project incubator-skywalking by apache.

the class PreparedStatementExecuteMethodsInterceptor method beforeMethod.

@Override
public final void beforeMethod(EnhancedInstance objInst, Method method, Object[] allArguments, Class<?>[] argumentsTypes, MethodInterceptResult result) throws Throwable {
    StatementEnhanceInfos cacheObject = (StatementEnhanceInfos) objInst.getSkyWalkingDynamicField();
    ConnectionInfo connectInfo = cacheObject.getConnectionInfo();
    /**
     * For avoid NPE. In this particular case, Execute sql inside the {@link com.mysql.jdbc.ConnectionImpl} constructor,
     * before the interceptor sets the connectionInfo.
     *
     * @see JDBCDriverInterceptor#afterMethod(EnhancedInstance, Method, Object[], Class[], Object)
     */
    if (connectInfo != null) {
        AbstractSpan span = ContextManager.createExitSpan(buildOperationName(connectInfo, method.getName(), cacheObject.getStatementName()), connectInfo.getDatabasePeer());
        Tags.DB_TYPE.set(span, "sql");
        Tags.DB_INSTANCE.set(span, connectInfo.getDatabaseName());
        Tags.DB_STATEMENT.set(span, cacheObject.getSql());
        span.setComponent(connectInfo.getComponent());
        SpanLayer.asDB(span);
    }
}

Example 2 with StatementEnhanceInfos

use of org.apache.skywalking.apm.plugin.jdbc.define.StatementEnhanceInfos in project incubator-skywalking by apache.

the class PreparedStatementExecuteMethodsInterceptor method beforeMethod.

@Override
public final void beforeMethod(EnhancedInstance objInst, Method method, Object[] allArguments, Class<?>[] argumentsTypes, MethodInterceptResult result) throws Throwable {
    StatementEnhanceInfos cacheObject = (StatementEnhanceInfos) objInst.getSkyWalkingDynamicField();
    ConnectionInfo connectInfo = cacheObject.getConnectionInfo();
    AbstractSpan span = ContextManager.createExitSpan(buildOperationName(connectInfo, method.getName(), cacheObject.getStatementName()), connectInfo.getDatabasePeer());
    Tags.DB_TYPE.set(span, "sql");
    Tags.DB_INSTANCE.set(span, connectInfo.getDatabaseName());
    Tags.DB_STATEMENT.set(span, cacheObject.getSql());
    span.setComponent(connectInfo.getComponent());
    SpanLayer.asDB(span);
}

Example 3 with StatementEnhanceInfos

use of org.apache.skywalking.apm.plugin.jdbc.define.StatementEnhanceInfos in project incubator-skywalking by apache.

the class StatementExecuteMethodsInterceptor method beforeMethod.

@Override
public final void beforeMethod(EnhancedInstance objInst, Method method, Object[] allArguments, Class<?>[] argumentsTypes, MethodInterceptResult result) throws Throwable {
    StatementEnhanceInfos cacheObject = (StatementEnhanceInfos) objInst.getSkyWalkingDynamicField();
    ConnectionInfo connectInfo = cacheObject.getConnectionInfo();
    /**
     * To protected the code occur NullPointException. because mysql execute system sql when constructor method in
     * {@link com.mysql.jdbc.ConnectionImpl} class executed. but the interceptor set the connection Info after
     * the constructor method executed.
     *
     * @see JDBCDriverInterceptor#afterMethod(EnhancedInstance, Method, Object[], Class[], Object)
     */
    if (connectInfo != null) {
        AbstractSpan span = ContextManager.createExitSpan(buildOperationName(connectInfo, method.getName(), cacheObject.getStatementName()), connectInfo.getDatabasePeer());
        Tags.DB_TYPE.set(span, "sql");
        Tags.DB_INSTANCE.set(span, connectInfo.getDatabaseName());
        /**
         * The first argument of all intercept method in `com.mysql.jdbc.StatementImpl` class is SQL, except the
         * `executeBatch` method that the jdbc plugin need to trace, because of this method argument size is zero.
         */
        String sql = "";
        if (allArguments.length > 0) {
            sql = (String) allArguments[0];
        }
        Tags.DB_STATEMENT.set(span, sql);
        span.setComponent(connectInfo.getComponent());
        SpanLayer.asDB(span);
    }
}

Example 4 with StatementEnhanceInfos

use of org.apache.skywalking.apm.plugin.jdbc.define.StatementEnhanceInfos in project incubator-skywalking by apache.

the class StatementExecuteMethodsInterceptorTest method setUp.

@Before
public void setUp() {
    serviceMethodInterceptor = new StatementExecuteMethodsInterceptor();
    enhanceRequireCacheObject = new StatementEnhanceInfos(connectionInfo, "SELECT * FROM test", "CallableStatement");
    when(objectInstance.getSkyWalkingDynamicField()).thenReturn(enhanceRequireCacheObject);
    when(method.getName()).thenReturn("executeQuery");
    when(connectionInfo.getComponent()).thenReturn(ComponentsDefine.H2);
    when(connectionInfo.getDBType()).thenReturn("H2");
    when(connectionInfo.getDatabaseName()).thenReturn("test");
    when(connectionInfo.getDatabasePeer()).thenReturn("localhost:3307");
}

Example 5 with StatementEnhanceInfos

use of org.apache.skywalking.apm.plugin.jdbc.define.StatementEnhanceInfos in project incubator-skywalking by apache.

the class PreparedStatementExecuteMethodsInterceptor method beforeMethod.

@Override
public final void beforeMethod(EnhancedInstance objInst, Method method, Object[] allArguments, Class<?>[] argumentsTypes, MethodInterceptResult result) throws Throwable {
    StatementEnhanceInfos cacheObject = (StatementEnhanceInfos) objInst.getSkyWalkingDynamicField();
    ConnectionInfo connectInfo = cacheObject.getConnectionInfo();
    AbstractSpan span = ContextManager.createExitSpan(buildOperationName(connectInfo, method.getName(), cacheObject.getStatementName()), connectInfo.getDatabasePeer());
    Tags.DB_TYPE.set(span, "sql");
    Tags.DB_INSTANCE.set(span, connectInfo.getDatabaseName());
    Tags.DB_STATEMENT.set(span, cacheObject.getSql());
    span.setComponent(connectInfo.getComponent());
    SpanLayer.asDB(span);
}