---

sduskis · web-flow · commit 0f2caa9c0734 · 2019-04-16T11:19:35.000-04:00
yaml --- r: 35813 b: refs/heads/autosynth-dataproc c: b1ebd2d h: refs/heads/master i: 35811: f2fa8c6
diff --git a/[refs] b/[refs]
@@ -165,7 +165,7 @@ refs/tags/v0.78.0: 62d4bd30605ab3578f9a08d84487fb0b33ac2ff5
 refs/tags/v0.79.0: 82287b570708748c411d05c40f3932cff9606feb
 refs/tags/v0.80.0: f745e744d38e4fe636f34d0e04795ba3d014287d
 refs/tags/v0.81.0: ed3a0c85339ea6b73560b9a570abfbb76b93a263
-refs/heads/autosynth-dataproc: 79b64df4705b7a47895ec02aae4e84e5b8107884
+refs/heads/autosynth-dataproc: b1ebd2d84384315f5788d1fa9f48c0a60265de20
 refs/heads/autosynth-securitycenter: b24087060036e623e57d2454ba5dabeaf1e530c5
 refs/heads/autosynth-talent: 4ca901879f86aab61091cea52e8a9b653639df24
 refs/tags/v0.82.0: 7b9807d5d0a400c757b8905fee768be4c85eba25
diff --git a/branches/autosynth-dataproc/google-cloud-clients/google-cloud-pubsub/src/main/java/com/google/cloud/pubsub/v1/Publisher.java b/branches/autosynth-dataproc/google-cloud-clients/google-cloud-pubsub/src/main/java/com/google/cloud/pubsub/v1/Publisher.java
@@ -86,8 +86,7 @@ public class Publisher {
   private final BatchingSettings batchingSettings;
 
   private final Lock messagesBatchLock;
-  private List<OutstandingPublish> messagesBatch;
-  private int batchedBytes;
+  private MessagesBatch messagesBatch;
 
   private final AtomicBoolean activeAlarm;
 
@@ -116,7 +115,7 @@ private Publisher(Builder builder) throws IOException {
     this.batchingSettings = builder.batchingSettings;
     this.messageTransform = builder.messageTransform;
 
-    messagesBatch = new LinkedList<>();
+    messagesBatch = new MessagesBatch();
     messagesBatchLock = new ReentrantLock();
     activeAlarm = new AtomicBoolean(false);
     executor = builder.executorProvider.getExecutor();
@@ -207,24 +206,19 @@ public ApiFuture<String> publish(PubsubMessage message) {
       // Check if the next message makes the current batch exceed the max batch byte size.
       if (!messagesBatch.isEmpty()
           && hasBatchingBytes()
-          && batchedBytes + messageSize >= getMaxBatchBytes()) {
-        batchToSend = new OutstandingBatch(messagesBatch, batchedBytes);
-        messagesBatch = new LinkedList<>();
-        batchedBytes = 0;
+          && messagesBatch.getBatchedBytes() + messageSize >= getMaxBatchBytes()) {
+        batchToSend = messagesBatch.popOutstandingBatch();
       }
 
       // Border case if the message to send is greater or equals to the max batch size then can't
       // be included in the current batch and instead sent immediately.
       if (!hasBatchingBytes() || messageSize < getMaxBatchBytes()) {
-        batchedBytes += messageSize;
-        messagesBatch.add(outstandingPublish);
+        messagesBatch.addMessage(outstandingPublish, messageSize);
 
         // If after adding the message we have reached the batch max messages then we have a batch
         // to send.
-        if (messagesBatch.size() == getBatchingSettings().getElementCountThreshold()) {
-          batchToSend = new OutstandingBatch(messagesBatch, batchedBytes);
-          messagesBatch = new LinkedList<>();
-          batchedBytes = 0;
+        if (messagesBatch.getMessagesCount() == getBatchingSettings().getElementCountThreshold()) {
+          batchToSend = messagesBatch.popOutstandingBatch();
         }
       }
       // Setup the next duration based delivery alarm if there are messages batched.
@@ -303,9 +297,7 @@ public void publishAllOutstanding() {
       if (messagesBatch.isEmpty()) {
         return;
       }
-      batchToSend = new OutstandingBatch(messagesBatch, batchedBytes);
-      messagesBatch = new LinkedList<>();
-      batchedBytes = 0;
+      batchToSend = messagesBatch.popOutstandingBatch();
     } finally {
       messagesBatchLock.unlock();
     }
@@ -640,4 +632,37 @@ public Publisher build() throws IOException {
       return new Publisher(this);
     }
   }
+
+  private static class MessagesBatch {
+    private List<OutstandingPublish> messages = new LinkedList<>();
+    private int batchedBytes;
+
+    private OutstandingBatch popOutstandingBatch() {
+      OutstandingBatch batch = new OutstandingBatch(messages, batchedBytes);
+      reset();
+      return batch;
+    }
+
+    private void reset() {
+      messages = new LinkedList<>();
+      batchedBytes = 0;
+    }
+
+    private boolean isEmpty() {
+      return messages.isEmpty();
+    }
+
+    private int getBatchedBytes() {
+      return batchedBytes;
+    }
+
+    private void addMessage(OutstandingPublish message, int messageSize) {
+      messages.add(message);
+      batchedBytes += messageSize;
+    }
+
+    private int getMessagesCount() {
+      return messages.size();
+    }
+  }
 }
