TransactionManagerImpl(SessionImpl session) { this.session = session; this.span = Tracing.getTracer().getCurrentSpan(); }
private AbstractReadContext(SessionImpl session, SpannerRpc rpc, int defaultPrefetchChunks) { this(session, rpc, defaultPrefetchChunks, Tracing.getTracer().getCurrentSpan()); }
@Override public void onCompleted() { tracer.getCurrentSpan().addAnnotation("Firestore.BatchGet: Complete"); List<DocumentSnapshot> documentSnapshots = new ArrayList<>(); for (DocumentReference documentReference : documentReferences) { documentSnapshots.add(resultMap.get(documentReference)); } futureList.set(documentSnapshots); } };
TransactionRunnerImpl( SessionImpl session, SpannerRpc rpc, Sleeper sleeper, int defaultPrefetchChunks) { this.session = session; this.sleeper = sleeper; this.span = Tracing.getTracer().getCurrentSpan(); this.txn = session.newTransaction(); }
@Override public void onError(Throwable throwable) { tracer.getCurrentSpan().addAnnotation("Firestore.BatchGet: Error"); futureList.setException(throwable); }
@Override public void onError(Throwable throwable) { Tracing.getTracer().getCurrentSpan().addAnnotation("Firestore.Query: Error"); documentObserver.onError(throwable); }
@Override public void onCompleted() { Tracing.getTracer() .getCurrentSpan() .addAnnotation( "Firestore.Query: Completed", ImmutableMap.of( "numDocuments", AttributeValue.longAttributeValue(numDocuments))); documentObserver.onCompleted(readTime); } };
@Override public void intercept(HttpRequest request) throws IOException { checkNotNull(request); if (this.interceptor != null) { this.interceptor.intercept(request); } if (propagationTextFormat != null && propagationTextFormatSetter != null) { SpanContext spanContext = tracer.getCurrentSpan().getContext(); if (!SpanContext.INVALID.equals(spanContext)) { propagationTextFormat.inject( spanContext, request.getHeaders(), propagationTextFormatSetter); } } } }
private static void backoffSleep(Context context, long backoffMillis) throws SpannerException { tracer .getCurrentSpan() .addAnnotation( "Backing off",
Span span = tracer.getCurrentSpan(); ExponentialBackOff backOff = newBackOff(); Context context = Context.current();
Span span = Tracing.getTracer().getCurrentSpan(); span.addAnnotation("Acquiring read write session"); Waiter waiter = null;
Span span = Tracing.getTracer().getCurrentSpan(); span.addAnnotation("Acquiring session"); Waiter waiter = null;
@Override public void onNext(RunQueryResponse response) { if (!firstResponse) { firstResponse = true; Tracing.getTracer().getCurrentSpan().addAnnotation("Firestore.Query: First response"); } if (response.hasDocument()) { numDocuments++; if (numDocuments % 100 == 0) { Tracing.getTracer() .getCurrentSpan() .addAnnotation("Firestore.Query: Received 100 documents"); } Document document = response.getDocument(); QueryDocumentSnapshot documentSnapshot = QueryDocumentSnapshot.fromDocument( firestore, Timestamp.fromProto(response.getReadTime()), document); documentObserver.onNext(documentSnapshot); } if (readTime == null) { readTime = Timestamp.fromProto(response.getReadTime()); } }
private void maybeCreateSession() { Span span = Tracing.getTracer().getCurrentSpan(); synchronized (lock) { if (numWaiters() >= numSessionsBeingCreated) { if (canCreateSession()) { span.addAnnotation("Creating session"); createSession(); } else if (options.isFailIfPoolExhausted()) { span.addAnnotation("Pool exhausted. Failing"); // throw specific exception throw newSpannerException( ErrorCode.RESOURCE_EXHAUSTED, "No session available in the pool. Maximum number of sessions in the pool can be" + " overridden by invoking SessionPoolOptions#Builder#setMaxSessions. Client can be made to block" + " rather than fail by setting SessionPoolOptions#Builder#setBlockIfPoolExhausted."); } } } } /**
.getCurrentSpan() .addAnnotation( "CloudFirestore.Commit",
.getCurrentSpan() .addAnnotation( "Firestore.BatchGet: Start",
tracer.getCurrentSpan().addAnnotation("Firestore.BatchGet: First response"); } else if (numResponses % 100 == 0) { tracer.getCurrentSpan().addAnnotation("Firestore.BatchGet: Received 100 responses");
.getCurrentSpan() .addAnnotation( "Firestore.Query: Start",
rewriteResponse.getTotalBytesRewritten().longValue()); } catch (IOException ex) { tracer.getCurrentSpan().setStatus(Status.UNKNOWN.withDescription(ex.getMessage())); throw translate(ex);
@Test public void getCurrentSpan_WithSpan() { assertThat(noopTracer.getCurrentSpan()).isSameAs(BlankSpan.INSTANCE); Scope ws = noopTracer.withSpan(span); try { assertThat(noopTracer.getCurrentSpan()).isSameAs(span); } finally { ws.close(); } assertThat(noopTracer.getCurrentSpan()).isSameAs(BlankSpan.INSTANCE); }