---

spkrka · sduskis · commit 0475ca8b4a86 · 2019-03-19T07:16:52.000-04:00
yaml --- r: 23977 b: refs/heads/autosynth-kms c: 03ba9c9 h: refs/heads/master i: 23975: cab7b05
diff --git a/[refs] b/[refs]
@@ -132,7 +132,7 @@ refs/heads/autosynth-dialogflow: cb1ceeb3ec1a132d3096fc88d44b930a52130e18
 refs/heads/autosynth-errorreporting: 3f176c20b55dfaaa8fc32f28d82b31784b93e636
 refs/heads/autosynth-firestore: d48d82caace227856b6cd85ac30ee474528733ea
 refs/heads/autosynth-iot: 4025d1804241e74d54950a324dc4f667aeaad4b3
-refs/heads/autosynth-kms: 0a6fc39ab502f112b3cbdd671d8c79189c51bcdf
+refs/heads/autosynth-kms: 03ba9c91cab9113f30bfc43d185fc87d1b515303
 refs/heads/autosynth-language: c3d990dd34d81e7e935041e7147fb9dd27f8a557
 refs/heads/autosynth-os-login: 092fdbed6d5317948f92b708e9f50dedd89fc666
 refs/heads/autosynth-redis: 9e1fe503973c7b4a9ba26afb1fcddc2a57ba795a
diff --git a/branches/autosynth-kms/google-cloud-clients/google-cloud-logging/src/main/java/com/google/cloud/logging/LoggingImpl.java b/branches/autosynth-kms/google-cloud-clients/google-cloud-logging/src/main/java/com/google/cloud/logging/LoggingImpl.java
@@ -37,6 +37,7 @@
 import com.google.cloud.MonitoredResourceDescriptor;
 import com.google.cloud.PageImpl;
 import com.google.cloud.logging.spi.v2.LoggingRpc;
+import com.google.common.annotations.VisibleForTesting;
 import com.google.common.base.Function;
 import com.google.common.base.Throwables;
 import com.google.common.collect.ImmutableList;
@@ -595,6 +596,9 @@ private void writeLogEntries(Iterable<LogEntry> logEntries, WriteOption... write
       case ASYNC:
       default:
         final ApiFuture<Void> writeFuture = writeAsync(logEntries, writeOptions);
+        synchronized (writeLock) {
+          pendingWrites.add(writeFuture);
+        }
         ApiFutures.addCallback(
             writeFuture,
             new ApiFutureCallback<Void>() {
@@ -619,9 +623,6 @@ public void onFailure(Throwable t) {
                 }
               }
             });
-        synchronized (writeLock) {
-          pendingWrites.add(writeFuture);
-        }
         break;
     }
   }
@@ -707,4 +708,11 @@ public void close() throws Exception {
     }
     return optionMap;
   }
+
+  @VisibleForTesting
+  int getNumPendingWrites() {
+    synchronized (writeLock) {
+      return pendingWrites.size();
+    }
+  }
 }
diff --git a/branches/autosynth-kms/google-cloud-clients/google-cloud-logging/src/test/java/com/google/cloud/logging/LoggingImplTest.java b/branches/autosynth-kms/google-cloud-clients/google-cloud-logging/src/test/java/com/google/cloud/logging/LoggingImplTest.java
@@ -1186,6 +1186,7 @@ public void testWriteLogEntries() {
     EasyMock.replay(rpcFactoryMock, loggingRpcMock);
     logging = options.getService();
     logging.write(ImmutableList.of(LOG_ENTRY1, LOG_ENTRY2));
+    assertEquals(0, ((LoggingImpl) logging).getNumPendingWrites());
   }
 
   @Test

Original file line number	Diff line number	Diff line change
`@@ -37,6 +37,7 @@`
`37`	`37`	`import com.google.cloud.MonitoredResourceDescriptor;`
`38`	`38`	`import com.google.cloud.PageImpl;`
`39`	`39`	`import com.google.cloud.logging.spi.v2.LoggingRpc;`
	`40`	`+import com.google.common.annotations.VisibleForTesting;`
`40`	`41`	`import com.google.common.base.Function;`
`41`	`42`	`import com.google.common.base.Throwables;`
`42`	`43`	`import com.google.common.collect.ImmutableList;`
`@@ -595,6 +596,9 @@ private void writeLogEntries(Iterable<LogEntry> logEntries, WriteOption... write`
`595`	`596`	`case ASYNC:`
`596`	`597`	`default:`
`597`	`598`	`final ApiFuture<Void> writeFuture = writeAsync(logEntries, writeOptions);`
	`599`	`+ synchronized (writeLock) {`
	`600`	`+ pendingWrites.add(writeFuture);`
	`601`	`+ }`
`598`	`602`	`ApiFutures.addCallback(`
`599`	`603`	`writeFuture,`
`600`	`604`	`new ApiFutureCallback<Void>() {`
`@@ -619,9 +623,6 @@ public void onFailure(Throwable t) {`
`619`	`623`	`}`
`620`	`624`	`}`
`621`	`625`	`});`
`622`		`- synchronized (writeLock) {`
`623`		`- pendingWrites.add(writeFuture);`
`624`		`- }`
`625`	`626`	`break;`
`626`	`627`	`}`
`627`	`628`	`}`
`@@ -707,4 +708,11 @@ public void close() throws Exception {`
`707`	`708`	`}`
`708`	`709`	`return optionMap;`
`709`	`710`	`}`
	`711`	`+`
	`712`	`+ @VisibleForTesting`
	`713`	`+ int getNumPendingWrites() {`
	`714`	`+ synchronized (writeLock) {`
	`715`	`+ return pendingWrites.size();`
	`716`	`+ }`
	`717`	`+ }`
`710`	`718`	`}`
Original file line number	Diff line number	Diff line change
`@@ -1186,6 +1186,7 @@ public void testWriteLogEntries() {`
`1186`	`1186`	`EasyMock.replay(rpcFactoryMock, loggingRpcMock);`
`1187`	`1187`	`logging = options.getService();`
`1188`	`1188`	`logging.write(ImmutableList.of(LOG_ENTRY1, LOG_ENTRY2));`
	`1189`	`+ assertEquals(0, ((LoggingImpl) logging).getNumPendingWrites());`
`1189`	`1190`	`}`
`1190`	`1191`
`1191`	`1192`	`@Test`