java - GCP 发布/订阅客户端、GrpcSubscriberStub、超时

Question

我正在尝试将 GrpcSubscriberStub 用于“同步拉动”，如示例https://cloud.google.com/pubsub/docs/pull

SubscriberStubSettings subscriberStubSettings =
    SubscriberStubSettings.newBuilder()
        .setTransportChannelProvider(
            SubscriberStubSettings.defaultGrpcTransportProviderBuilder()
                .setMaxInboundMessageSize(20 << 20) // 20MB
                .build())
        .build();

try (SubscriberStub subscriber = GrpcSubscriberStub.create(subscriberStubSettings)) {
  // String projectId = "my-project-id";
  // String subscriptionId = "my-subscription-id";
  // int numOfMessages = 10;   // max number of messages to be pulled
  String subscriptionName = ProjectSubscriptionName.format(projectId, subscriptionId);
  PullRequest pullRequest =
      PullRequest.newBuilder()
          .setMaxMessages(numOfMessages)
          .setReturnImmediately(false) // return immediately if messages are not available
          .setSubscription(subscriptionName)
          .build();

  // use pullCallable().futureCall to asynchronously perform this operation
  PullResponse pullResponse = subscriber.pullCallable().call(pullRequest);
  List<String> ackIds = new ArrayList<>();
  for (ReceivedMessage message : pullResponse.getReceivedMessagesList()) {
    // handle received message
    // ...
    ackIds.add(message.getAckId());
  }
  // acknowledge received messages
  AcknowledgeRequest acknowledgeRequest =
      AcknowledgeRequest.newBuilder()
          .setSubscription(subscriptionName)
          .addAllAckIds(ackIds)
          .build();
  // use acknowledgeCallable().futureCall to asynchronously perform this operation
  subscriber.acknowledgeCallable().call(acknowledgeRequest);
  return pullResponse.getReceivedMessagesList();
}

但我不知道如何使用选项“setReturnImmediately(false)”设置拉取超时。

你能帮助我吗？

Answer 1

要更改拉取超时，您需要修改SubscriberStubSettings.

例如，将 pull 的总超时设置为 30 秒：

SubscriberStubSettings.Builder subscriberStubSettingsBuilder =
    SubscriberStubSettings.newBuilder();
subscriberStubSettingsBuilder
    .pullSettings()
    .setRetrySettings(
        subscriberStubSettingsBuilder.pullSettings().getRetrySettings().toBuilder()
            .setTotalTimeout(Duration.ofSeconds(30))
            .build());
SubscriberStubSettings subscriberStubSettings = subscriberStubSettingsBuilder.build();

这是基于这里的 SubscriberStubSettings 文档。