Skip to content

Commit 409d556

Browse files
spkrkasduskis
authored andcommitted
---
yaml --- r: 20243 b: refs/heads/autosynth-redis c: bd8d2ce h: refs/heads/master i: 20241: bbbf5bf 20239: 07d2790
1 parent 9fcdbab commit 409d556

2 files changed

Lines changed: 8 additions & 19 deletions

File tree

  • branches/autosynth-redis/google-cloud-clients/google-cloud-logging/src/main/java/com/google/cloud/logging

[refs]

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,7 @@ refs/heads/autosynth-iot: f03bdd338a9056ca3b7ea6d9ca901649ba9aab78
136136
refs/heads/autosynth-kms: 2828edfe3d2c53dd6e71912eae8a53c87bf40c87
137137
refs/heads/autosynth-language: c3d990dd34d81e7e935041e7147fb9dd27f8a557
138138
refs/heads/autosynth-os-login: 092fdbed6d5317948f92b708e9f50dedd89fc666
139-
refs/heads/autosynth-redis: 03ba9c91cab9113f30bfc43d185fc87d1b515303
139+
refs/heads/autosynth-redis: bd8d2cecdfb56b35d41a86697eda279e34f19bcf
140140
refs/heads/autosynth-scheduler: d97f8743ba965c7d5e492c8dc1f51d023104e260
141141
refs/heads/autosynth-spanner: 9bff86d057df31e04c76d72865e8e073ac5794fb
142142
refs/heads/autosynth-speech: 75d6c62a9d07d3a3642980502a25d07fbde0f232

branches/autosynth-redis/google-cloud-clients/google-cloud-logging/src/main/java/com/google/cloud/logging/LoggingImpl.java

Lines changed: 7 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -70,11 +70,9 @@
7070
import com.google.logging.v2.WriteLogEntriesResponse;
7171
import com.google.protobuf.Empty;
7272
import java.util.ArrayList;
73-
import java.util.Collections;
74-
import java.util.IdentityHashMap;
7573
import java.util.List;
7674
import java.util.Map;
77-
import java.util.Set;
75+
import java.util.concurrent.ConcurrentHashMap;
7876
import java.util.concurrent.ExecutionException;
7977
import java.util.concurrent.TimeUnit;
8078
import java.util.concurrent.TimeoutException;
@@ -83,9 +81,7 @@ class LoggingImpl extends BaseService<LoggingOptions> implements Logging {
8381

8482
private static final int FLUSH_WAIT_TIMEOUT_SECONDS = 6;
8583
private final LoggingRpc rpc;
86-
private final Object writeLock = new Object();
87-
private final Set<ApiFuture<Void>> pendingWrites =
88-
Collections.newSetFromMap(new IdentityHashMap<ApiFuture<Void>, Boolean>());
84+
private final Map<Object, ApiFuture<Void>> pendingWrites = new ConcurrentHashMap<>();
8985

9086
private volatile Synchronicity writeSynchronicity = Synchronicity.ASYNC;
9187
private volatile Severity flushSeverity = null;
@@ -575,9 +571,7 @@ public void flush() {
575571
// BUG(1795): We should force batcher to issue RPC call for buffered messages,
576572
// so the code below doesn't wait uselessly.
577573
ArrayList<ApiFuture<Void>> writesToFlush = new ArrayList<>();
578-
synchronized (writeLock) {
579-
writesToFlush.addAll(pendingWrites);
580-
}
574+
writesToFlush.addAll(pendingWrites.values());
581575

582576
try {
583577
ApiFutures.allAsList(writesToFlush).get(FLUSH_WAIT_TIMEOUT_SECONDS, TimeUnit.SECONDS);
@@ -596,16 +590,13 @@ private void writeLogEntries(Iterable<LogEntry> logEntries, WriteOption... write
596590
case ASYNC:
597591
default:
598592
final ApiFuture<Void> writeFuture = writeAsync(logEntries, writeOptions);
599-
synchronized (writeLock) {
600-
pendingWrites.add(writeFuture);
601-
}
593+
final Object pendingKey = new Object();
594+
pendingWrites.put(pendingKey, writeFuture);
602595
ApiFutures.addCallback(
603596
writeFuture,
604597
new ApiFutureCallback<Void>() {
605598
private void removeFromPending() {
606-
synchronized (writeLock) {
607-
pendingWrites.remove(writeFuture);
608-
}
599+
pendingWrites.remove(pendingKey);
609600
}
610601

611602
@Override
@@ -711,8 +702,6 @@ public void close() throws Exception {
711702

712703
@VisibleForTesting
713704
int getNumPendingWrites() {
714-
synchronized (writeLock) {
715-
return pendingWrites.size();
716-
}
705+
return pendingWrites.size();
717706
}
718707
}

0 commit comments

Comments
 (0)