fix: Propogate context in SinglePartitionPublisherBuilder. (#63)

Co-authored-by: dpcollins-google <[email protected]>

Author: dpcollins-google

google-cloud-pubsublite/src/main/java/com/google/cloud/pubsublite/internal/wire/SinglePartitionPublisherBuilder.java

@@ -64,7 +64,10 @@ public abstract static class Builder {
     public SinglePartitionPublisher build() throws StatusException {
       SinglePartitionPublisherBuilder builder = autoBuild();
       PublisherBuilder.Builder publisherBuilder =
-          PublisherBuilder.builder().setTopic(builder.topic()).setPartition(builder.partition());
+          PublisherBuilder.builder()
+              .setTopic(builder.topic())
+              .setPartition(builder.partition())
+              .setContext(builder.context());
       builder.stub().ifPresent(publisherBuilder::setStub);
       builder.batchingSettings().ifPresent(publisherBuilder::setBatching);
       return new SinglePartitionPublisher(publisherBuilder.build(), builder.partition());

pubsublite-beam-io/src/main/java/com/google/cloud/pubsublite/beam/PubsubLiteSink.java

@@ -115,7 +115,8 @@ public void onFailure(Throwable t) {
               errorsSinceLastFinish.push(ExtractStatus.toCanonical(t));
             }
           }
-        }, executor);
+        },
+        executor);
   }
 
   // Intentionally don't flush on bundle finish to allow multi-sink client reuse.