Java Span类代码示例

import org.htrace.Span; //导入依赖的package包/类
/** Construct a new output stream for creating a file. */
private DFSOutputStream(DFSClient dfsClient, String src, HdfsFileStatus stat,
    EnumSet<CreateFlag> flag, Progressable progress,
    DataChecksum checksum, String[] favoredNodes) throws IOException {
  this(dfsClient, src, progress, stat, checksum);
  this.shouldSyncBlock = flag.contains(CreateFlag.SYNC_BLOCK);

  computePacketChunkSize(dfsClient.getConf().writePacketSize, bytesPerChecksum);

  Span traceSpan = null;
  if (Trace.isTracing()) {
    traceSpan = Trace.startSpan(this.getClass().getSimpleName()).detach();
  }
  streamer = new DataStreamer(stat, traceSpan);
  if (favoredNodes != null && favoredNodes.length != 0) {
    streamer.setFavoredNodes(favoredNodes);
  }
}
 

import org.htrace.Span; //导入依赖的package包/类
public static RpcRequestHeaderProto makeRpcRequestHeader(RPC.RpcKind rpcKind,
    RpcRequestHeaderProto.OperationProto operation, int callId,
    int retryCount, byte[] uuid) {
  RpcRequestHeaderProto.Builder result = RpcRequestHeaderProto.newBuilder();
  result.setRpcKind(convert(rpcKind)).setRpcOp(operation).setCallId(callId)
      .setRetryCount(retryCount).setClientId(ByteString.copyFrom(uuid));

  // Add tracing info if we are currently tracing.
  if (Trace.isTracing()) {
    Span s = Trace.currentSpan();
    result.setTraceInfo(RPCTraceInfoProto.newBuilder()
        .setParentId(s.getSpanId())
        .setTraceId(s.getTraceId()).build());
  }

  return result.build();
}
 

import org.htrace.Span; //导入依赖的package包/类
/**
 * construction with tracing info
 */
private DataStreamer(HdfsFileStatus stat, Span span) {
  isAppend = false;
  isLazyPersistFile = isLazyPersist(stat);
  stage = BlockConstructionStage.PIPELINE_SETUP_CREATE;
  traceSpan = span;
}
 

import org.htrace.Span; //导入依赖的package包/类
@Override
public void releaseShortCircuitFds(SlotId slotId) throws IOException {
  ReleaseShortCircuitAccessRequestProto.Builder builder =
      ReleaseShortCircuitAccessRequestProto.newBuilder().
      setSlotId(PBHelper.convert(slotId));
  if (Trace.isTracing()) {
    Span s = Trace.currentSpan();
    builder.setTraceInfo(DataTransferTraceInfoProto.newBuilder()
        .setTraceId(s.getTraceId()).setParentId(s.getSpanId()));
  }
  ReleaseShortCircuitAccessRequestProto proto = builder.build();
  send(out, Op.RELEASE_SHORT_CIRCUIT_FDS, proto);
}
 

import org.htrace.Span; //导入依赖的package包/类
@Override
public void requestShortCircuitShm(String clientName) throws IOException {
  ShortCircuitShmRequestProto.Builder builder =
      ShortCircuitShmRequestProto.newBuilder().
      setClientName(clientName);
  if (Trace.isTracing()) {
    Span s = Trace.currentSpan();
    builder.setTraceInfo(DataTransferTraceInfoProto.newBuilder()
        .setTraceId(s.getTraceId()).setParentId(s.getSpanId()));
  }
  ShortCircuitShmRequestProto proto = builder.build();
  send(out, Op.REQUEST_SHORT_CIRCUIT_SHM, proto);
}
 

import org.htrace.Span; //导入依赖的package包/类
static BaseHeaderProto buildBaseHeader(ExtendedBlock blk,
    Token<BlockTokenIdentifier> blockToken) {
  BaseHeaderProto.Builder builder =  BaseHeaderProto.newBuilder()
    .setBlock(PBHelper.convert(blk))
    .setToken(PBHelper.convert(blockToken));
  if (Trace.isTracing()) {
    Span s = Trace.currentSpan();
    builder.setTraceInfo(DataTransferTraceInfoProto.newBuilder()
        .setTraceId(s.getTraceId())
        .setParentId(s.getSpanId()));
  }
  return builder.build();
}
 

import org.htrace.Span; //导入依赖的package包/类
public static Map<String, List<Span>> getMap() {
  Map<String, List<Span>> map = new HashMap<String, List<Span>>();

  for (Span s : spans.values()) {
    List<Span> l = map.get(s.getDescription());
    if (l == null) {
      l = new LinkedList<Span>();
      map.put(s.getDescription(), l);
    }
    l.add(s);
  }
  return map;
}
 

import org.htrace.Span; //导入依赖的package包/类
public Call(int id, int retryCount, Writable param, Connection connection,
    RPC.RpcKind kind, byte[] clientId, Span span) {
  this.callId = id;
  this.retryCount = retryCount;
  this.rpcRequest = param;
  this.connection = connection;
  this.timestamp = Time.now();
  this.rpcResponse = null;
  this.rpcKind = kind;
  this.clientId = clientId;
  this.traceSpan = span;
}
 

import org.htrace.Span; //导入依赖的package包/类
private SyncFuture publishSyncOnRingBuffer(Span span) {
  long sequence = this.disruptor.getRingBuffer().next();
  SyncFuture syncFuture = getSyncFuture(sequence, span);
  try {
    RingBufferTruck truck = this.disruptor.getRingBuffer().get(sequence);
    truck.loadPayload(syncFuture);
  } finally {
    this.disruptor.getRingBuffer().publish(sequence);
  }
  return syncFuture;
}
 

import org.htrace.Span; //导入依赖的package包/类
private Span blockOnSync(final SyncFuture syncFuture) throws IOException {
  // Now we have published the ringbuffer, halt the current thread until we get an answer back.
  try {
    syncFuture.get();
    return syncFuture.getSpan();
  } catch (InterruptedException ie) {
    LOG.warn("Interrupted", ie);
    throw convertInterruptedExceptionToIOException(ie);
  } catch (ExecutionException e) {
    throw ensureIOException(e.getCause());
  }
}
 

import org.htrace.Span; //导入依赖的package包/类
private SyncFuture getSyncFuture(final long sequence, Span span) {
  SyncFuture syncFuture = this.syncFuturesByHandler.get(Thread.currentThread());
  if (syncFuture == null) {
    syncFuture = new SyncFuture();
    this.syncFuturesByHandler.put(Thread.currentThread(), syncFuture);
  }
  return syncFuture.reset(sequence, span);
}
 

import org.htrace.Span; //导入依赖的package包/类
/**
 * Call this method to clear old usage and get it ready for new deploy. Call
 * this method even if it is being used for the first time.
 *
 * @param sequence sequenceId from this Future's position in the RingBuffer
 * @param span curren span, detached from caller. Don't forget to attach it when
 *             resuming after a call to {@link #get()}.
 * @return this
 */
synchronized SyncFuture reset(final long sequence, Span span) {
  if (t != null && t != Thread.currentThread()) throw new IllegalStateException();
  t = Thread.currentThread();
  if (!isDone()) throw new IllegalStateException("" + sequence + " " + Thread.currentThread());
  this.doneSequence = NOT_DONE;
  this.ringBufferSequence = sequence;
  this.span = span;
  return this;
}
 

import org.htrace.Span; //导入依赖的package包/类
/**
 * Add a result to this aggregate result.
 * @param time Time in nanoseconds
 * @param span Span.  To be kept if the time taken was over 1 second
 */
public void addResult(long time, Span span) {
  stats.addValue(TimeUnit.MILLISECONDS.convert(time, TimeUnit.NANOSECONDS));
  if (TimeUnit.SECONDS.convert(time, TimeUnit.NANOSECONDS) >= 1) {
    traces.add(span.getTraceId());
  }
}
 

import org.htrace.Span; //导入依赖的package包/类
public CallFuture sendCall(Call call, int priority, Span span)
    throws InterruptedException, IOException {
  CallFuture cts = new CallFuture(call, priority, span);
  callsToWrite.add(cts);
  checkIsOpen(); // We check after the put, to be sure that the call we added won't stay
                 //  in the list while the cleanup was already done.
  return cts;
}
 

import org.htrace.Span; //导入依赖的package包/类
protected void tracedWriteRequest(Call call, int priority, Span span) throws IOException {
  TraceScope ts = Trace.continueSpan(span);
  try {
    writeRequest(call, priority, span);
  } finally {
    ts.close();
  }
}