---

spkrka · sduskis · commit 0c3cfa32929d · 2019-03-19T12:26:43.000-04:00
yaml --- r: 23835 b: refs/heads/autosynth-iot c: bd8d2ce h: refs/heads/master i: 23833: 41acc64 23831: f0c2b67
diff --git a/[refs] b/[refs]
@@ -131,7 +131,7 @@ refs/heads/autosynth-datastore: d0fc1187000c7a50d8bddd89c3b7cdf1187ecb96
 refs/heads/autosynth-dialogflow: cb1ceeb3ec1a132d3096fc88d44b930a52130e18
 refs/heads/autosynth-errorreporting: 3f176c20b55dfaaa8fc32f28d82b31784b93e636
 refs/heads/autosynth-firestore: d48d82caace227856b6cd85ac30ee474528733ea
-refs/heads/autosynth-iot: 03ba9c91cab9113f30bfc43d185fc87d1b515303
+refs/heads/autosynth-iot: bd8d2cecdfb56b35d41a86697eda279e34f19bcf
 refs/heads/autosynth-kms: 2828edfe3d2c53dd6e71912eae8a53c87bf40c87
 refs/heads/autosynth-language: c3d990dd34d81e7e935041e7147fb9dd27f8a557
 refs/heads/autosynth-os-login: 092fdbed6d5317948f92b708e9f50dedd89fc666
diff --git a/branches/autosynth-iot/google-cloud-clients/google-cloud-logging/src/main/java/com/google/cloud/logging/LoggingImpl.java b/branches/autosynth-iot/google-cloud-clients/google-cloud-logging/src/main/java/com/google/cloud/logging/LoggingImpl.java
@@ -70,11 +70,9 @@
 import com.google.logging.v2.WriteLogEntriesResponse;
 import com.google.protobuf.Empty;
 import java.util.ArrayList;
-import java.util.Collections;
-import java.util.IdentityHashMap;
 import java.util.List;
 import java.util.Map;
-import java.util.Set;
+import java.util.concurrent.ConcurrentHashMap;
 import java.util.concurrent.ExecutionException;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.TimeoutException;
@@ -83,9 +81,7 @@ class LoggingImpl extends BaseService<LoggingOptions> implements Logging {
 
   private static final int FLUSH_WAIT_TIMEOUT_SECONDS = 6;
   private final LoggingRpc rpc;
-  private final Object writeLock = new Object();
-  private final Set<ApiFuture<Void>> pendingWrites =
-      Collections.newSetFromMap(new IdentityHashMap<ApiFuture<Void>, Boolean>());
+  private final Map<Object, ApiFuture<Void>> pendingWrites = new ConcurrentHashMap<>();
 
   private volatile Synchronicity writeSynchronicity = Synchronicity.ASYNC;
   private volatile Severity flushSeverity = null;
@@ -575,9 +571,7 @@ public void flush() {
     // BUG(1795): We should force batcher to issue RPC call for buffered messages,
     // so the code below doesn't wait uselessly.
     ArrayList<ApiFuture<Void>> writesToFlush = new ArrayList<>();
-    synchronized (writeLock) {
-      writesToFlush.addAll(pendingWrites);
-    }
+    writesToFlush.addAll(pendingWrites.values());
 
     try {
       ApiFutures.allAsList(writesToFlush).get(FLUSH_WAIT_TIMEOUT_SECONDS, TimeUnit.SECONDS);
@@ -596,16 +590,13 @@ private void writeLogEntries(Iterable<LogEntry> logEntries, WriteOption... write
       case ASYNC:
       default:
         final ApiFuture<Void> writeFuture = writeAsync(logEntries, writeOptions);
-        synchronized (writeLock) {
-          pendingWrites.add(writeFuture);
-        }
+        final Object pendingKey = new Object();
+        pendingWrites.put(pendingKey, writeFuture);
         ApiFutures.addCallback(
             writeFuture,
             new ApiFutureCallback<Void>() {
               private void removeFromPending() {
-                synchronized (writeLock) {
-                  pendingWrites.remove(writeFuture);
-                }
+                pendingWrites.remove(pendingKey);
               }
 
               @Override
@@ -711,8 +702,6 @@ public void close() throws Exception {
 
   @VisibleForTesting
   int getNumPendingWrites() {
-    synchronized (writeLock) {
-      return pendingWrites.size();
-    }
+    return pendingWrites.size();
   }
 }
