Examples with AbstractSpan org.apache.skywalking.apm.agent.core.context.trace.AbstractSpan used on opensource projects

Example 71 with AbstractSpan

use of org.apache.skywalking.apm.agent.core.context.trace.AbstractSpan in project incubator-skywalking by apache.

the class TracingContext method capture.

/**
 * Capture the snapshot of current context.
 *
 * @return the snapshot of context for cross-thread propagation
 * Ref to {@link AbstractTracerContext#capture()}
 */
@Override
public ContextSnapshot capture() {
    List<TraceSegmentRef> refs = this.segment.getRefs();
    ContextSnapshot snapshot = new ContextSnapshot(segment.getTraceSegmentId(), activeSpan().getSpanId(), segment.getRelatedGlobalTraces());
    int entryOperationId;
    String entryOperationName;
    int entryApplicationInstanceId;
    AbstractSpan firstSpan = first();
    if (refs != null && refs.size() > 0) {
        TraceSegmentRef ref = refs.get(0);
        entryOperationId = ref.getEntryOperationId();
        entryOperationName = ref.getEntryOperationName();
        entryApplicationInstanceId = ref.getEntryApplicationInstanceId();
    } else {
        entryOperationId = firstSpan.getOperationId();
        entryOperationName = firstSpan.getOperationName();
        entryApplicationInstanceId = this.segment.getApplicationInstanceId();
    }
    snapshot.setEntryApplicationInstanceId(entryApplicationInstanceId);
    if (entryOperationId == DictionaryUtil.nullValue()) {
        snapshot.setEntryOperationName(entryOperationName);
    } else {
        snapshot.setEntryOperationId(entryOperationId);
    }
    if (firstSpan.getOperationId() == DictionaryUtil.nullValue()) {
        snapshot.setParentOperationName(firstSpan.getOperationName());
    } else {
        snapshot.setParentOperationId(firstSpan.getOperationId());
    }
    return snapshot;
}

Example 72 with AbstractSpan

use of org.apache.skywalking.apm.agent.core.context.trace.AbstractSpan in project incubator-skywalking by apache.

the class TracingContext method createLocalSpan.

/**
 * Create a local span
 *
 * @param operationName most likely a local method signature, or business name.
 * @return the span represents a local logic block.
 * Ref to {@link LocalSpan}
 */
@Override
public AbstractSpan createLocalSpan(final String operationName) {
    if (isLimitMechanismWorking()) {
        NoopSpan span = new NoopSpan();
        return push(span);
    }
    AbstractSpan parentSpan = peek();
    final int parentSpanId = parentSpan == null ? -1 : parentSpan.getSpanId();
    AbstractTracingSpan span = (AbstractTracingSpan) DictionaryManager.findOperationNameCodeSection().findOrPrepare4Register(segment.getApplicationId(), operationName, false, false).doInCondition(new PossibleFound.FoundAndObtain() {

        @Override
        public Object doProcess(int operationId) {
            return new LocalSpan(spanIdGenerator++, parentSpanId, operationId);
        }
    }, new PossibleFound.NotFoundAndObtain() {

        @Override
        public Object doProcess() {
            return new LocalSpan(spanIdGenerator++, parentSpanId, operationName);
        }
    });
    span.start();
    return push(span);
}

Example 73 with AbstractSpan

use of org.apache.skywalking.apm.agent.core.context.trace.AbstractSpan in project incubator-skywalking by apache.

the class TracingContext method createExitSpan.

/**
 * Create an exit span
 *
 * @param operationName most likely a service name of remote
 * @param remotePeer the network id(ip:port, hostname:port or ip1:port1,ip2,port, etc.)
 * @return the span represent an exit point of this segment.
 * @see  ExitSpan
 */
@Override
public AbstractSpan createExitSpan(final String operationName, final String remotePeer) {
    AbstractSpan exitSpan;
    AbstractSpan parentSpan = peek();
    if (parentSpan != null && parentSpan.isExit()) {
        exitSpan = parentSpan;
    } else {
        final int parentSpanId = parentSpan == null ? -1 : parentSpan.getSpanId();
        exitSpan = (AbstractSpan) DictionaryManager.findNetworkAddressSection().find(remotePeer).doInCondition(new PossibleFound.FoundAndObtain() {

            @Override
            public Object doProcess(final int peerId) {
                if (isLimitMechanismWorking()) {
                    return new NoopExitSpan(peerId);
                }
                return DictionaryManager.findOperationNameCodeSection().findOnly(segment.getApplicationId(), operationName).doInCondition(new PossibleFound.FoundAndObtain() {

                    @Override
                    public Object doProcess(int operationId) {
                        return new ExitSpan(spanIdGenerator++, parentSpanId, operationId, peerId);
                    }
                }, new PossibleFound.NotFoundAndObtain() {

                    @Override
                    public Object doProcess() {
                        return new ExitSpan(spanIdGenerator++, parentSpanId, operationName, peerId);
                    }
                });
            }
        }, new PossibleFound.NotFoundAndObtain() {

            @Override
            public Object doProcess() {
                if (isLimitMechanismWorking()) {
                    return new NoopExitSpan(remotePeer);
                }
                return DictionaryManager.findOperationNameCodeSection().findOnly(segment.getApplicationId(), operationName).doInCondition(new PossibleFound.FoundAndObtain() {

                    @Override
                    public Object doProcess(int operationId) {
                        return new ExitSpan(spanIdGenerator++, parentSpanId, operationId, remotePeer);
                    }
                }, new PossibleFound.NotFoundAndObtain() {

                    @Override
                    public Object doProcess() {
                        return new ExitSpan(spanIdGenerator++, parentSpanId, operationName, remotePeer);
                    }
                });
            }
        });
        push(exitSpan);
    }
    exitSpan.start();
    return exitSpan;
}

Example 74 with AbstractSpan

use of org.apache.skywalking.apm.agent.core.context.trace.AbstractSpan in project incubator-skywalking by apache.

the class TracingContext method createEntrySpan.

/**
 * Create an entry span
 *
 * @param operationName most likely a service name
 * @return span instance.
 * Ref to {@link EntrySpan}
 */
@Override
public AbstractSpan createEntrySpan(final String operationName) {
    if (isLimitMechanismWorking()) {
        NoopSpan span = new NoopSpan();
        return push(span);
    }
    AbstractSpan entrySpan;
    final AbstractSpan parentSpan = peek();
    final int parentSpanId = parentSpan == null ? -1 : parentSpan.getSpanId();
    if (parentSpan != null && parentSpan.isEntry()) {
        entrySpan = (AbstractTracingSpan) DictionaryManager.findOperationNameCodeSection().findOnly(segment.getApplicationId(), operationName).doInCondition(new PossibleFound.FoundAndObtain() {

            @Override
            public Object doProcess(int operationId) {
                return parentSpan.setOperationId(operationId);
            }
        }, new PossibleFound.NotFoundAndObtain() {

            @Override
            public Object doProcess() {
                return parentSpan.setOperationName(operationName);
            }
        });
        return entrySpan.start();
    } else {
        entrySpan = (AbstractTracingSpan) DictionaryManager.findOperationNameCodeSection().findOnly(segment.getApplicationId(), operationName).doInCondition(new PossibleFound.FoundAndObtain() {

            @Override
            public Object doProcess(int operationId) {
                return new EntrySpan(spanIdGenerator++, parentSpanId, operationId);
            }
        }, new PossibleFound.NotFoundAndObtain() {

            @Override
            public Object doProcess() {
                return new EntrySpan(spanIdGenerator++, parentSpanId, operationName);
            }
        });
        entrySpan.start();
        return push(entrySpan);
    }
}

Example 75 with AbstractSpan

use of org.apache.skywalking.apm.agent.core.context.trace.AbstractSpan in project incubator-skywalking by apache.

the class ContextManager method createEntrySpan.

public static AbstractSpan createEntrySpan(String operationName, ContextCarrier carrier) {
    SamplingService samplingService = ServiceManager.INSTANCE.findService(SamplingService.class);
    AbstractSpan span;
    AbstractTracerContext context;
    if (carrier != null && carrier.isValid()) {
        samplingService.forceSampled();
        context = getOrCreate(operationName, true);
        span = context.createEntrySpan(operationName);
        context.extract(carrier);
    } else {
        context = getOrCreate(operationName, false);
        span = context.createEntrySpan(operationName);
    }
    return span;
}