Skip to content

Commit 6d5f172

Browse files
Add runner logs
1 parent 21733d5 commit 6d5f172

File tree

5 files changed

+32
-0
lines changed

5 files changed

+32
-0
lines changed

google-cloud-spanner/src/main/java/com/google/cloud/spanner/AsyncResultSetImpl.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -340,6 +340,7 @@ public void run() {
340340
private class ProduceRowsRunnable implements Runnable {
341341
@Override
342342
public void run() {
343+
System.out.println("ProduceRowsRunnable");
343344
boolean stop = false;
344345
boolean hasNext = false;
345346
try {
@@ -515,6 +516,7 @@ private void initiateProduceRows() {
515516
}
516517
produceRowsInitiated = true;
517518
}
519+
System.out.println("initiateProduceRows");
518520
this.service.execute(new ProduceRowsRunnable());
519521
}
520522

google-cloud-spanner/src/main/java/com/google/cloud/spanner/GrpcResultSet.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,7 @@ public boolean next() throws SpannerException {
7878
if (error != null) {
7979
throw newSpannerException(error);
8080
}
81+
System.out.println("GrpcResultSet next");
8182
try {
8283
if (currRow == null) {
8384
metadata = iterator.getMetadata();

google-cloud-spanner/src/main/java/com/google/cloud/spanner/ResumableStreamIterator.java

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -226,6 +226,7 @@ public boolean isWithBeginTransaction() {
226226
@Override
227227
@InternalApi
228228
public boolean initiateStreaming(AsyncResultSet.StreamMessageListener streamMessageListener) {
229+
System.out.println("Inside initiateStreaming");
229230
this.streamMessageListener = streamMessageListener;
230231
startGrpcStreaming();
231232
return true;
@@ -317,6 +318,7 @@ && prepareIteratorForRetryOnDifferentGrpcChannel()) {
317318
}
318319

319320
private void startGrpcStreaming() {
321+
System.out.println("Starting Stream");
320322
if (stream == null) {
321323
span.addAnnotation(
322324
"Starting/Resuming stream",
@@ -325,6 +327,7 @@ private void startGrpcStreaming() {
325327
try (IScope scope = tracer.withSpan(span)) {
326328
// When start a new stream set the Span as current to make the gRPC Span a child of
327329
// this Span.
330+
System.out.println("Creating Stream");
328331
stream = checkNotNull(startStream(resumeToken, streamMessageListener));
329332
}
330333
}

google-cloud-spanner/src/test/java/com/google/cloud/spanner/MockSpannerServiceImpl.java

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -651,6 +651,7 @@ private ByteString generateTransactionName(String session) {
651651
counter = new AtomicLong();
652652
transactionCounters.put(session, counter);
653653
}
654+
System.out.printf("Generating session using Session ID %s\n", session);
654655
return ByteString.copyFromUtf8(
655656
String.format("%s/transactions/%d", session, counter.incrementAndGet()));
656657
}
@@ -1658,6 +1659,7 @@ public void read(final ReadRequest request, StreamObserver<ResultSet> responseOb
16581659
@Override
16591660
public void streamingRead(
16601661
final ReadRequest request, StreamObserver<PartialResultSet> responseObserver) {
1662+
System.out.println("StreamingRead");
16611663
requests.add(request);
16621664
Preconditions.checkNotNull(request.getSession());
16631665
Session session = getSession(request.getSession());
@@ -1999,6 +2001,8 @@ private void ensureMostRecentTransaction(Session session, ByteString transaction
19992001
if (index > -1) {
20002002
long id = Long.parseLong(transactionId.toStringUtf8().substring(index + 1));
20012003
if (id != counter.get()) {
2004+
System.out.printf(
2005+
"Session ID %s TransactionId %s\n", session.getName(), transactionId.toStringUtf8());
20022006
throw Status.FAILED_PRECONDITION
20032007
.withDescription(
20042008
String.format(

google-cloud-spanner/src/test/java/com/google/cloud/spanner/RetryOnInvalidatedSessionTest.java

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,10 @@
6060
import org.junit.AfterClass;
6161
import org.junit.Before;
6262
import org.junit.BeforeClass;
63+
import org.junit.Rule;
6364
import org.junit.Test;
65+
import org.junit.rules.TestWatcher;
66+
import org.junit.runner.Description;
6467
import org.junit.runner.RunWith;
6568
import org.junit.runners.Parameterized;
6669
import org.junit.runners.Parameterized.Parameter;
@@ -75,6 +78,25 @@ public Long apply(StructReader input) {
7578
}
7679
}
7780

81+
@Rule
82+
public TestWatcher testWatcher =
83+
new TestWatcher() {
84+
@Override
85+
protected void succeeded(Description description) {
86+
System.out.println("Succeeded test " + description.getMethodName());
87+
}
88+
89+
@Override
90+
protected void failed(Throwable e, Description description) {
91+
System.out.println("Failed test " + description.getMethodName());
92+
}
93+
94+
@Override
95+
protected void starting(Description description) {
96+
System.out.println("Starting test " + description.getMethodName());
97+
}
98+
};
99+
78100
private static final ToLongTransformer TO_LONG = new ToLongTransformer();
79101

80102
@Parameter(0)

0 commit comments

Comments
 (0)