TracingListener => OpenTracingListener

 * </p>

 */

package io.shardingsphere.opentracing.tag;

package io.shardingsphere.opentracing;

import io.opentracing.tag.StringTag;

import lombok.AccessLevel;

import lombok.NoArgsConstructor;

/**

 * Local tags.

 * Sharding tags.

 *

 * @author chenqingyang

 */

@NoArgsConstructor(access = AccessLevel.PRIVATE)

public final class LocalTags {

public final class ShardingTags {

    public static final String COMPONENT_NAME = "SHARDING-SPHERE";

    /**

     * DB_BIND_VARIABLES records the bind variables of sql statement.

     * Records the bind variables of SQL.

     */

    public static final StringTag DB_BIND_VARIABLES = new StringTag("db.bind_vars");

}

package io.shardingsphere.opentracing.listener;

import io.opentracing.BaseSpan;

import io.opentracing.tag.Tags;

import io.shardingsphere.core.event.ShardingEvent;

import io.shardingsphere.core.event.ShardingEventBusInstance;

import java.util.Map;

/**

 * Tracing listener.

 * Opentracing listener.

 * 

 * @author zhangliang

 * 

 * @param <T> type of sharding event

 */

public abstract class TracingListener<T extends ShardingEvent> {

public abstract class OpenTracingListener<T extends ShardingEvent> {

    /**

     * Register listener.

        ShardingEventBusInstance.getInstance().register(this);

    }

    @SuppressWarnings("unchecked")

    protected final void tracing(final T event) {

        switch (event.getEventType()) {

            case BEFORE_EXECUTE:

                tracingFinish();

                break;

            case EXECUTE_FAILURE:

                tracingFailure(event);

                getFailureSpan().setTag(Tags.ERROR.getKey(), true).log(System.currentTimeMillis(), getReason(event.getException()));

                tracingFinish();

                break;

            default:

    protected abstract void tracingFinish();

    protected abstract void tracingFailure(T event);

    protected abstract BaseSpan<?> getFailureSpan();

    protected final Map<String, ?> log(final Throwable cause) {

    private Map<String, ?> getReason(final Throwable cause) {

        Map<String, String> result = new HashMap<>(3, 1);

        result.put("event", "error");

        result.put("error.kind", cause.getClass().getName());

    public void listen(final DMLExecutionEvent event) {

        tracing(event);

    }

    @Override

    protected String getOperation() {

        return "MODIFY";

    }

}

    public void listen(final DQLExecutionEvent event) {

        tracing(event);

    }

    @Override

    protected String getOperation() {

        return "QUERY";

    }

}

import com.google.common.base.Joiner;

import io.opentracing.ActiveSpan;

import io.opentracing.Span;

import io.opentracing.Tracer;

import io.opentracing.tag.Tags;

import io.shardingsphere.core.executor.event.SQLExecutionEvent;

import io.shardingsphere.core.executor.threadlocal.ExecutorDataMap;

import io.shardingsphere.opentracing.ShardingTracer;

import io.shardingsphere.opentracing.listener.TracingListener;

import io.shardingsphere.opentracing.tag.LocalTags;

import io.shardingsphere.opentracing.listener.OpenTracingListener;

import io.shardingsphere.opentracing.ShardingTags;

/**

 * SQL execute event listener.

 * @author wangkai

 * @author maxiaoguang

 */

public abstract class ExecuteEventListener extends TracingListener<SQLExecutionEvent> {

    private static final String SNAPSHOT_DATA_KEY = "OPENTRACING_SNAPSHOT_DATA";

public abstract class ExecuteEventListener extends OpenTracingListener<SQLExecutionEvent> {

    private static final String OPERATION_NAME_PREFIX = "/SHARDING-SPHERE/EXECUTE/";

    private static final String SNAPSHOT_DATA_KEY = "OPENTRACING_SNAPSHOT_DATA";

    private final ThreadLocal<ActiveSpan> trunkSpan = new ThreadLocal<>();

    private final ThreadLocal<Span> branchSpan = new ThreadLocal<>();

    @Override

    protected final void beforeExecute(final SQLExecutionEvent event) {

        Tracer tracer = ShardingTracer.get();

        if (ExecutorDataMap.getDataMap().containsKey(SNAPSHOT_DATA_KEY) && null == trunkSpan.get() && null == branchSpan.get()) {

            trunkInBranchSpan.set(((ActiveSpan.Continuation) ExecutorDataMap.getDataMap().get(SNAPSHOT_DATA_KEY)).activate());

        }

        if (null == branchSpan.get()) {

            branchSpan.set(tracer.buildSpan(OPERATION_NAME_PREFIX + getOperation()).withTag(Tags.SPAN_KIND.getKey(), Tags.SPAN_KIND_CLIENT)

                    .withTag(Tags.PEER_HOSTNAME.getKey(), event.getDataSource()).withTag(Tags.COMPONENT.getKey(), LocalTags.COMPONENT_NAME)

            branchSpan.set(ShardingTracer.get().buildSpan(OPERATION_NAME_PREFIX).withTag(Tags.SPAN_KIND.getKey(), Tags.SPAN_KIND_CLIENT)

                    .withTag(Tags.PEER_HOSTNAME.getKey(), event.getDataSource()).withTag(Tags.COMPONENT.getKey(), ShardingTags.COMPONENT_NAME)

                    .withTag(Tags.DB_INSTANCE.getKey(), event.getDataSource()).withTag(Tags.DB_TYPE.getKey(), "sql")

                    .withTag(LocalTags.DB_BIND_VARIABLES.getKey(), event.getParameters().isEmpty() ? "" : Joiner.on(",").join(event.getParameters()))

                    .withTag(ShardingTags.DB_BIND_VARIABLES.getKey(), event.getParameters().isEmpty() ? "" : Joiner.on(",").join(event.getParameters()))

                    .withTag(Tags.DB_STATEMENT.getKey(), event.getSqlUnit().getSql()).startManual());

        }

    }

    }

    @Override

    protected final void tracingFailure(final SQLExecutionEvent event) {

        branchSpan.get().setTag(Tags.ERROR.getKey(), true).log(System.currentTimeMillis(), log(event.getException()));

    protected final Span getFailureSpan() {

        return branchSpan.get();

    }

    protected abstract String getOperation();

}